a077466b1e
`assertEquals` has been deprecated, while not fully removed it will still keep giving warnings in tests / CI. ref: https://docs.python.org/3/library/unittest.html#deprecated-aliases
102 lines
3.1 KiB
Python
102 lines
3.1 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import unittest
|
|
|
|
import frappe
|
|
from frappe.website import render
|
|
from frappe.website.utils import get_home_page
|
|
from frappe.utils import set_request
|
|
|
|
|
|
class TestWebsite(unittest.TestCase):
|
|
|
|
def test_home_page_for_role(self):
|
|
frappe.delete_doc_if_exists('User', 'test-user-for-home-page@example.com')
|
|
frappe.delete_doc_if_exists('Role', 'home-page-test')
|
|
frappe.delete_doc_if_exists('Web Page', 'home-page-test')
|
|
user = frappe.get_doc(dict(
|
|
doctype='User',
|
|
email='test-user-for-home-page@example.com',
|
|
first_name='test')).insert()
|
|
|
|
role = frappe.get_doc(dict(
|
|
doctype = 'Role',
|
|
role_name = 'home-page-test',
|
|
desk_access = 0,
|
|
home_page = '/home-page-test'
|
|
)).insert()
|
|
|
|
user.add_roles(role.name)
|
|
user.save()
|
|
|
|
frappe.set_user('test-user-for-home-page@example.com')
|
|
self.assertEqual(get_home_page(), 'home-page-test')
|
|
|
|
frappe.set_user('Administrator')
|
|
role.home_page = ''
|
|
role.save()
|
|
|
|
# home page via portal settings
|
|
frappe.db.set_value('Portal Settings', None, 'default_portal_home', 'test-portal-home')
|
|
|
|
frappe.set_user('test-user-for-home-page@example.com')
|
|
frappe.cache().hdel('home_page', frappe.session.user)
|
|
self.assertEqual(get_home_page(), 'test-portal-home')
|
|
|
|
def test_page_load(self):
|
|
frappe.set_user('Guest')
|
|
set_request(method='POST', path='login')
|
|
response = render.render()
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
html = frappe.safe_decode(response.get_data())
|
|
|
|
self.assertTrue('// login.js' in html)
|
|
self.assertTrue('<!-- login.html -->' in html)
|
|
frappe.set_user('Administrator')
|
|
|
|
def test_redirect(self):
|
|
import frappe.hooks
|
|
frappe.hooks.website_redirects = [
|
|
dict(source=r'/testfrom', target=r'://testto1'),
|
|
dict(source=r'/testfromregex.*', target=r'://testto2'),
|
|
dict(source=r'/testsub/(.*)', target=r'://testto3/\1')
|
|
]
|
|
|
|
website_settings = frappe.get_doc('Website Settings')
|
|
website_settings.append('route_redirects', {
|
|
'source': '/testsource',
|
|
'target': '/testtarget'
|
|
})
|
|
website_settings.save()
|
|
|
|
frappe.cache().delete_key('app_hooks')
|
|
frappe.cache().delete_key('website_redirects')
|
|
|
|
set_request(method='GET', path='/testfrom')
|
|
response = render.render()
|
|
self.assertEqual(response.status_code, 301)
|
|
self.assertEqual(response.headers.get('Location'), r'://testto1')
|
|
|
|
set_request(method='GET', path='/testfromregex/test')
|
|
response = render.render()
|
|
self.assertEqual(response.status_code, 301)
|
|
self.assertEqual(response.headers.get('Location'), r'://testto2')
|
|
|
|
set_request(method='GET', path='/testsub/me')
|
|
response = render.render()
|
|
self.assertEqual(response.status_code, 301)
|
|
self.assertEqual(response.headers.get('Location'), r'://testto3/me')
|
|
|
|
set_request(method='GET', path='/test404')
|
|
response = render.render()
|
|
self.assertEqual(response.status_code, 404)
|
|
|
|
set_request(method='GET', path='/testsource')
|
|
response = render.render()
|
|
self.assertEqual(response.status_code, 301)
|
|
self.assertEqual(response.headers.get('Location'), '/testtarget')
|
|
|
|
delattr(frappe.hooks, 'website_redirects')
|
|
frappe.cache().delete_key('app_hooks')
|