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))