def test_get_gl_transactions_happy_path(self):
expense = Expense(company=self.default_company,
amount='123.45',
account_id='7890',
counterparty_id='testcp1')
transactions = expense.get_gl_transactions()
self.assertEqual(len(transactions), 1)
lines = set((l[0].id, l[1], l[2].id) for l in transactions[0]['lines'])
self.assertEqual(len(lines), 2)
should_be = [('7890', Decimal('123.45'), 'testcp1'),
('3000', Decimal('-123.45'), 'testcp1')]
self.assertEqual(set(should_be), set(lines))
def test_get_gl_transactions_multiperiod_small(self):
expense = Expense(company=self.default_company,
amount='123.45',
account_id='7890',
start_date=datetime.date(2015,1,1),
end_date=datetime.date(2015,10,1),
counterparty_id='testcp1')
transactions = expense.get_gl_transactions()
self.assertEqual(len(transactions), 1)
lines = set((l[0].id, l[1], l[2].id) for l in transactions[0]['lines'])
self.assertEqual(len(lines), 2)
should_be = [('7890', Decimal('123.45'), 'testcp1'),
('3000', Decimal('-123.45'), 'testcp1')]
self.assertEqual(set(should_be), set(lines))
