def test_get_product_ok_response( self, mock_product_repo_get ): '''Test get_product OK response. ''' # Arrange with self.app.app_context(): mock_product_repo_get.return_value = self.mock_product product_controller = ProductController(self.request_context) # Act result = product_controller.get_product(1) # Assert assert result.status_code == 200 assert result.get_json()['msg'] == 'OK'
def test_get_product_when_product_doesnot_exist( self, mock_product_repo_get ): '''Test get_product when the 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.get_product(1) # Assert assert result.status_code == 400 assert result.get_json()['msg'] == 'Bad Request - Invalid or ' \ 'Missing product_id'