from cherrypy.test import test test.prefer_parent_path() import cherrypy script_names = ["", "/foo", "/users/fred/blog", "/corp/blog"] def setup_server(): class Root: def index(self, name="world"): return name index.exposed = True def foobar(self): return "bar" foobar.exposed = True def default(self, *params): return "default:" + repr(params) default.exposed = True def other(self): return "other" other.exposed = True def extra(self, *p): return repr(p) extra.exposed = True def redirect(self): raise cherrypy.HTTPRedirect('dir1/', 302) redirect.exposed = True def notExposed(self): return "not exposed" def confvalue(self): return cherrypy.request.config.get("user") confvalue.exposed = True def mapped_func(self, ID=None): return "ID is %s" % ID mapped_func.exposed = True setattr(Root, "Von B\xfclow", mapped_func) class Exposing: def base(self): return "expose works!" cherrypy.expose(base) cherrypy.expose(base, "1") cherrypy.expose(base, "2") class ExposingNewStyle(object): def base(self): return "expose works!" cherrypy.expose(base) cherrypy.expose(base, "1") cherrypy.expose(base, "2") class Dir1: def index(self): return "index for dir1" index.exposed = True def myMethod(self): return "myMethod from dir1, path_info is:" + repr(cherrypy.request.path_info) myMethod.exposed = True myMethod._cp_config = {'tools.trailing_slash.extra': True} def default(self, *params): return "default for dir1, param is:" + repr(params) default.exposed = True class Dir2: def index(self): return "index for dir2, path is:" + cherrypy.request.path_info index.exposed = True def script_name(self): return cherrypy.tree.script_name() script_name.exposed = True def cherrypy_url(self): return cherrypy.url("/extra") cherrypy_url.exposed = True def posparam(self, *vpath): return "/".join(vpath) posparam.exposed = True class Dir3: def default(self): return "default for dir3, not exposed" class Dir4: def index(self): return "index for dir4, not exposed" class DefNoIndex: def default(self, *args): raise cherrypy.HTTPRedirect("contact") default.exposed = True # MethodDispatcher code class ByMethod: exposed = True def __init__(self, *things): self.things = list(things) def GET(self): return repr(self.things) def POST(self, thing): self.things.append(thing) class Collection: default = ByMethod('a', 'bit') Root.exposing = Exposing() Root.exposingnew = ExposingNewStyle() Root.dir1 = Dir1() Root.dir1.dir2 = Dir2() Root.dir1.dir2.dir3 = Dir3() Root.dir1.dir2.dir3.dir4 = Dir4() Root.defnoindex = DefNoIndex() Root.bymethod = ByMethod('another') Root.collection = Collection() for url in script_names: d = cherrypy.dispatch.MethodDispatcher() conf = {'/': {'user': (url or "/").split("/")[-2]}, '/bymethod': {'request.dispatch': d}, '/collection': {'request.dispatch': d}, } cherrypy.tree.mount(Root(), url, conf) cherrypy.config.update({'environment': "test_suite"}) class Isolated: def index(self): return "made it!" index.exposed = True cherrypy.tree.mount(Isolated(), "/isolated") from cherrypy.test import helper class ObjectMappingTest(helper.CPWebCase): def testObjectMapping(self): for url in script_names: prefix = self.script_name = url self.getPage('/') self.assertBody('world') self.getPage("/dir1/myMethod") self.assertBody("myMethod from dir1, path_info is:'/dir1/myMethod'") self.getPage("/this/method/does/not/exist") self.assertBody("default:('this', 'method', 'does', 'not', 'exist')") self.getPage("/extra/too/much") self.assertBody("('too', 'much')") self.getPage("/other") self.assertBody('other') self.getPage("/notExposed") self.assertBody("default:('notExposed',)") self.getPage("/dir1/dir2/") self.assertBody('index for dir2, path is:/dir1/dir2/') # Test omitted trailing slash (should be redirected by default). self.getPage("/dir1/dir2") self.assertStatus((302, 303)) self.assertHeader('Location', '%s/dir1/dir2/' % self.base()) # Test extra trailing slash (should be redirected if configured). self.getPage("/dir1/myMethod/") self.assertStatus((302, 303)) self.assertHeader('Location', '%s/dir1/myMethod' % self.base()) # Test that default method must be exposed in order to match. self.getPage("/dir1/dir2/dir3/dir4/index") self.assertBody("default for dir1, param is:('dir2', 'dir3', 'dir4', 'index')") # Test *vpath when default() is defined but not index() # This also tests HTTPRedirect with default. self.getPage("/defnoindex") self.assertStatus((302, 303)) self.assertHeader('Location', '%s/contact' % self.base()) self.getPage("/defnoindex/") self.assertStatus((302, 303)) self.assertHeader('Location', '%s/defnoindex/contact' % self.base()) self.getPage("/defnoindex/page") self.assertStatus((302, 303)) self.assertHeader('Location', '%s/defnoindex/contact' % self.base()) self.getPage("/redirect") self.assertStatus('302 Found') self.assertHeader('Location', '%s/dir1/' % self.base()) # Test that we can use URL's which aren't all valid Python identifiers # This should also test the %XX-unquoting of URL's. self.getPage("/Von%20B%fclow?ID=14") self.assertBody("ID is 14") # Test that %2F in the path doesn't get unquoted too early; # that is, it should not be used to separate path components. # See ticket #393. self.getPage("/page%2Fname") self.assertBody("default:('page/name',)") self.getPage("/dir1/dir2/script_name") self.assertBody(url) self.getPage("/dir1/dir2/cherrypy_url") self.assertBody("%s/extra" % self.base()) # Test that configs don't overwrite each other from diferent apps self.getPage("/confvalue") self.assertBody((url or "/").split("/")[-2]) self.script_name = "" # Test absoluteURI's in the Request-Line self.getPage('http://localhost/') self.assertBody('world') # Test that the "isolated" app doesn't leak url's into the root app. # If it did leak, Root.default() would answer with # "default:('isolated', 'doesnt', 'exist')". self.getPage("/isolated/") self.assertStatus("200 OK") self.assertBody("made it!") self.getPage("/isolated/doesnt/exist") self.assertStatus("404 Not Found") # Make sure /foobar maps to Root.foobar and not to the app # mounted at /foo. See http://www.cherrypy.org/ticket/573 self.getPage("/foobar") self.assertBody("bar") def testPositionalParams(self): self.getPage("/dir1/dir2/posparam/18/24/hut/hike") self.assertBody("18/24/hut/hike") # intermediate index methods should not receive posparams; # only the "final" index method should do so. self.getPage("/dir1/dir2/5/3/sir") self.assertBody("default for dir1, param is:('dir2', '5', '3', 'sir')") def testExpose(self): # Test the cherrypy.expose function/decorator self.getPage("/exposing/base") self.assertBody("expose works!") self.getPage("/exposing/1") self.assertBody("expose works!") self.getPage("/exposing/2") self.assertBody("expose works!") self.getPage("/exposingnew/base") self.assertBody("expose works!") self.getPage("/exposingnew/1") self.assertBody("expose works!") self.getPage("/exposingnew/2") self.assertBody("expose works!") def testMethodDispatch(self): self.getPage("/bymethod") self.assertBody("['another']") self.assertHeader('Allow', 'GET, HEAD, POST') self.getPage("/bymethod", method="HEAD") self.assertBody("") self.assertHeader('Allow', 'GET, HEAD, POST') self.getPage("/bymethod", method="POST", body="thing=one") self.assertBody("") self.assertHeader('Allow', 'GET, HEAD, POST') self.getPage("/bymethod") self.assertBody("['another', 'one']") self.assertHeader('Allow', 'GET, HEAD, POST') self.getPage("/bymethod", method="PUT") self.assertErrorPage(405) self.assertHeader('Allow', 'GET, HEAD, POST') # Test default with posparams self.getPage("/collection/silly", method="POST") self.getPage("/collection", method="GET") self.assertBody("['a', 'bit', 'silly']") if __name__ == "__main__": setup_server() helper.testmain()