def setUp(self):
self.default_company = Company(name='TEST_COMPANY', id='TEST')
self.default_company.save()
Account(id='1001',
path='test.one',
display_name='TestOne',
role='asset').save()
Account(id='1002',
path='test.two',
display_name='TestTwo',
role='expense').save()
Account(id='1003',
path='test.three',
display_name='TestThree',
role='liability').save()
def setUp(self):
# Create a company for the expense
self.default_company = Company(name="TEST_COMPANY", id="TEST")
self.default_company.save()
# Create accounts required to generate expense transactions
self.ap = Account(id="3000", path="AP", display_name="AP")
self.chk = Account(id="1001", path="CHECKING1", display_name="CHECKING1")
self.ap.save()
self.chk.save()
self.cp1 = Counterparty(id="jpmc", name="Test C/P")
self.cp1.save()
self.cp2 = Counterparty(id="test_cp", name="Test C/P")
self.cp2.save()
ExternalAccount(
company=self.default_company, gl_account=self.chk, counterparty=self.cp1, label="CHK", name="Checking"
).save()
class CashflowTestCase(TestCase):
def setUp(self):
# Create a company for the expense
self.default_company = Company(name="TEST_COMPANY", id="TEST")
self.default_company.save()
# Create accounts required to generate expense transactions
self.ap = Account(id="3000", path="AP", display_name="AP")
self.chk = Account(id="1001", path="CHECKING1", display_name="CHECKING1")
self.ap.save()
self.chk.save()
self.cp1 = Counterparty(id="jpmc", name="Test C/P")
self.cp1.save()
self.cp2 = Counterparty(id="test_cp", name="Test C/P")
self.cp2.save()
ExternalAccount(
company=self.default_company, gl_account=self.chk, counterparty=self.cp1, label="CHK", name="Checking"
).save()
def test_get_gl_transactions_checking(self):
trans = Cashflow(
ext_account=ExternalAccount.objects.get(label="CHK"),
post_date=datetime.date(2015, 1, 1),
description="test",
trans_type=Account.objects.get(id="3000"),
amount="100.2",
counterparty=Counterparty.objects.get(id="test_cp"),
)
transactions = trans.get_gl_transactions()
self.assertEqual(len(transactions), 1)
lines = transactions[0]["lines"]
self.assertEqual(len(lines), 2)
self.assertEqual(set([(self.chk, 100.2, self.cp1), (self.ap, -100.2, self.cp2)]), set(lines))
def test_get_gl_transactions_happy_path(self):
nom_trans = NominalTransaction(id='98765',company=self.default_company, date=datetime.date(2015,1,1))
debit = Account(id='7000', path='capital.opexp.retearnings.debit', display_name='7000')
credit = Account(id='7010', path='capital.opexp.retearnings.credit', display_name='7010')
debit.save()
credit.save()
cp = Counterparty(id='test_cp', name="Test C/P")
cp.save()
debit_line = NominalTranLine(transaction_id=nom_trans.id, account=debit, amount='123.4', counterparty=cp)
debit_line.save()
nom_trans.nominaltranline_set.add(debit_line)
credit_line = NominalTranLine(transaction_id=nom_trans.id, account=credit, amount='-123.4', counterparty=cp)
credit_line.save()
nom_trans.nominaltranline_set.add(credit_line)
transactions = nom_trans.get_gl_transactions()
self.assertEqual(len(transactions), 1)
self.assertEqual(len(transactions[0]['lines']), 2)