commit 8922813142cd4578b2aec70193c6f5905b185dce
parent d389e32b85a59aa72b173907bf2724f2a40b0678
Author: Sheng <webmaster0115@gmail.com>
Date: Wed, 13 Feb 2019 11:40:44 +0800
Refactored tests
Diffstat:
M | tests/test_app.py | | | 228 | ++++++++++++++++++++++++++++--------------------------------------------------- |
1 file changed, 81 insertions(+), 147 deletions(-)
diff --git a/tests/test_app.py b/tests/test_app.py
@@ -27,7 +27,54 @@ handler.DELAY = 0.1
swallow_http_errors = handler.swallow_http_errors
-class TestAppBasic(AsyncHTTPTestCase):
+class TestAppBase(AsyncHTTPTestCase):
+
+ def get_httpserver_options(self):
+ return get_server_settings(options)
+
+ def assert_response(self, bstr, response):
+ if swallow_http_errors:
+ self.assertEqual(response.code, 200)
+ self.assertIn(bstr, response.body)
+ else:
+ self.assertEqual(response.code, 400)
+ self.assertIn(b'Bad Request', response.body)
+
+ def assert_status_in(self, data, status):
+ self.assertIsNone(data['encoding'])
+ self.assertIsNone(data['id'])
+ self.assertIn(status, data['status'])
+
+ def assert_status_equal(self, data, status=''):
+ if not status:
+ self.assertIsNotNone(data['encoding'])
+ self.assertIsNotNone(data['id'])
+ else:
+ self.assertIsNone(data['encoding'])
+ self.assertIsNone(data['id'])
+ self.assertEqual(status, data['status'])
+
+ def sync_post(self, body, headers={}, url='/', method='POST'):
+ headers.update(self.headers)
+ return self.fetch(url, method=method, body=body, headers=headers)
+
+ def async_post(self, url, body, headers={}, method='POST'):
+ if url.startswith('/'):
+ url = self.get_url('/')
+
+ if isinstance(body, dict):
+ body = urlencode(body)
+
+ if headers:
+ headers.update(self.headers)
+ else:
+ headers = self.headers
+
+ client = self.get_http_client()
+ return client.fetch(url, method=method, body=body, headers=headers)
+
+
+class TestAppBasic(TestAppBase):
running = [True]
sshserver_port = 2200
@@ -65,26 +112,6 @@ class TestAppBasic(AsyncHTTPTestCase):
cls.running.pop()
print('='*20)
- def get_httpserver_options(self):
- return get_server_settings(options)
-
- def assert_response(self, bstr, response):
- if swallow_http_errors:
- self.assertEqual(response.code, 200)
- self.assertIn(bstr, response.body)
- else:
- self.assertEqual(response.code, 400)
- self.assertIn(b'Bad Request', response.body)
-
- def sync_post(self, body, headers={}, url='/', method='POST'):
- headers.update(self.headers)
- return self.fetch(url, method=method, body=body, headers=headers)
-
- def async_post(self, url, body, headers={}, method='POST'):
- client = self.get_http_client()
- headers.update(self.headers)
- return client.fetch(url, method=method, body=body, headers=headers)
-
def test_app_with_invalid_form_for_missing_argument(self):
response = self.fetch('/')
self.assertEqual(response.code, 200)
@@ -142,17 +169,11 @@ class TestAppBasic(AsyncHTTPTestCase):
def test_app_with_wrong_credentials(self):
response = self.sync_post(self.body + 's')
- data = json.loads(to_str(response.body))
- self.assertIsNone(data['encoding'])
- self.assertIsNone(data['id'])
- self.assertIn('Authentication failed.', data['status'])
+ self.assert_status_in(json.loads(to_str(response.body)), 'Authentication failed.') # noqa
def test_app_with_correct_credentials(self):
response = self.sync_post(self.body)
- data = json.loads(to_str(response.body))
- self.assertIsNone(data['status'])
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
+ self.assert_status_equal(json.loads(to_str(response.body)), None)
def test_app_with_correct_credentials_but_with_no_port(self):
default_port = handler.DEFAULT_PORT
@@ -161,18 +182,12 @@ class TestAppBasic(AsyncHTTPTestCase):
# with no port value
body = self.body.replace(str(self.sshserver_port), '')
response = self.sync_post(body)
- data = json.loads(to_str(response.body))
- self.assertIsNone(data['status'])
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
+ self.assert_status_equal(json.loads(to_str(response.body)), None)
# with no port argument
body = body.replace('port=&', '')
response = self.sync_post(body)
- data = json.loads(to_str(response.body))
- self.assertIsNone(data['status'])
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
+ self.assert_status_equal(json.loads(to_str(response.body)), None)
handler.DEFAULT_PORT = default_port
@@ -181,9 +196,7 @@ class TestAppBasic(AsyncHTTPTestCase):
url = self.get_url('/')
response = yield self.async_post(url, self.body)
data = json.loads(to_str(response.body))
- self.assertIsNone(data['status'])
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
+ self.assert_status_equal(data, None)
url = url.replace('http', 'ws')
ws_url = url + 'ws?id=' + data['id']
@@ -198,9 +211,7 @@ class TestAppBasic(AsyncHTTPTestCase):
url = self.get_url('/')
response = yield self.async_post(url, self.body)
data = json.loads(to_str(response.body))
- self.assertIsNone(data['status'])
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
+ self.assert_status_equal(data, None)
url = url.replace('http', 'ws')
ws_url = url + 'ws?id=' + data['id']
@@ -214,9 +225,7 @@ class TestAppBasic(AsyncHTTPTestCase):
url = self.get_url('/')
response = yield self.async_post(url, self.body)
data = json.loads(to_str(response.body))
- self.assertIsNone(data['status'])
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
+ self.assert_status_equal(data, None)
url = url.replace('http', 'ws')
ws_url = url + 'ws'
@@ -230,9 +239,7 @@ class TestAppBasic(AsyncHTTPTestCase):
url = self.get_url('/')
response = yield self.async_post(url, self.body)
data = json.loads(to_str(response.body))
- self.assertIsNone(data['status'])
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
+ self.assert_status_equal(data, None)
url = url.replace('http', 'ws')
ws_url = url + 'ws?id='
@@ -246,9 +253,7 @@ class TestAppBasic(AsyncHTTPTestCase):
url = self.get_url('/')
response = yield self.async_post(url, self.body)
data = json.loads(to_str(response.body))
- self.assertIsNone(data['status'])
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
+ self.assert_status_equal(data, None)
url = url.replace('http', 'ws')
ws_url = url + 'ws?id=1' + data['id']
@@ -263,9 +268,7 @@ class TestAppBasic(AsyncHTTPTestCase):
url = self.get_url('/')
response = yield self.async_post(url, body)
data = json.loads(to_str(response.body))
- self.assertIsNone(data['status'])
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
+ self.assert_status_equal(data, None)
url = url.replace('http', 'ws')
ws_url = url + 'ws?id=' + data['id']
@@ -310,9 +313,7 @@ class TestAppBasic(AsyncHTTPTestCase):
body = urlencode(self.body_dict)
response = yield self.async_post(url, body)
data = json.loads(to_str(response.body))
- self.assertIsNone(data['status'])
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
+ self.assert_status_equal(data, None)
url = url.replace('http', 'ws')
ws_url = url + 'ws?id=' + data['id']
@@ -333,9 +334,7 @@ class TestAppBasic(AsyncHTTPTestCase):
}
response = yield self.async_post(url, body, headers=headers)
data = json.loads(to_str(response.body))
- self.assertIsNone(data['status'])
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
+ self.assert_status_equal(data, None)
url = url.replace('http', 'ws')
ws_url = url + 'ws?id=' + data['id']
@@ -404,7 +403,6 @@ class TestAppBasic(AsyncHTTPTestCase):
@tornado.testing.gen_test
def test_app_post_form_with_large_body_size_by_multipart_form(self):
- url = self.get_url('/')
privatekey = 'h' * (2 * max_body_size)
files = [('privatekey', 'user_rsa_key', privatekey)]
content_type, body = encode_multipart_formdata(self.body_dict.items(),
@@ -414,32 +412,27 @@ class TestAppBasic(AsyncHTTPTestCase):
}
with self.assertRaises(HTTPError) as ctx:
- yield self.async_post(url, body, headers=headers)
+ yield self.async_post('/', body, headers=headers)
self.assertIn('Bad Request', ctx.exception.message)
@tornado.testing.gen_test
def test_app_post_form_with_large_body_size_by_urlencoded_form(self):
- url = self.get_url('/')
privatekey = 'h' * (2 * max_body_size)
body = self.body + '&privatekey=' + privatekey
with self.assertRaises(HTTPError) as ctx:
- yield self.async_post(url, body)
+ yield self.async_post('/', body)
self.assertIn('Bad Request', ctx.exception.message)
@tornado.testing.gen_test
def test_app_with_user_keyonly_for_bad_authentication_type(self):
- url = self.get_url('/')
self.body_dict.update(username='keyonly', password='foo')
body = urlencode(self.body_dict)
- response = yield self.async_post(url, body)
+ response = yield self.async_post('/', body)
self.assertEqual(response.code, 200)
- data = json.loads(to_str(response.body))
- self.assertIsNone(data['id'])
- self.assertIsNone(data['encoding'])
- self.assertIn('Bad authentication type', data['status'])
+ self.assert_status_in(json.loads(to_str(response.body)), 'Bad authentication type') # noqa
-class OtherTestBase(AsyncHTTPTestCase):
+class OtherTestBase(TestAppBase):
sshserver_port = 3300
headers = {'Cookie': '_xsrf=yummy'}
debug = False
@@ -472,9 +465,6 @@ class OtherTestBase(AsyncHTTPTestCase):
app = make_app(make_handlers(loop, options), get_app_settings(options))
return app
- def get_httpserver_options(self):
- return get_server_settings(options)
-
def setUp(self):
print('='*20)
self.running = True
@@ -524,14 +514,10 @@ class TestAppWithLargeBuffer(OtherTestBase):
@tornado.testing.gen_test
def test_app_for_sending_message_with_large_size(self):
url = self.get_url('/')
- client = self.get_http_client()
body = urlencode(dict(self.body, username='foo'))
- response = yield client.fetch(url, method='POST', body=body,
- headers=self.headers)
+ response = yield self.async_post(url, body, self.headers)
data = json.loads(to_str(response.body))
- self.assertIsNone(data['status'])
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
+ self.assert_status_equal(data, None)
url = url.replace('http', 'ws')
ws_url = url + 'ws?id=' + data['id']
@@ -559,14 +545,8 @@ class TestAppWithRejectPolicy(OtherTestBase):
@tornado.testing.gen_test
def test_app_with_hostname_not_in_hostkeys(self):
- url = self.get_url('/')
- client = self.get_http_client()
- body = urlencode(self.body)
- response = yield client.fetch(url, method='POST', body=body,
- headers=self.headers)
+ response = yield self.async_post('/', urlencode(self.body), self.headers) # noqa
data = json.loads(to_str(response.body))
- self.assertIsNone(data['id'])
- self.assertIsNone(data['encoding'])
message = 'Connection to {}:{} is not allowed.'.format(self.body['hostname'], self.sshserver_port) # noqa
self.assertEqual(message, data['status'])
@@ -582,14 +562,8 @@ class TestAppWithBadHostKey(OtherTestBase):
@tornado.testing.gen_test
def test_app_with_bad_host_key(self):
- url = self.get_url('/')
- client = self.get_http_client()
- body = urlencode(self.body)
- response = yield client.fetch(url, method='POST', body=body,
- headers=self.headers)
+ response = yield self.async_post('/', urlencode(self.body), self.headers) # noqa
data = json.loads(to_str(response.body))
- self.assertIsNone(data['id'])
- self.assertIsNone(data['encoding'])
self.assertEqual('Bad host key.', data['status'])
@@ -690,23 +664,13 @@ class TestAppWithTooManyConnections(OtherTestBase):
clients['127.0.0.1'] = {'fake_worker_id': None}
url = self.get_url('/')
- client = self.get_http_client()
- body = urlencode(self.body)
- response = yield client.fetch(url, method='POST', body=body,
- headers=self.headers)
+ response = yield self.async_post(url, urlencode(self.body), self.headers) # noqa
data = json.loads(to_str(response.body))
- self.assertIsNone(data['id'])
- self.assertIsNone(data['encoding'])
- self.assertEqual(data['status'], 'Too many connections.')
+ self.assertEqual('Too many connections.', data['status']) # noqa
clients['127.0.0.1'].clear()
-
- response = yield client.fetch(url, method='POST', body=body,
- headers=self.headers)
- data = json.loads(to_str(response.body))
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
- self.assertIsNone(data['status'])
+ response = yield self.async_post(url, urlencode(self.body), self.headers) # noqa
+ self.assert_status_equal(json.loads(to_str(response.body)), None)
class TestAppWithCrossOriginOperation(OtherTestBase):
@@ -715,58 +679,28 @@ class TestAppWithCrossOriginOperation(OtherTestBase):
@tornado.testing.gen_test
def test_app_with_wrong_event_origin(self):
- url = self.get_url('/')
- client = self.get_http_client()
body = urlencode(dict(self.body, _origin='localhost'))
- response = yield client.fetch(url, method='POST', body=body,
- headers=self.headers)
- data = json.loads(to_str(response.body))
- self.assertIsNone(data['id'])
- self.assertIsNone(data['encoding'])
- self.assertEqual(
- 'Cross origin operation is not allowed.', data['status']
- )
+ response = yield self.async_post('/', body, self.headers)
+ self.assert_status_equal(json.loads(to_str(response.body)), 'Cross origin operation is not allowed.') # noqa
@tornado.testing.gen_test
def test_app_with_wrong_header_origin(self):
- url = self.get_url('/')
- client = self.get_http_client()
- body = urlencode(self.body)
headers = dict(self.headers, Origin='localhost')
- response = yield client.fetch(url, method='POST', body=body,
- headers=headers)
- data = json.loads(to_str(response.body))
- self.assertIsNone(data['id'])
- self.assertIsNone(data['encoding'])
- self.assertEqual(
- 'Cross origin operation is not allowed.', data['status']
- )
+ response = yield self.async_post('/', urlencode(self.body), headers)
+ self.assert_status_equal(json.loads(to_str(response.body)), 'Cross origin operation is not allowed.') # noqa
@tornado.testing.gen_test
def test_app_with_correct_event_origin(self):
- url = self.get_url('/')
- client = self.get_http_client()
body = urlencode(dict(self.body, _origin=self.origin))
- response = yield client.fetch(url, method='POST', body=body,
- headers=self.headers)
- data = json.loads(to_str(response.body))
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
- self.assertIsNone(data['status'])
+ response = yield self.async_post('/', body, self.headers)
+ self.assert_status_equal(json.loads(to_str(response.body)), None)
self.assertIsNone(response.headers.get('Access-Control-Allow-Origin'))
@tornado.testing.gen_test
def test_app_with_correct_header_origin(self):
- url = self.get_url('/')
- client = self.get_http_client()
- body = urlencode(self.body)
headers = dict(self.headers, Origin=self.origin)
- response = yield client.fetch(url, method='POST', body=body,
- headers=headers)
- data = json.loads(to_str(response.body))
- self.assertIsNotNone(data['id'])
- self.assertIsNotNone(data['encoding'])
- self.assertIsNone(data['status'])
+ response = yield self.async_post('/', urlencode(self.body), headers)
+ self.assert_status_equal(json.loads(to_str(response.body)), None)
self.assertEqual(
response.headers.get('Access-Control-Allow-Origin'), self.origin
)