78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
import ast
|
|
from dataclasses import dataclass, field
|
|
from enum import Enum
|
|
from pathlib import Path
|
|
|
|
class FixtureClass(str, Enum):
|
|
UNIT = "unit"
|
|
MOCK_APP = "mock_app"
|
|
LIVE_GUI = "live_gui"
|
|
HEADLESS = "headless"
|
|
OPT_IN = "opt_in"
|
|
PERFORMANCE = "performance"
|
|
|
|
class Speed(str, Enum):
|
|
FAST = "fast"
|
|
MEDIUM = "medium"
|
|
SLOW = "slow"
|
|
VERY_SLOW = "very_slow"
|
|
|
|
@dataclass(frozen=True)
|
|
class CategoryRecord:
|
|
filename: str
|
|
fixture_class: FixtureClass
|
|
subsystems: list[str]
|
|
speed: Speed
|
|
batch_group: str
|
|
notes: str = ""
|
|
test_order: dict[str, int] = field(default_factory=dict)
|
|
source: str = "auto"
|
|
warnings: list[str] = field(default_factory=list)
|
|
|
|
_OPT_IN_PREFIXES: tuple[str, ...] = ("test_clean_install", "test_docker_build")
|
|
_PERF_KEYWORDS: tuple[str, ...] = ("perf", "stress", "phase_3_final", "phase_4_stress")
|
|
_FIXTURE_ARGS: dict[FixtureClass, frozenset[str]] = {
|
|
FixtureClass.LIVE_GUI: frozenset({"live_gui"}),
|
|
FixtureClass.MOCK_APP: frozenset({"mock_app", "app_instance"}),
|
|
}
|
|
|
|
def _fixture_args_used(source: str) -> set[str]:
|
|
try:
|
|
tree = ast.parse(source)
|
|
except SyntaxError:
|
|
return set()
|
|
found: set[str] = set()
|
|
for node in ast.walk(tree):
|
|
if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef)):
|
|
found.update(a.arg for a in node.args.args)
|
|
found.update(a.arg for a in node.args.kwonlyargs)
|
|
found.update(a.arg for a in node.args.posonlyargs)
|
|
return found
|
|
|
|
def _classify_fixture_class(path: Path, source: str) -> FixtureClass:
|
|
name = path.name
|
|
for prefix in _OPT_IN_PREFIXES:
|
|
if name.startswith(prefix):
|
|
return FixtureClass.OPT_IN
|
|
args = _fixture_args_used(source)
|
|
for fc, fixture_names in _FIXTURE_ARGS.items():
|
|
if args & fixture_names:
|
|
return fc
|
|
lowered = name.lower()
|
|
for kw in _PERF_KEYWORDS:
|
|
if kw in lowered:
|
|
return FixtureClass.PERFORMANCE
|
|
return FixtureClass.UNIT
|
|
|
|
def auto_classify(path: Path, durations: dict[str, float] | None = None) -> CategoryRecord:
|
|
source = path.read_text(encoding="utf-8", errors="replace")
|
|
fixture_class = _classify_fixture_class(path, source)
|
|
return CategoryRecord(
|
|
filename=path.name,
|
|
fixture_class=fixture_class,
|
|
subsystems=[],
|
|
speed=Speed.MEDIUM,
|
|
batch_group="",
|
|
source="auto",
|
|
)
|