def test_get_all_no_connection(self): with closing(TestDatabase()) as test_db: test_db.use_table('users') expected_users = ["user_A", "user_B", "user_C"] with unittest.mock.patch('microservice.components.get_psql') as db_mock: db_mock.return_value = test_db.connection self.assertListEqual(expected_users, users.get_all()) with unittest.mock.patch('microservice.components.get_psql') as db_mock: db_mock.return_value.cursor.side_effect = DatabaseError self.assertListEqual(expected_users, users.get_all())
def test_get_all_no_connection(self): with closing(TestDatabase()) as test_db: test_db.use_table('users') expected_users = ["user_A", "user_B", "user_C"] with unittest.mock.patch( 'microservice.components.get_psql') as db_mock: db_mock.return_value = test_db.connection self.assertListEqual(expected_users, users.get_all()) with unittest.mock.patch( 'microservice.components.get_psql') as db_mock: db_mock.return_value.cursor.side_effect = DatabaseError self.assertListEqual(expected_users, users.get_all())
def test_get_all(self): with closing(TestDatabase()) as test_db: redis = components.get_redis() test_db.use_table('users') expected_users = ["user_A", "user_B", "user_C"] with unittest.mock.patch('microservice.components.get_psql') as db_mock: db_mock.return_value = test_db.connection self.assertListEqual(expected_users, users.get_all()) self.assertEqual(json.dumps(expected_users), redis.get(users.REDIS_KEY))
def test_get_all(self): with closing(TestDatabase()) as test_db: redis = components.get_redis() test_db.use_table('users') expected_users = ["user_A", "user_B", "user_C"] with unittest.mock.patch( 'microservice.components.get_psql') as db_mock: db_mock.return_value = test_db.connection self.assertListEqual(expected_users, users.get_all()) self.assertEqual(json.dumps(expected_users), redis.get(users.REDIS_KEY))
def get_users(**kwargs): raven = components.get_raven() with components.get_riemann() as riemann: try: result = users.get_all() except Exception as e: raven.captureException(e) raise e riemann.event(service="microservice_users") return result