Package cherrypy :: Package test :: Module test_wsgi_vhost
[hide private]
[frames] | no frames]

Source Code for Module cherrypy.test.test_wsgi_vhost

 1  import cherrypy 
 2  from cherrypy.test import helper 
 3   
 4   
5 -class WSGI_VirtualHost_Test(helper.CPWebCase):
6
7 - def setup_server():
8 9 class ClassOfRoot(object): 10 11 def __init__(self, name): 12 self.name = name
13 14 def index(self): 15 return "Welcome to the %s website!" % self.name
16 index.exposed = True 17 18 default = cherrypy.Application(None) 19 20 domains = {} 21 for year in range(1997, 2008): 22 app = cherrypy.Application(ClassOfRoot('Class of %s' % year)) 23 domains['www.classof%s.example' % year] = app 24 25 cherrypy.tree.graft(cherrypy._cpwsgi.VirtualHost(default, domains)) 26 setup_server = staticmethod(setup_server) 27
28 - def test_welcome(self):
29 if not cherrypy.server.using_wsgi: 30 return self.skip("skipped (not using WSGI)... ") 31 32 for year in range(1997, 2008): 33 self.getPage( 34 "/", headers=[('Host', 'www.classof%s.example' % year)]) 35 self.assertBody("Welcome to the Class of %s website!" % year)
36