You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
bazarr/libs/waitress/tests/test_runner.py

192 lines
5.6 KiB

5 years ago
import contextlib
import os
import sys
if sys.version_info[:2] == (2, 6): # pragma: no cover
import unittest2 as unittest
else: # pragma: no cover
import unittest
from waitress import runner
class Test_match(unittest.TestCase):
def test_empty(self):
self.assertRaisesRegexp(
ValueError, "^Malformed application ''$", runner.match, ""
)
def test_module_only(self):
self.assertRaisesRegexp(
ValueError, r"^Malformed application 'foo\.bar'$", runner.match, "foo.bar"
)
def test_bad_module(self):
self.assertRaisesRegexp(
ValueError,
r"^Malformed application 'foo#bar:barney'$",
runner.match,
"foo#bar:barney",
)
def test_module_obj(self):
self.assertTupleEqual(
runner.match("foo.bar:fred.barney"), ("foo.bar", "fred.barney")
)
class Test_resolve(unittest.TestCase):
def test_bad_module(self):
self.assertRaises(
ImportError, runner.resolve, "nonexistent", "nonexistent_function"
)
def test_nonexistent_function(self):
self.assertRaisesRegexp(
AttributeError,
r"has no attribute 'nonexistent_function'",
runner.resolve,
"os.path",
"nonexistent_function",
)
def test_simple_happy_path(self):
from os.path import exists
self.assertIs(runner.resolve("os.path", "exists"), exists)
def test_complex_happy_path(self):
# Ensure we can recursively resolve object attributes if necessary.
self.assertEquals(runner.resolve("os.path", "exists.__name__"), "exists")
class Test_run(unittest.TestCase):
def match_output(self, argv, code, regex):
argv = ["waitress-serve"] + argv
with capture() as captured:
self.assertEqual(runner.run(argv=argv), code)
self.assertRegexpMatches(captured.getvalue(), regex)
captured.close()
def test_bad(self):
self.match_output(["--bad-opt"], 1, "^Error: option --bad-opt not recognized")
def test_help(self):
self.match_output(["--help"], 0, "^Usage:\n\n waitress-serve")
def test_no_app(self):
self.match_output([], 1, "^Error: Specify one application only")
def test_multiple_apps_app(self):
self.match_output(["a:a", "b:b"], 1, "^Error: Specify one application only")
def test_bad_apps_app(self):
self.match_output(["a"], 1, "^Error: Malformed application 'a'")
def test_bad_app_module(self):
self.match_output(["nonexistent:a"], 1, "^Error: Bad module 'nonexistent'")
self.match_output(
["nonexistent:a"],
1,
(
r"There was an exception \((ImportError|ModuleNotFoundError)\) "
"importing your module.\n\nIt had these arguments: \n"
"1. No module named '?nonexistent'?"
),
)
def test_cwd_added_to_path(self):
def null_serve(app, **kw):
pass
sys_path = sys.path
current_dir = os.getcwd()
try:
os.chdir(os.path.dirname(__file__))
argv = [
"waitress-serve",
"fixtureapps.runner:app",
]
self.assertEqual(runner.run(argv=argv, _serve=null_serve), 0)
finally:
sys.path = sys_path
os.chdir(current_dir)
def test_bad_app_object(self):
self.match_output(
["waitress.tests.fixtureapps.runner:a"], 1, "^Error: Bad object name 'a'"
)
def test_simple_call(self):
import waitress.tests.fixtureapps.runner as _apps
def check_server(app, **kw):
self.assertIs(app, _apps.app)
self.assertDictEqual(kw, {"port": "80"})
argv = [
"waitress-serve",
"--port=80",
"waitress.tests.fixtureapps.runner:app",
]
self.assertEqual(runner.run(argv=argv, _serve=check_server), 0)
def test_returned_app(self):
import waitress.tests.fixtureapps.runner as _apps
def check_server(app, **kw):
self.assertIs(app, _apps.app)
self.assertDictEqual(kw, {"port": "80"})
argv = [
"waitress-serve",
"--port=80",
"--call",
"waitress.tests.fixtureapps.runner:returns_app",
]
self.assertEqual(runner.run(argv=argv, _serve=check_server), 0)
class Test_helper(unittest.TestCase):
def test_exception_logging(self):
from waitress.runner import show_exception
regex = (
r"There was an exception \(ImportError\) importing your module."
r"\n\nIt had these arguments: \n1. My reason"
)
with capture() as captured:
try:
raise ImportError("My reason")
except ImportError:
self.assertEqual(show_exception(sys.stderr), None)
self.assertRegexpMatches(captured.getvalue(), regex)
captured.close()
regex = (
r"There was an exception \(ImportError\) importing your module."
r"\n\nIt had no arguments."
)
with capture() as captured:
try:
raise ImportError
except ImportError:
self.assertEqual(show_exception(sys.stderr), None)
self.assertRegexpMatches(captured.getvalue(), regex)
captured.close()
@contextlib.contextmanager
def capture():
from waitress.compat import NativeIO
fd = NativeIO()
sys.stdout = fd
sys.stderr = fd
yield fd
sys.stdout = sys.__stdout__
sys.stderr = sys.__stderr__