test_business.py

from unittest import main
 
from datetime import datetime, timezone, timedelta
 
from gnucash import Account, \
    ACCT_TYPE_RECEIVABLE, ACCT_TYPE_INCOME, ACCT_TYPE_BANK, \
    GncNumeric
from gnucash.gnucash_business import Vendor, Employee, Customer, Job, Invoice, Entry
 
from test_book import BookSession
 
class BusinessSession(BookSession):
    def setUp(self):
        BookSession.setUp(self)
 
        self.today = datetime.today()
 
        self.bank = Account(self.book)
        self.bank.SetType(ACCT_TYPE_BANK)
        self.bank.SetCommodity(self.currencycurrency)
        self.income = Account(self.book)
        self.income.SetType(ACCT_TYPE_INCOME)
        self.income.SetCommodity(self.currencycurrency)
        self.receivable = Account(self.book)
        self.receivable.SetType(ACCT_TYPE_RECEIVABLE)
        self.receivable.SetCommodity(self.currencycurrency)
 
        self.customer = Customer(self.book,'CustomerID',self.currencycurrency)
        self.vendor = Vendor(self.book,'VendorID',self.currencycurrency)
        self.employee = Employee(self.book,'EmployeeID',self.currencycurrency)
        self.job = Job(self.book,'JobID',self.customer)
 
        self.invoice = Invoice(self.book,'InvoiceID',self.currencycurrency,self.customer)
        self.invoice.SetDateOpened(self.today)
        entry = Entry(self.book)
        entry.SetDate(self.today)
        entry.SetDescription("Some income")
        entry.SetQuantity(GncNumeric(1))
        entry.SetInvAccount(self.income)
        entry.SetInvPrice(GncNumeric(100))
        self.invoice.AddEntry(entry)
 
        self.invoice.PostToAccount(self.receivable,
            self.today, self.today, "", True, False)
 
class TestBusiness(BusinessSession):
    def test_equal(self):
        self.assertTrue( self.vendor.Equal( self.vendor.GetVendor() ) )
        self.assertTrue( self.customer.Equal( self.job.GetOwner() ) )
        self.assertTrue( self.customer.Equal( self.invoice.GetOwner() ) )
 
    def test_employee_name(self):
        NAME = 'John Doe'
        self.assertEqual( '', self.employee.GetUsername() )
        self.employee.SetUsername(NAME)
        self.assertEqual( NAME, self.employee.GetUsername() )
 
    def test_post(self):
        utc_offset = datetime.now().astimezone().utcoffset()
        now = datetime.now().astimezone()
        neutral_time = (now + utc_offset).astimezone(timezone.utc).replace(hour=10, minute=59, second=0, microsecond=0)
        if utc_offset > timedelta(hours=13):
            neutral_time -= utc_offset - timedelta(hours=13);
        if utc_offset < timedelta(hours=-10):
            neutral_time += timedelta(hours=-10) - utc_offset
        self.assertEqual(neutral_time,
                         self.invoice.GetDatePosted().astimezone(timezone.utc))
        self.assertTrue( self.invoice.IsPosted() )
 
    def test_owner(self):
        OWNER = self.invoice.GetOwner()
        self.assertTrue( self.customer.Equal( OWNER ) )
 
    def test_commodities(self):
        self.assertTrue( self.currencycurrency.equal( self.customer.GetCommoditiesList()[0] ) )
 
    def test_invoice_transaction(self):
        """
        Test that you can get the posted transaction from a posted invoice and that you can get the invoice back from the transaction.
        """
        posted_transaction = self.invoice.GetPostedTxn()
        self.assertTrue( posted_transaction != None )
        invoice_from_transaction = posted_transaction.GetInvoiceFromTxn()
        self.assertTrue( invoice_from_transaction != None and invoice_from_transaction.GetID() == self.invoice.GetID() )
 
    def test_invoice_payment(self):
        """
        Test that you can pay an invoice into a transfer account (this also tests conversion of Python lists to GList).
        """
        self.receivablereceivable.RecomputeBalance()
        self.assertTrue( int(self.receivablereceivable.GetBalance().to_double()) == 100 )
        lots = [self.invoice.GetPostedLot()]
        self.customer.ApplyPayment(None, lots, self.receivablereceivable, self.bankbank, GncNumeric(100), GncNumeric(1), self.invoice.GetDatePosted(), "Paid into bank account.", "1234", False)
        self.receivablereceivable.RecomputeBalance()
        self.assertTrue( self.receivablereceivable.GetBalance().to_fraction().numerator == 0 )
 
if __name__ == '__main__':
    main()