825b0120fa
* fix(login): redirect user from login page if already logged in the user should not be able to access the login page if a user session already exists. closes #6500. Signed-off-by: Chinmay Pai <chinmaydpai@gmail.com> * fix(test-website): fix website test what is the point in writing tests if they don't really work/function as intended? Signed-off-by: Chinmay Pai <chinmaydpai@gmail.com> * fix(regex): do not replace '\' in rules that defeats the entire purpose of creating rules, wtf? Signed-off-by: Chinmay Pai <chinmaydpai@gmail.com> * fix(test_website): change user using set_user() Signed-off-by: Chinmay Pai <chinmaydpai@gmail.com> * redirect: prefix string with r to escape string literals Signed-off-by: Chinmay Pai <chinmaydpai@gmail.com>
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import frappe, unittest
|
|
from werkzeug.wrappers import Request
|
|
from werkzeug.test import EnvironBuilder
|
|
|
|
from frappe.website import render
|
|
|
|
def set_request(**kwargs):
|
|
builder = EnvironBuilder(**kwargs)
|
|
frappe.local.request = Request(builder.get_environ())
|
|
|
|
class TestWebsite(unittest.TestCase):
|
|
|
|
def test_page_load(self):
|
|
frappe.set_user('Guest')
|
|
set_request(method='POST', path='login')
|
|
response = render.render()
|
|
|
|
self.assertEquals(response.status_code, 200)
|
|
|
|
html = frappe.safe_decode(response.get_data())
|
|
|
|
self.assertTrue('/* login-css */' in html)
|
|
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')
|
|
]
|
|
frappe.cache().delete_key('app_hooks')
|
|
frappe.cache().delete_key('website_redirects')
|
|
|
|
set_request(method='GET', path='/testfrom')
|
|
response = render.render()
|
|
self.assertEquals(response.status_code, 301)
|
|
self.assertEquals(response.headers.get('Location'), r'://testto1')
|
|
|
|
set_request(method='GET', path='/testfromregex/test')
|
|
response = render.render()
|
|
self.assertEquals(response.status_code, 301)
|
|
self.assertEquals(response.headers.get('Location'), r'://testto2')
|
|
|
|
set_request(method='GET', path='/testsub/me')
|
|
response = render.render()
|
|
self.assertEquals(response.status_code, 301)
|
|
self.assertEquals(response.headers.get('Location'), r'://testto3/me')
|
|
|
|
set_request(method='GET', path='/test404')
|
|
response = render.render()
|
|
self.assertEquals(response.status_code, 404)
|
|
|
|
delattr(frappe.hooks, 'website_redirects')
|
|
frappe.cache().delete_key('app_hooks')
|
|
|