jinql
/
webssh
mirror of https://github.com/huashengdun/webssh
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Enhanced tests

Browse Source
pull/58/head
Sheng 6 years ago
parent a1c9378048
commit d389e32b85
1 changed files with 14 additions and 25 deletions
Show Stats Download Patch File Download Diff File

39
tests/test_app.py
Unescape View File

@ -561,7 +561,7 @@ class TestAppWithRejectPolicy(OtherTestBase):
def test_app_with_hostname_not_in_hostkeys(self): def test_app_with_hostname_not_in_hostkeys(self):
url = self.get_url('/') url = self.get_url('/')
client = self.get_http_client() client = self.get_http_client()
body = urlencode(dict(self.body, username='foo')) body = urlencode(self.body)
response = yield client.fetch(url, method='POST', body=body, response = yield client.fetch(url, method='POST', body=body,
headers=self.headers) headers=self.headers)
data = json.loads(to_str(response.body)) data = json.loads(to_str(response.body))
@ -584,7 +584,7 @@ class TestAppWithBadHostKey(OtherTestBase):
def test_app_with_bad_host_key(self): def test_app_with_bad_host_key(self):
url = self.get_url('/') url = self.get_url('/')
client = self.get_http_client() client = self.get_http_client()
body = urlencode(dict(self.body, username='foo')) body = urlencode(self.body)
response = yield client.fetch(url, method='POST', body=body, response = yield client.fetch(url, method='POST', body=body,
headers=self.headers) headers=self.headers)
data = json.loads(to_str(response.body)) data = json.loads(to_str(response.body))
@ -668,7 +668,7 @@ class TestAppWithPutRequest(OtherTestBase):
def test_app_with_method_not_supported(self): def test_app_with_method_not_supported(self):
url = self.get_url('/') url = self.get_url('/')
client = self.get_http_client() client = self.get_http_client()
body = urlencode(dict(self.body, username='foo')) body = urlencode(self.body)
with self.assertRaises(HTTPError) as ctx: with self.assertRaises(HTTPError) as ctx:
yield client.fetch( yield client.fetch(
@ -687,17 +687,11 @@ class TestAppWithTooManyConnections(OtherTestBase):
@tornado.testing.gen_test @tornado.testing.gen_test
def test_app_with_too_many_connections(self): def test_app_with_too_many_connections(self):
clients['127.0.0.1'] = {'fake_worker_id': None}
url = self.get_url('/') url = self.get_url('/')
client = self.get_http_client() client = self.get_http_client()
body = urlencode(dict(self.body, username='foo')) body = urlencode(self.body)
response = yield client.fetch(url, method='POST', body=body,
headers=self.headers)
data = json.loads(to_str(response.body))
worker_id = data['id']
self.assertIsNotNone(worker_id)
self.assertIsNotNone(data['encoding'])
self.assertIsNone(data['status'])
response = yield client.fetch(url, method='POST', body=body, response = yield client.fetch(url, method='POST', body=body,
headers=self.headers) headers=self.headers)
data = json.loads(to_str(response.body)) data = json.loads(to_str(response.body))
@ -705,19 +699,14 @@ class TestAppWithTooManyConnections(OtherTestBase):
self.assertIsNone(data['encoding']) self.assertIsNone(data['encoding'])
self.assertEqual(data['status'], 'Too many connections.') self.assertEqual(data['status'], 'Too many connections.')
ws_url = url.replace('http', 'ws') + 'ws?id=' + worker_id clients['127.0.0.1'].clear()
ws = yield tornado.websocket.websocket_connect(ws_url)
msg = yield ws.read_message()
self.assertIsNotNone(msg)
response = yield client.fetch(url, method='POST', body=body, response = yield client.fetch(url, method='POST', body=body,
headers=self.headers) headers=self.headers)
data = json.loads(to_str(response.body)) data = json.loads(to_str(response.body))
self.assertIsNone(data['id']) self.assertIsNotNone(data['id'])
self.assertIsNone(data['encoding']) self.assertIsNotNone(data['encoding'])
self.assertEqual(data['status'], 'Too many connections.') self.assertIsNone(data['status'])
ws.close()
class TestAppWithCrossOriginOperation(OtherTestBase): class TestAppWithCrossOriginOperation(OtherTestBase):
@ -728,7 +717,7 @@ class TestAppWithCrossOriginOperation(OtherTestBase):
def test_app_with_wrong_event_origin(self): def test_app_with_wrong_event_origin(self):
url = self.get_url('/') url = self.get_url('/')
client = self.get_http_client() client = self.get_http_client()
body = urlencode(dict(self.body, username='foo', _origin='localhost')) body = urlencode(dict(self.body, _origin='localhost'))
response = yield client.fetch(url, method='POST', body=body, response = yield client.fetch(url, method='POST', body=body,
headers=self.headers) headers=self.headers)
data = json.loads(to_str(response.body)) data = json.loads(to_str(response.body))
@ -742,7 +731,7 @@ class TestAppWithCrossOriginOperation(OtherTestBase):
def test_app_with_wrong_header_origin(self): def test_app_with_wrong_header_origin(self):
url = self.get_url('/') url = self.get_url('/')
client = self.get_http_client() client = self.get_http_client()
body = urlencode(dict(self.body, username='foo')) body = urlencode(self.body)
headers = dict(self.headers, Origin='localhost') headers = dict(self.headers, Origin='localhost')
response = yield client.fetch(url, method='POST', body=body, response = yield client.fetch(url, method='POST', body=body,
headers=headers) headers=headers)
@ -757,7 +746,7 @@ class TestAppWithCrossOriginOperation(OtherTestBase):
def test_app_with_correct_event_origin(self): def test_app_with_correct_event_origin(self):
url = self.get_url('/') url = self.get_url('/')
client = self.get_http_client() client = self.get_http_client()
body = urlencode(dict(self.body, username='foo', _origin=self.origin)) body = urlencode(dict(self.body, _origin=self.origin))
response = yield client.fetch(url, method='POST', body=body, response = yield client.fetch(url, method='POST', body=body,
headers=self.headers) headers=self.headers)
data = json.loads(to_str(response.body)) data = json.loads(to_str(response.body))
@ -770,7 +759,7 @@ class TestAppWithCrossOriginOperation(OtherTestBase):
def test_app_with_correct_header_origin(self): def test_app_with_correct_header_origin(self):
url = self.get_url('/') url = self.get_url('/')
client = self.get_http_client() client = self.get_http_client()
body = urlencode(dict(self.body, username='foo')) body = urlencode(self.body)
headers = dict(self.headers, Origin=self.origin) headers = dict(self.headers, Origin=self.origin)
response = yield client.fetch(url, method='POST', body=body, response = yield client.fetch(url, method='POST', body=body,
headers=headers) headers=headers)

Write Preview
Loading…
Cancel
Save