def test_get_vendor_rating_ok_response(self, mock_vendor_rating_repo_get): '''Test get_vendor_rating OK response. ''' # Arrange with self.app.app_context(): mock_vendor_rating = VendorRating(vendor_id=1, user_id=1, comment='Mock comment', service_date=datetime.now(), rating=1.0, channel='Mock channel', rating_type='engagement', type_id=0, engagement_id=1, id=1, created_at=datetime.now(), updated_at=datetime.now(), main_meal_id=1) mock_vendor_rating_repo_get.return_value = mock_vendor_rating vendor_rating_controller = VendorRatingController( self.request_context) # Act result = vendor_rating_controller.get_vendor_rating(1) # Assert assert result.status_code == 200 assert result.get_json()['msg'] == 'OK'
def test_get_vendor_rating_when_rating_doesnot_exist( self, mock_vendor_rating_repo_get): '''Test get_vendor_rating when the rating does not exist. ''' # Arrange with self.app.app_context(): mock_vendor_rating_repo_get.return_value = None vendor_rating_controller = VendorRatingController( self.request_context) # Act result = vendor_rating_controller.get_vendor_rating(1) # Assert assert result.status_code == 400 assert result.get_json()['msg'] == 'Bad Request'