mirror of
https://github.com/frappe/gunicorn.git
synced 2026-01-14 11:09:11 +08:00
19cb68f4c3
* load application from factory function Use `ast.parse` to validate that the string passed to the CLI is either an attribute name or a function call. Use `ast.literal_eval` to parse any positional and keyword arguments to the function. Call the function to get the real application. Co-authored-by: Connor Brinton <connor.brinton@gmail.com> * test coverage for util.import_app * document app factory pattern
133 lines
4.3 KiB
Python
133 lines
4.3 KiB
Python
# -*- coding: utf-8 -
|
|
#
|
|
# This file is part of gunicorn released under the MIT license.
|
|
# See the NOTICE for more information.
|
|
import os
|
|
|
|
import pytest
|
|
|
|
from gunicorn import util
|
|
from gunicorn.errors import AppImportError
|
|
from urllib.parse import SplitResult
|
|
|
|
|
|
@pytest.mark.parametrize('test_input, expected', [
|
|
('unix://var/run/test.sock', 'var/run/test.sock'),
|
|
('unix:/var/run/test.sock', '/var/run/test.sock'),
|
|
('tcp://localhost', ('localhost', 8000)),
|
|
('tcp://localhost:5000', ('localhost', 5000)),
|
|
('', ('0.0.0.0', 8000)),
|
|
('[::1]:8000', ('::1', 8000)),
|
|
('[::1]:5000', ('::1', 5000)),
|
|
('[::1]', ('::1', 8000)),
|
|
('localhost:8000', ('localhost', 8000)),
|
|
('127.0.0.1:8000', ('127.0.0.1', 8000)),
|
|
('localhost', ('localhost', 8000)),
|
|
('fd://33', 33),
|
|
])
|
|
def test_parse_address(test_input, expected):
|
|
assert util.parse_address(test_input) == expected
|
|
|
|
|
|
def test_parse_address_invalid():
|
|
with pytest.raises(RuntimeError) as exc_info:
|
|
util.parse_address('127.0.0.1:test')
|
|
assert "'test' is not a valid port number." in str(exc_info.value)
|
|
|
|
|
|
def test_parse_fd_invalid():
|
|
with pytest.raises(RuntimeError) as exc_info:
|
|
util.parse_address('fd://asd')
|
|
assert "'asd' is not a valid file descriptor." in str(exc_info.value)
|
|
|
|
|
|
def test_http_date():
|
|
assert util.http_date(1508607753.740316) == 'Sat, 21 Oct 2017 17:42:33 GMT'
|
|
|
|
|
|
@pytest.mark.parametrize('test_input, expected', [
|
|
('1200:0000:AB00:1234:0000:2552:7777:1313', True),
|
|
('1200::AB00:1234::2552:7777:1313', False),
|
|
('21DA:D3:0:2F3B:2AA:FF:FE28:9C5A', True),
|
|
('1200:0000:AB00:1234:O000:2552:7777:1313', False),
|
|
])
|
|
def test_is_ipv6(test_input, expected):
|
|
assert util.is_ipv6(test_input) == expected
|
|
|
|
|
|
def test_warn(capsys):
|
|
util.warn('test warn')
|
|
_, err = capsys.readouterr()
|
|
assert '!!! WARNING: test warn' in err
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"value",
|
|
[
|
|
"support",
|
|
"support:app",
|
|
"support:create_app()",
|
|
"support:create_app('Gunicorn', 3)",
|
|
"support:create_app(count=3)",
|
|
],
|
|
)
|
|
def test_import_app_good(value):
|
|
assert util.import_app(value)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("value", "exc_type", "msg"),
|
|
[
|
|
("a:app", ImportError, "No module"),
|
|
("support:create_app(", AppImportError, "Failed to parse"),
|
|
("support:create.app()", AppImportError, "Function reference"),
|
|
("support:create_app(Gunicorn)", AppImportError, "literal values"),
|
|
("support:create.app", AppImportError, "attribute name"),
|
|
("support:wrong_app", AppImportError, "find attribute"),
|
|
("support:error_factory(1)", AppImportError, "error_factory() takes"),
|
|
("support:error_factory()", TypeError, "inner"),
|
|
("support:none_app", AppImportError, "find application object"),
|
|
("support:HOST", AppImportError, "callable"),
|
|
],
|
|
)
|
|
def test_import_app_bad(value, exc_type, msg):
|
|
with pytest.raises(exc_type) as exc_info:
|
|
util.import_app(value)
|
|
|
|
assert msg in str(exc_info.value)
|
|
|
|
|
|
def test_import_app_py_ext(monkeypatch):
|
|
monkeypatch.chdir(os.path.dirname(__file__))
|
|
|
|
with pytest.raises(ImportError) as exc_info:
|
|
util.import_app("support.py")
|
|
|
|
assert "did you mean" in str(exc_info.value)
|
|
|
|
|
|
def test_to_bytestring():
|
|
assert util.to_bytestring('test_str', 'ascii') == b'test_str'
|
|
assert util.to_bytestring('test_str®') == b'test_str\xc2\xae'
|
|
assert util.to_bytestring(b'byte_test_str') == b'byte_test_str'
|
|
with pytest.raises(TypeError) as exc_info:
|
|
util.to_bytestring(100)
|
|
msg = '100 is not a string'
|
|
assert msg in str(exc_info.value)
|
|
|
|
|
|
@pytest.mark.parametrize('test_input, expected', [
|
|
('https://example.org/a/b?c=1#d',
|
|
SplitResult(scheme='https', netloc='example.org', path='/a/b', query='c=1', fragment='d')),
|
|
('a/b?c=1#d',
|
|
SplitResult(scheme='', netloc='', path='a/b', query='c=1', fragment='d')),
|
|
('/a/b?c=1#d',
|
|
SplitResult(scheme='', netloc='', path='/a/b', query='c=1', fragment='d')),
|
|
('//a/b?c=1#d',
|
|
SplitResult(scheme='', netloc='', path='//a/b', query='c=1', fragment='d')),
|
|
('///a/b?c=1#d',
|
|
SplitResult(scheme='', netloc='', path='///a/b', query='c=1', fragment='d')),
|
|
])
|
|
def test_split_request_uri(test_input, expected):
|
|
assert util.split_request_uri(test_input) == expected
|