def test_suspend_product_when_product_doesnot_exist( self, mock_product_repo_get ): '''Test suspend_product when product doesnot exist. ''' # Arrange with self.app.app_context(): mock_product_repo_get.return_value = None product_controller = ProductController(self.request_context) # Act result = product_controller.suspend_product(1) # Assert assert result.status_code == 400 assert result.get_json()['msg'] == 'Invalid or incorrect ' \ 'product_id provided'
def test_suspend_product_ok_response( self, mock_product_repo_update, mock_product_repo_get ): '''Test suspend_product OK response. ''' # Arrange with self.app.app_context(): mock_product_repo_get.return_value = self.mock_product mock_product_repo_update.return_value = self.mock_product product_controller = ProductController(self.request_context) # Act result = product_controller.suspend_product(1) # Assert assert result.status_code == 200 assert result.get_json()['msg'] == 'OK'