1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
# -*- coding: utf-8 -*-
import webtest
from webob import Request, Response
from tests.compat import unittest
from webtest.compat import to_bytes
def application(environ, start_response):
req = Request(environ)
if req.path_info == '/redirect':
req.path_info = '/path'
resp = Response()
resp.status = '302 Found'
resp.location = req.path
else:
resp = Response()
resp.body = to_bytes('<html><body><a href="%s">link</a></body></html>' % req.path)
return resp(environ, start_response)
class TestScriptName(unittest.TestCase):
def test_script_name(self):
app = webtest.TestApp(application)
resp = app.get('/script', extra_environ={'SCRIPT_NAME':'/script'})
resp.mustcontain('href="/script"')
resp = app.get('/script/redirect', extra_environ={'SCRIPT_NAME':'/script'})
self.assertEqual(resp.status_int, 302)
self.assertEqual(resp.location, 'http://localhost/script/path', resp.location)
resp = resp.follow(extra_environ={'SCRIPT_NAME':'/script'})
resp.mustcontain('href="/script/path"')
resp = resp.click('link')
resp.mustcontain('href="/script/path"')
def test_app_script_name(self):
app = webtest.TestApp(application, extra_environ={'SCRIPT_NAME':'/script'})
resp = app.get('/script/redirect')
self.assertEqual(resp.status_int, 302)
self.assertEqual(resp.location, 'http://localhost/script/path', resp.location)
resp = resp.follow()
resp.mustcontain('href="/script/path"')
resp = resp.click('link')
resp.mustcontain('href="/script/path"')
def test_script_name_doesnt_match(self):
app = webtest.TestApp(application)
resp = app.get('/path', extra_environ={'SCRIPT_NAME':'/script'})
resp.mustcontain('href="/script/path"')
|
