def test_user_profile_db_conn_failed(self, mock_is_authenticated):
mock_is_authenticated.return_value = True
user_id = '1'
UserDataModel.get_user_data_by_user_id = mock.MagicMock(side_effect=DBConnectionError(""))
ErrorHandler.create_error_response = mock.MagicMock(return_value=no_db_conn_mock)
service = ProfileController()
ResponseBuilder.get_build_response = mock.MagicMock(return_value=profile_successful_mock)
self.assertEqual(service.get(user_id), no_db_conn_mock)
def get_mongodb_instance(user, passwd): try: conn=pymongo.MongoClient(os.environ['MONGO_URI']) conn.testdb.authenticate(user, passwd, mechanism = 'SCRAM-SHA-1') db = conn.testdb return db except pymongo.errors.ConnectionFailure as e: raise DBConnectionError(e)
def test_user_friends_db_conn_failed(self):
user_id = '1'
FriendModel.get_friends_by_user_id = mock.MagicMock(
side_effect=DBConnectionError(''))
ErrorHandler.create_error_response = mock.MagicMock(
return_value=no_db_conn_mock)
service = FriendController()
service._create_get_friends_response = mock.MagicMock(
return_value=friends_successful_mock)
self.assertEqual(service.get_friends_by_user_id(user_id),
no_db_conn_mock)
def test_user_stories_db_conn_failed(self):
user_id = '1'
StorieModel.get_profile_stories_by_user_id = mock.MagicMock(
side_effect=DBConnectionError(""))
ErrorHandler.create_error_response = mock.MagicMock(
return_value=no_db_conn_mock)
service = StorieDetailController()
service._create_get_stories_response = mock.MagicMock(
return_value=stories_successful_mock)
self.assertEqual(service.get_stories_by_user_id(user_id),
no_db_conn_mock)