sarj-python-lint 0.81.8__tar.gz → 0.81.9__tar.gz

{sarj_python_lint-0.81.8 → sarj_python_lint-0.81.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: sarj-python-lint
-Version: 0.81.8
+Version: 0.81.9
 Summary: Custom Python lint rules — AST-based, pre-commit-friendly, hypermodern defaults
 Project-URL: Homepage, https://code-standards.sarj.ai/rules/python/
 Project-URL: Documentation, https://code-standards.sarj.ai/rules/python/

{sarj_python_lint-0.81.8 → sarj_python_lint-0.81.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sarj-python-lint"
-version = "0.81.8"
+version = "0.81.9"
 description = "Custom Python lint rules — AST-based, pre-commit-friendly, hypermodern defaults"
 readme = "README.md"
 authors = [{ name = "sarj-ai" }]

{sarj_python_lint-0.81.8 → sarj_python_lint-0.81.9}/src/sarj_python_lint/rules/no_raw_connection_in_tests.py

@@ -42,7 +42,7 @@ class NoRawConnectionInTests(Rule):
         category=RuleCategory.TESTING,
         autofix=AutofixPolicy.NONE,
         limitations=(
-            "Only test paths outside conftest.py and conventional shared test-support modules are inspected.",
+            "Only collected test modules outside conftest.py, conventional shared test-support modules, and migration-test trees are inspected.",
             "A receiver is reported only when a parameter, annotated local, or constructor call proves it is a psycopg ConnectionPool or AsyncConnectionPool.",
             "Pytest fixtures may use a connection internally for setup and cleanup, but fixtures that return or yield the connection remain reportable.",
         ),
@@ -81,12 +81,16 @@ class NoRawConnectionInTests(Rule):
 
     @override
     def check(self, path: Path, source: str) -> list[Diagnostic]:
-        if (
-            not is_test_path(path)
-            or path.name == "conftest.py"
-            or is_test_support_path(path)
-            or is_generated(path, source)
-        ):
+        excluded_path = any(
+            (
+                not is_test_path(path),
+                path.name == "conftest.py",
+                is_test_support_path(path),
+                _is_non_collected_test_support(path),
+                _is_migration_test(path),
+            )
+        )
+        if excluded_path or is_generated(path, source):
             return []
         tree = parse_or_none(path, source)
         if tree is None:
@@ -120,6 +124,18 @@ class NoRawConnectionInTests(Rule):
         return sorted(diagnostics, key=lambda item: (item.line, item.col))
 
 
+def _is_non_collected_test_support(path: Path) -> bool:
+    return "tests" in path.parts and not (path.name.startswith("test_") or path.name.endswith("_test.py"))
+
+
+def _is_migration_test(path: Path) -> bool:
+    parts = path.parts
+    return any(
+        part == "tests" and index + 1 < len(parts) and parts[index + 1] == "migrations"
+        for index, part in enumerate(parts)
+    )
+
+
 def _proven_pool_names(scope: ast.Module | ast.FunctionDef | ast.AsyncFunctionDef) -> frozenset[str]:
     names: set[str] = set()
     if isinstance(scope, (ast.FunctionDef, ast.AsyncFunctionDef)):