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 19 default = cherrypy.Application(None) 20 21 domains = {} 22 for year in range(1997, 2008): 23 app = cherrypy.Application(ClassOfRoot('Class of %s' % year)) 24 domains['www.classof%s.example' % year] = app 25 26 cherrypy.tree.graft(cherrypy._cpwsgi.VirtualHost(default, domains)) 27 setup_server = staticmethod(setup_server) 28
29 - def test_welcome(self):
30 if not cherrypy.server.using_wsgi: 31 return self.skip("skipped (not using WSGI)... ") 32 33 for year in range(1997, 2008): 34 self.getPage("/", headers=[('Host', 'www.classof%s.example' % year)]) 35 self.assertBody("Welcome to the Class of %s website!" % year)
36