128 lines
3.4 KiB
Python
128 lines
3.4 KiB
Python
|
# flake8: noqa: E501
|
||
|
|
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from pathlib import Path
|
||
|
|
from typing import List, Tuple
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
CURRENT_DIR = Path(__file__).parent
|
||
|
|
|
||
|
|
|
||
|
|
def run_check(path_location: str) -> str:
|
||
|
|
"""Use mypy to check the given path location and return the output."""
|
||
|
|
|
||
|
|
mypy_path = "mypy"
|
||
|
|
path = CURRENT_DIR / path_location
|
||
|
|
command = [mypy_path, path.resolve().as_posix()]
|
||
|
|
|
||
|
|
process = subprocess.run(
|
||
|
|
command,
|
||
|
|
stdout=subprocess.PIPE,
|
||
|
|
stderr=subprocess.PIPE,
|
||
|
|
universal_newlines=True,
|
||
|
|
)
|
||
|
|
output = process.stdout + process.stderr
|
||
|
|
return output
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"path_location,expected",
|
||
|
|
(
|
||
|
|
(
|
||
|
|
"app_default.py",
|
||
|
|
[
|
||
|
|
(
|
||
|
|
"sanic.app.Sanic[sanic.config.Config, types.SimpleNamespace]",
|
||
|
|
5,
|
||
|
|
)
|
||
|
|
],
|
||
|
|
),
|
||
|
|
(
|
||
|
|
"app_custom_config.py",
|
||
|
|
[
|
||
|
|
(
|
||
|
|
"sanic.app.Sanic[app_custom_config.CustomConfig, types.SimpleNamespace]",
|
||
|
|
10,
|
||
|
|
)
|
||
|
|
],
|
||
|
|
),
|
||
|
|
(
|
||
|
|
"app_custom_ctx.py",
|
||
|
|
[("sanic.app.Sanic[sanic.config.Config, app_custom_ctx.Foo]", 9)],
|
||
|
|
),
|
||
|
|
(
|
||
|
|
"app_fully_custom.py",
|
||
|
|
[
|
||
|
|
(
|
||
|
|
"sanic.app.Sanic[app_fully_custom.CustomConfig, app_fully_custom.Foo]",
|
||
|
|
14,
|
||
|
|
)
|
||
|
|
],
|
||
|
|
),
|
||
|
|
(
|
||
|
|
"request_custom_sanic.py",
|
||
|
|
[
|
||
|
|
("types.SimpleNamespace", 18),
|
||
|
|
(
|
||
|
|
"sanic.app.Sanic[request_custom_sanic.CustomConfig, types.SimpleNamespace]",
|
||
|
|
19,
|
||
|
|
),
|
||
|
|
],
|
||
|
|
),
|
||
|
|
(
|
||
|
|
"request_custom_ctx.py",
|
||
|
|
[
|
||
|
|
("request_custom_ctx.Foo", 16),
|
||
|
|
(
|
||
|
|
"sanic.app.Sanic[sanic.config.Config, types.SimpleNamespace]",
|
||
|
|
17,
|
||
|
|
),
|
||
|
|
],
|
||
|
|
),
|
||
|
|
(
|
||
|
|
"request_fully_custom.py",
|
||
|
|
[
|
||
|
|
("request_fully_custom.CustomRequest", 32),
|
||
|
|
("request_fully_custom.RequestContext", 33),
|
||
|
|
(
|
||
|
|
"sanic.app.Sanic[request_fully_custom.CustomConfig, request_fully_custom.Foo]",
|
||
|
|
34,
|
||
|
|
),
|
||
|
|
],
|
||
|
|
),
|
||
|
|
),
|
||
|
|
)
|
||
|
|
def test_check_app_default(
|
||
|
|
path_location: str, expected: List[Tuple[str, int]]
|
||
|
|
) -> None:
|
||
|
|
output = run_check(f"samples/{path_location}")
|
||
|
|
|
||
|
|
for text, number in expected:
|
||
|
|
current = CURRENT_DIR / f"samples/{path_location}"
|
||
|
|
path = current.relative_to(CURRENT_DIR.parent)
|
||
|
|
|
||
|
|
target = Path.cwd()
|
||
|
|
while True:
|
||
|
|
note = _text_from_path(current, path, target, number, text)
|
||
|
|
try:
|
||
|
|
assert note in output, output
|
||
|
|
except AssertionError:
|
||
|
|
target = target.parent
|
||
|
|
if not target.exists():
|
||
|
|
raise
|
||
|
|
else:
|
||
|
|
break
|
||
|
|
|
||
|
|
|
||
|
|
def _text_from_path(
|
||
|
|
base: Path, path: Path, target: Path, number: int, text: str
|
||
|
|
) -> str:
|
||
|
|
relative_to_cwd = base.relative_to(target)
|
||
|
|
prefix = ".".join(relative_to_cwd.parts[:-1])
|
||
|
|
text = text.replace(path.stem, f"{prefix}.{path.stem}")
|
||
|
|
return f'{path}:{number}: note: Revealed type is "{text}"'
|