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