def test_update_vendor_rating_ok_response( self, mock_auth_user, mock_vendor_rating_controller_get_json, mock_vendor_rating_repo_get, mock_vendor_rating_repo_update): '''Test update_vendor_rating when rating is forbidden. ''' # Arrange with self.app.app_context(): mock_auth_user.return_value = 1 mock_vendor_rating_controller_get_json.return_value = { 'comment': 'Mock comment' } mock_vendor_rating = VendorRating(id=1, created_at=datetime.now(), updated_at=datetime.now(), vendor_id=1, user_id=1, comment='Mock comment', service_date=datetime.now(), rating=3.0, channel='web', rating_type='meal', type_id=1, engagement_id=1, main_meal_id=1) mock_vendor_rating_repo_get.return_value = mock_vendor_rating mock_vendor_rating_repo_update.return_value = mock_vendor_rating vendor_rating_controller = VendorRatingController( self.request_context) # Act result = vendor_rating_controller.update_vendor_rating(1) # Assert assert result.status_code == 200 assert result.get_json()['msg'] == 'OK'
def test_update_vendor_rating_when_rating_is_invalid( self, mock_vendor_rating_repo_get, mock_vendor_rating_controller_get_json ): '''Test update_vendor_rating when rating is invalid. ''' # Arrange with self.app.app_context(): mock_rating_id = 1 mock_vendor_rating_controller_get_json.return_value = { 'comment': 'Mock comment' } mock_vendor_rating_repo_get.return_value = None vendor_rating_controller = VendorRatingController( self.request_context ) # Act result = vendor_rating_controller.update_vendor_rating( mock_rating_id ) # Assert assert result.status_code == 404 assert result.get_json()['msg'] == 'Invalid or incorrect ' \ 'rating_id provided'