forked from orbit-oss/flask
fix subdomain_matching=False behavior
This commit is contained in:
parent
07c7d5730a
commit
4995a775df
5 changed files with 98 additions and 43 deletions
|
|
@ -4,6 +4,7 @@ import typing as t
|
|||
import uuid
|
||||
import warnings
|
||||
import weakref
|
||||
from contextlib import nullcontext
|
||||
from datetime import datetime
|
||||
from datetime import timezone
|
||||
from platform import python_implementation
|
||||
|
|
@ -1483,6 +1484,48 @@ def test_request_locals():
|
|||
assert not flask.g
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
("subdomain_matching", "host_matching", "expect_base", "expect_abc", "expect_xyz"),
|
||||
[
|
||||
(False, False, "default", "default", "default"),
|
||||
(True, False, "default", "abc", "<invalid>"),
|
||||
(False, True, "default", "abc", "default"),
|
||||
],
|
||||
)
|
||||
def test_server_name_matching(
|
||||
subdomain_matching: bool,
|
||||
host_matching: bool,
|
||||
expect_base: str,
|
||||
expect_abc: str,
|
||||
expect_xyz: str,
|
||||
) -> None:
|
||||
app = flask.Flask(
|
||||
__name__,
|
||||
subdomain_matching=subdomain_matching,
|
||||
host_matching=host_matching,
|
||||
static_host="example.test" if host_matching else None,
|
||||
)
|
||||
app.config["SERVER_NAME"] = "example.test"
|
||||
|
||||
@app.route("/", defaults={"name": "default"}, host="<name>")
|
||||
@app.route("/", subdomain="<name>", host="<name>.example.test")
|
||||
def index(name: str) -> str:
|
||||
return name
|
||||
|
||||
client = app.test_client()
|
||||
|
||||
r = client.get(base_url="http://example.test")
|
||||
assert r.text == expect_base
|
||||
|
||||
r = client.get(base_url="http://abc.example.test")
|
||||
assert r.text == expect_abc
|
||||
|
||||
with pytest.warns() if subdomain_matching else nullcontext():
|
||||
r = client.get(base_url="http://xyz.other.test")
|
||||
|
||||
assert r.text == expect_xyz
|
||||
|
||||
|
||||
def test_server_name_subdomain():
|
||||
app = flask.Flask(__name__, subdomain_matching=True)
|
||||
client = app.test_client()
|
||||
|
|
|
|||
|
|
@ -951,7 +951,10 @@ def test_nesting_url_prefixes(
|
|||
|
||||
|
||||
def test_nesting_subdomains(app, client) -> None:
|
||||
subdomain = "api"
|
||||
app.subdomain_matching = True
|
||||
app.config["SERVER_NAME"] = "example.test"
|
||||
client.allow_subdomain_redirects = True
|
||||
|
||||
parent = flask.Blueprint("parent", __name__)
|
||||
child = flask.Blueprint("child", __name__)
|
||||
|
||||
|
|
@ -960,42 +963,31 @@ def test_nesting_subdomains(app, client) -> None:
|
|||
return "child"
|
||||
|
||||
parent.register_blueprint(child)
|
||||
app.register_blueprint(parent, subdomain=subdomain)
|
||||
|
||||
client.allow_subdomain_redirects = True
|
||||
|
||||
domain_name = "domain.tld"
|
||||
app.config["SERVER_NAME"] = domain_name
|
||||
response = client.get("/child/", base_url="http://api." + domain_name)
|
||||
app.register_blueprint(parent, subdomain="api")
|
||||
|
||||
response = client.get("/child/", base_url="http://api.example.test")
|
||||
assert response.status_code == 200
|
||||
|
||||
|
||||
def test_child_and_parent_subdomain(app, client) -> None:
|
||||
child_subdomain = "api"
|
||||
parent_subdomain = "parent"
|
||||
app.subdomain_matching = True
|
||||
app.config["SERVER_NAME"] = "example.test"
|
||||
client.allow_subdomain_redirects = True
|
||||
|
||||
parent = flask.Blueprint("parent", __name__)
|
||||
child = flask.Blueprint("child", __name__, subdomain=child_subdomain)
|
||||
child = flask.Blueprint("child", __name__, subdomain="api")
|
||||
|
||||
@child.route("/")
|
||||
def index():
|
||||
return "child"
|
||||
|
||||
parent.register_blueprint(child)
|
||||
app.register_blueprint(parent, subdomain=parent_subdomain)
|
||||
|
||||
client.allow_subdomain_redirects = True
|
||||
|
||||
domain_name = "domain.tld"
|
||||
app.config["SERVER_NAME"] = domain_name
|
||||
response = client.get(
|
||||
"/", base_url=f"http://{child_subdomain}.{parent_subdomain}.{domain_name}"
|
||||
)
|
||||
app.register_blueprint(parent, subdomain="parent")
|
||||
|
||||
response = client.get("/", base_url="http://api.parent.example.test")
|
||||
assert response.status_code == 200
|
||||
|
||||
response = client.get("/", base_url=f"http://{parent_subdomain}.{domain_name}")
|
||||
|
||||
response = client.get("/", base_url="http://parent.example.test")
|
||||
assert response.status_code == 404
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue