Commit 12f4c0d0 by Sheng

Pass server settings for test

parent 021a3a62
...@@ -12,7 +12,7 @@ from tests.sshserver import run_ssh_server, banner ...@@ -12,7 +12,7 @@ from tests.sshserver import run_ssh_server, banner
from tests.utils import encode_multipart_formdata, read_file, make_tests_data_path # noqa from tests.utils import encode_multipart_formdata, read_file, make_tests_data_path # noqa
from webssh.main import make_app, make_handlers from webssh.main import make_app, make_handlers
from webssh.settings import ( from webssh.settings import (
get_app_settings, max_body_size, swallow_http_errors get_app_settings, get_server_settings, max_body_size, swallow_http_errors
) )
from webssh.utils import to_str from webssh.utils import to_str
...@@ -45,6 +45,7 @@ class TestAppBasic(AsyncHTTPTestCase): ...@@ -45,6 +45,7 @@ class TestAppBasic(AsyncHTTPTestCase):
options.policy = random.choice(['warning', 'autoadd']) options.policy = random.choice(['warning', 'autoadd'])
options.hostFile = '' options.hostFile = ''
options.sysHostFile = '' options.sysHostFile = ''
options.proxies = ''
app = make_app(make_handlers(loop, options), get_app_settings(options)) app = make_app(make_handlers(loop, options), get_app_settings(options))
return app return app
...@@ -63,9 +64,7 @@ class TestAppBasic(AsyncHTTPTestCase): ...@@ -63,9 +64,7 @@ class TestAppBasic(AsyncHTTPTestCase):
print('='*20) print('='*20)
def get_httpserver_options(self): def get_httpserver_options(self):
options = super(TestAppBasic, self).get_httpserver_options() return get_server_settings(options)
options.update(max_body_size=max_body_size)
return options
def assert_response(self, bstr, response): def assert_response(self, bstr, response):
if swallow_http_errors: if swallow_http_errors:
...@@ -443,8 +442,9 @@ class OtherTestBase(AsyncHTTPTestCase): ...@@ -443,8 +442,9 @@ class OtherTestBase(AsyncHTTPTestCase):
headers = {'Cookie': '_xsrf=yummy'} headers = {'Cookie': '_xsrf=yummy'}
debug = False debug = False
policy = None policy = None
hostFile = None hostFile = ''
sysHostFile = None sysHostFile = ''
proxies = ''
body = { body = {
'hostname': '127.0.0.1', 'hostname': '127.0.0.1',
'port': '', 'port': '',
...@@ -458,11 +458,15 @@ class OtherTestBase(AsyncHTTPTestCase): ...@@ -458,11 +458,15 @@ class OtherTestBase(AsyncHTTPTestCase):
loop = self.io_loop loop = self.io_loop
options.debug = self.debug options.debug = self.debug
options.policy = self.policy if self.policy else random.choice(['warning', 'autoadd']) # noqa options.policy = self.policy if self.policy else random.choice(['warning', 'autoadd']) # noqa
options.hostFile = self.hostFile if self.hostFile else '' options.hostFile = self.hostFile
options.sysHostFile = self.sysHostFile if self.sysHostFile else '' options.sysHostFile = self.sysHostFile
options.proxies = self.proxies
app = make_app(make_handlers(loop, options), get_app_settings(options)) app = make_app(make_handlers(loop, options), get_app_settings(options))
return app return app
def get_httpserver_options(self):
return get_server_settings(options)
def setUp(self): def setUp(self):
print('='*20) print('='*20)
self.running = True self.running = True
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment