def test_web_page_get_not_in_database_return_none(
self, mock_get_web_page_by_type):
# Arrange
mock_get_web_page_by_type.side_effect = DoesNotExist
# Act
result = web_page_api.get("fake_type")
# Assert
self.assertEqual(None, result)
def test_web_page_get_term_of_use(self, mock_get_web_page_by_type):
# Arrange
content = "content web page terms"
mock_get_web_page_by_type.return_value = _create_mock_web_page(
WEB_PAGE_TYPES["terms_of_use"], content)
# Act
result = web_page_api.get("terms_of_use")
# Assert
self.assertEqual("content web page terms", result.content)
def test_web_page_get_login(self, mock_get_web_page_by_type):
# Arrange
content = "content web page login"
mock_get_web_page_by_type.return_value = _create_mock_web_page(
WEB_PAGE_TYPES["login"], content)
# Act
result = web_page_api.get("login")
# Assert
self.assertEqual("content web page login", result.content)
def test_web_page_get_policy_privacy(self, mock_get_web_page_by_type):
# Arrange
content = "content web page privacy"
mock_get_web_page_by_type.return_value = _create_mock_web_page(
WEB_PAGE_TYPES["privacy_policy"], content)
# Act
result = web_page_api.get("privacy_policy")
# Assert
self.assertEqual("content web page privacy", result.content)
def get_object(self):
""" Get the web page from db
Args:
Returns:
Web page
"""
try:
return web_page_api.get(self.web_page_type)
except exceptions.DoesNotExist:
raise Http404
def get():
""" Get the terms of use if exist
Returns: Terms of use web page
"""
return web_page_api.get(TERMS_PAGE_NAME)
def test_web_page_get_with_wrong_type_return_none(self):
# Arrange
# Act
page = web_page_api.get("wrong_type")
# Assert
self.assertEqual(None, page)
def test_web_page_get_not_in_database_return_none(self):
# Act
result = web_page_api.get("fake_type")
# Assert
self.assertEqual(None, result)
def get():
"""Get the help if exist
Returns: help web page
"""
return web_page_api.get(HELP_PAGE_NAME)
def get():
""" Get the privacy policy if exist
Returns: privacy policy web page
"""
return web_page_api.get(PRIVACY_PAGE_NAME)
def get():
"""Get the login page if exist
Returns: login web page
"""
return web_page_api.get(LOGIN_PAGE_NAME)
def get():
"""Get the help if exist
Returns: rules of behavior web page
"""
return web_page_api.get(RULES_OF_BEHAVIOR_PAGE_NAME)
