def test_get_users_dict(self):
self.assertEqual(fs.get_users_dict(self.cursor), {})
fs.insert_user(self.cursor, 'user1')
fs.insert_user(self.cursor, 'user2')
self.connection.commit()
users_expected = {1: 'user1', 2: 'user2'}
self.assertEqual(fs.get_users_dict(self.cursor), users_expected)
def test_get_status_dict(self):
users = fs.get_users_dict(self.cursor)
self.assertEqual(fs.get_status_dict(self.cursor, users), {})
fs.insert_user(self.cursor, 'user1')
fs.insert_user(self.cursor, 'user2')
fs.insert_user(self.cursor, 'user3')
fs.insert_expense(self.cursor, 'test', 20, '20130101', 1, (1, 2))
fs.insert_expense(self.cursor, 'test', 30, '20130101', 1, (1, 2))
fs.insert_expense(self.cursor, 'test', 40, '20130101', 2, (1, 2))
fs.insert_expense(self.cursor, 'test', 40, '20130101', 2, (1, 2))
fs.insert_expense(self.cursor, 'test', 40, '20130101', 2, (1, 2))
fs.insert_expense(self.cursor, 'test', 5, '20130101', 1, (1, 2))
fs.insert_expense(self.cursor, 'test', 105, '20130101', 3, (1, 2))
fs.insert_expense(self.cursor, 'test', 210, '20130101', 3, (1, 2, 3))
self.connection.commit()
expected_status = {
'user1': {'user1': 0, 'user2': 0, 'user3': 0},
'user2': {'user1': 32.5, 'user2': 0, 'user3': 0},
'user3': {'user1': 122.5, 'user2': 122.5, 'user3': 0},
}
users = fs.get_users_dict(self.cursor)
self.assertEqual(fs.get_status_dict(self.cursor, users),
expected_status)
