formatting

Author: BarnabasG

src/pytest_api_cov/cli.py

@@ -89,7 +89,7 @@ def generate_pyproject_config(framework: str) -> str:
 """
 
 
-def cmd_init():
+def cmd_init() -> int:
     """Initialize pytest-api-cov setup in current directory."""
     print("🚀 pytest-api-cov Setup Wizard")
     print("=" * 40)
@@ -171,7 +171,7 @@ def read_root():
     return 0
 
 
-def main():
+def main() -> int:
     """Main CLI entry point."""
     parser = argparse.ArgumentParser(prog="pytest-api-cov", description="pytest API coverage plugin CLI tools")
 

src/pytest_api_cov/frameworks.py

@@ -31,7 +31,7 @@ def get_tracked_client(self, recorder: Optional["ApiCallRecorder"], test_name: s
             return self.app.test_client()
 
         class TrackingFlaskClient(FlaskClient):
-            def open(self, *args, **kwargs) -> Any:
+            def open(self, *args: Any, **kwargs: Any) -> Any:
                 path = kwargs.get("path") or (args[0] if args else None)
                 if path and hasattr(self.application.url_map, "bind"):
                     try:
@@ -58,9 +58,10 @@ def get_tracked_client(self, recorder: Optional["ApiCallRecorder"], test_name: s
             return TestClient(self.app)
 
         class TrackingFastAPIClient(TestClient):
-            def send(self, *args, **kwargs) -> Any:
+            def send(self, *args: Any, **kwargs: Any) -> Any:
                 request = args[0]
-                recorder.record_call(request.url.path, test_name)
+                if recorder is not None:
+                    recorder.record_call(request.url.path, test_name)
                 return super().send(*args, **kwargs)
 
         return TrackingFastAPIClient(self.app)

src/pytest_api_cov/models.py

@@ -1,6 +1,6 @@
 """Data models for pytest-api-cov."""
 
-from typing import Any, Dict, Iterator, List, Set, Tuple
+from typing import Any, Dict, Iterable, List, Set, Tuple
 
 from pydantic import BaseModel, Field
 
@@ -51,15 +51,15 @@ def __contains__(self, endpoint: str) -> bool:
         """Check if an endpoint has been recorded."""
         return endpoint in self.calls
 
-    def items(self) -> Iterator[Tuple[str, Set[str]]]:
+    def items(self) -> Iterable[Tuple[str, Set[str]]]:
         """Iterate over endpoint, callers pairs."""
         return self.calls.items()
 
-    def keys(self) -> Iterator[str]:
+    def keys(self) -> Iterable[str]:
         """Get all recorded endpoints."""
         return self.calls.keys()
 
-    def values(self) -> Iterator[Set[str]]:
+    def values(self) -> Iterable[Set[str]]:
         """Get all caller sets."""
         return self.calls.values()
 
@@ -84,7 +84,7 @@ def __len__(self) -> int:
         """Return number of discovered endpoints."""
         return len(self.endpoints)
 
-    def __iter__(self) -> Iterator[str]:  # type: ignore[override]
+    def __iter__(self) -> Iterable[str]:  # type: ignore[override]
         """Iterate over discovered endpoints."""
         return iter(self.endpoints)