Mirror
/
zulip
mirror of https://github.com/zulip/zulip.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

tornado: Add support for testing with subdomains.

This commit is contained in:
Tim Abbott 2017-08-25 15:40:17 -07:00
parent acdeaca2a8
commit 7a5eb9dd9f
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
zerver/lib/test_classes.py
View File

@ -116,7 +116,7 @@ class ZulipTestCase(TestCase):
kwargs["HTTP_HOST"] = "%s.%s" % (kwargs["subdomain"], settings.EXTERNAL_HOST)
else:
kwargs["HTTP_HOST"] = settings.EXTERNAL_HOST
del kwargs["subdomain"]
del kwargs['subdomain']
elif 'HTTP_HOST' not in kwargs:
kwargs["HTTP_HOST"] = "zulip.%s" % (settings.EXTERNAL_HOST,)

11
zerver/tests/test_tornado.py
View File

@ -60,11 +60,19 @@ class TornadoWebTestCase(AsyncHTTPTestCase, ZulipTestCase):
def client_get(self, path, **kwargs):
# type: (Text, **Any) -> HTTPResponse
self.add_session_cookie(kwargs)
self.set_http_host(kwargs)
if 'HTTP_HOST' in kwargs:
kwargs['headers']['Host'] = kwargs['HTTP_HOST']
del kwargs['HTTP_HOST']
return self.fetch(path, method='GET', **kwargs)
def fetch_async(self, method, path, **kwargs):
# type: (Text, Text, **Any) -> None
self.add_session_cookie(kwargs)
self.set_http_host(kwargs)
if 'HTTP_HOST' in kwargs:
kwargs['headers']['Host'] = kwargs['HTTP_HOST']
del kwargs['HTTP_HOST']
self.http_client.fetch(
self.get_url(path),
self.stop,
@ -74,6 +82,7 @@ class TornadoWebTestCase(AsyncHTTPTestCase, ZulipTestCase):
def client_get_async(self, path, **kwargs):
# type: (Text, **Any) -> None
self.set_http_host(kwargs)
self.fetch_async('GET', path, **kwargs)
def login(self, *args, **kwargs):
@ -98,7 +107,7 @@ class TornadoWebTestCase(AsyncHTTPTestCase, ZulipTestCase):
def create_queue(self, **kwargs):
# type: (**Any) -> str
response = self.client_get('/json/events?dont_block=true')
response = self.client_get('/json/events?dont_block=true', subdomain="zulip")
self.assertEqual(response.code, 200)
body = ujson.loads(response.body)
self.assertEqual(body['events'], [])