def test_un_suspend_vendor_ok_response(self, mock_vendor_repo_update, mock_vendor_repo_get): '''Test un_suspend_vendor OK response. ''' # Arrange with self.app.app_context(): mock_vendor_repo_get.return_value = self.mock_vendor mock_vendor_repo_update.return_value = self.mock_vendor vendor_controller = VendorController(self.request_context) # Act result = vendor_controller.un_suspend_vendor(1) # Assert assert result.status_code == 200 assert result.get_json()['msg'] == 'OK'
def test_un_suspend_vendor_when_vendor_doesnot_exist( self, mock_vendor_repo_get): '''Test un_suspend_vendor when vendor doesnot exist. ''' # Arrange with self.app.app_context(): mock_vendor_repo_get.return_value = None vendor_controller = VendorController(self.request_context) # Act result = vendor_controller.un_suspend_vendor(1) # Assert assert result.status_code == 400 assert result.get_json()['msg'] == 'Invalid or incorrect ' \ 'vendor_id provided'