def test_update_expense(self):
fs.insert_user(self.cursor, 'user1')
fs.insert_user(self.cursor, 'user2')
fs.insert_expense(self.cursor, 'test1', 20, '20130101', 1, (1, 2))
self.connection.commit()
fs.update_expense(self.cursor, 1, 'edit1', 30, '20130102', 2, (1,))
expected_expense = {'id': 1, 'title': 'edit1', 'cost': 30,
'date': '20130102', 'payer': 2}
expected_debtors = [1,]
self.assertEqual(fs.get_expense(self.cursor, 1), expected_expense)
self.assertEqual(fs.get_debtors(self.cursor, 1), expected_debtors)
def test_get_expense(self):
fs.insert_user(self.cursor, 'user1')
fs.insert_user(self.cursor, 'user2')
self.connection.commit()
fs.insert_expense(self.cursor, 'test1', 20, '20130101', 1, (1, 2))
fs.insert_expense(self.cursor, 'test2', 30, '20130102', 1, (1, 2))
fs.insert_expense(self.cursor, 'test3', 40, '20130103', 2, (1, 2))
self.connection.commit()
expected_exp = (
{'id': 1, 'date': '20130101', 'title': 'test1', 'payer': 1,
'cost': 20},
{'id': 2, 'date': '20130102', 'title': 'test2', 'payer': 1,
'cost': 30},
{'id': 3, 'date': '20130103', 'title': 'test3', 'payer': 2,
'cost': 40},
)
for i in range(len(expected_exp)):
self.assertEqual(fs.get_expense(self.cursor, i+1), expected_exp[i])
