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())
Example #2
0
 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))
Example #4
0
 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))
Example #5
0
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
Example #6
0
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