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'