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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
|
import bottle
from tools import ServerTestBase
from bottle import response
class TestAppMounting(ServerTestBase):
def setUp(self):
ServerTestBase.setUp(self)
self.subapp = bottle.Bottle()
@self.subapp.route('/')
@self.subapp.route('/test/:test')
def test(test='foo'):
return test
def test_mount_order_bug581(self):
self.app.mount('/test/', self.subapp)
# This should not match
self.app.route('/<test:path>', callback=lambda test: test)
self.assertStatus(200, '/test/')
self.assertBody('foo', '/test/')
def test_mount(self):
self.app.mount('/test/', self.subapp)
self.assertStatus(404, '/')
self.assertStatus(404, '/test')
self.assertStatus(200, '/test/')
self.assertBody('foo', '/test/')
self.assertStatus(200, '/test/test/bar')
self.assertBody('bar', '/test/test/bar')
def test_mount_meta(self):
self.app.mount('/test/', self.subapp)
self.assertEqual(
self.app.routes[0].config['mountpoint.prefix'],
'/test/')
self.assertEqual(
self.app.routes[0].config['mountpoint.target'],
self.subapp)
def test_no_slash_prefix(self):
self.app.mount('/test', self.subapp)
self.assertStatus(404, '/')
self.assertStatus(200, '/test')
self.assertBody('foo', '/test')
self.assertStatus(200, '/test/')
self.assertBody('foo', '/test/')
self.assertStatus(200, '/test/test/bar')
self.assertBody('bar', '/test/test/bar')
def test_mount_no_plugins(self):
def plugin(func):
def wrapper(*a, **ka):
return 'Plugin'
return wrapper
self.app.install(plugin)
self.app.route('/foo', callback=lambda: 'baz')
self.app.mount('/test/', self.subapp)
self.assertBody('Plugin', '/foo')
self.assertBody('foo', '/test/')
def test_mount_wsgi(self):
status = {}
def app(environ, start_response):
start_response('200 OK', [('X-Test', 'WSGI')])
return 'WSGI ' + environ['PATH_INFO']
self.app.mount('/test', app)
self.assertStatus(200, '/test/')
self.assertBody('WSGI /', '/test')
self.assertBody('WSGI /', '/test/')
self.assertHeader('X-Test', 'WSGI', '/test/')
self.assertBody('WSGI /test/bar', '/test/test/bar')
def test_mount_wsgi(self):
@self.subapp.route('/cookie')
def test_cookie():
response.set_cookie('a', 'a')
response.set_cookie('b', 'b')
self.app.mount('/test', self.subapp)
c = self.urlopen('/test/cookie')['header']['Set-Cookie']
self.assertEqual(['a=a', 'b=b'], list(sorted(c.split(', '))))
def test_mount_wsgi_ctype_bug(self):
status = {}
def app(environ, start_response):
start_response('200 OK', [('Content-Type', 'test/test')])
return 'WSGI ' + environ['PATH_INFO']
self.app.mount('/test', app)
self.assertHeader('Content-Type', 'test/test', '/test/')
def test_mount_json_bug(self):
@self.subapp.route('/json')
def test_cookie():
return {'a':5}
self.app.mount('/test', self.subapp)
self.assertHeader('Content-Type', 'application/json', '/test/json')
class TestAppMerging(ServerTestBase):
def setUp(self):
ServerTestBase.setUp(self)
self.subapp = bottle.Bottle()
@self.subapp.route('/')
@self.subapp.route('/test/:test')
def test(test='foo'):
return test
def test_merge(self):
self.app.merge(self.subapp)
self.assertStatus(200, '/')
self.assertBody('foo', '/')
self.assertStatus(200, '/test/bar')
self.assertBody('bar', '/test/bar')
if __name__ == '__main__': #pragma: no cover
unittest.main()
|