def test_logs_database_failure_during_select(self):
self.mock_select_user.side_effect = helpers.create_database_error()
with self.assertRaises(DatabaseError):
users.get_by_id('test-user-id')
self.assertEqual([
'ERROR - Database query for user ID "test-user-id" failed',
], self.logger.lines)
def test_assigns_correct_api_key(self):
self.mock_select_user.return_value.fetchone.return_value = create_user_db_record(
)
user = users.get_by_id('test-user-id')
self.assertEqual(API_KEY, user.api_key)
def test_assigns_correct_user_name(self):
self.mock_select_user.return_value.fetchone.return_value = create_user_db_record(
)
user = users.get_by_id('test-user-id')
self.assertEqual('test-user-name', user.name)
def test_returns_a_user(self):
self.mock_select_user.return_value.fetchone.return_value = create_user_db_record(
)
user = users.get_by_id('test-user-id')
self.assertIsInstance(user, users.User)
def test_returns_nothing_if_record_not_found(self):
self.mock_select_user.return_value.fetchone.return_value = None
user = users.get_by_id('test-user-id')
self.assertEqual(None, user)
def test_throws_when_database_query_fails(self):
self.mock_select_user.side_effect = helpers.create_database_error()
with self.assertRaises(DatabaseError):
users.get_by_id('test-user-id')