Formatting

Author: alukach

src/stac_auth_proxy/app.py

@@ -49,6 +49,7 @@ def configure_app(
     **settings_kwargs : Any
         Keyword arguments used to configure the health and conformance checks if
         ``settings`` is not provided.
+
     """
     settings = settings or Settings(**settings_kwargs)
 

src/stac_auth_proxy/lifespan.py

@@ -72,7 +72,6 @@ async def check_conformance(
     """Check if the upstream API supports a given conformance class."""
     required_conformances: dict[str, list[str]] = {}
     for middleware in middleware_classes:
-
         for conformance in getattr(middleware.cls, attr_name, []):
             required_conformances.setdefault(conformance, []).append(
                 middleware.cls.__name__
@@ -127,6 +126,7 @@ def build_lifespan(settings: Settings | None = None, **settings_kwargs: Any):
     -------
     Callable[[FastAPI], AsyncContextManager[Any]]
         A callable suitable for the ``lifespan`` parameter of ``FastAPI``.
+
     """
     if settings is None:
         settings = Settings(**settings_kwargs)

src/stac_auth_proxy/utils/cache.py

@@ -78,9 +78,10 @@ def get_value_by_path(obj: dict, path: str, default: Any = None) -> Any:
         path: The dot notation path (e.g. "payload.sub")
         default: Default value to return if path doesn't exist
 
-    Returns
+    Returns:
     -------
         The value at the specified path or default if path doesn't exist
+
     """
     try:
         for key in path.split("."):

src/stac_auth_proxy/utils/middleware.py

@@ -29,6 +29,7 @@ def should_transform_response(
         Returns
         -------
             bool: True if the response should be transformed
+
         """
         ...
 
@@ -39,10 +40,12 @@ def transform_json(self, data: Any, request: Request) -> Any:
 
         Args:
             data: The parsed JSON data
+            request: The HTTP request object
 
-        Returns
+        Returns:
         -------
             The transformed JSON data
+
         """
         ...
 

tests/test_authn.py

@@ -312,14 +312,14 @@ def test_with_invalid_tokens_fails(invalid_token, expected_status, source_api_se
     )
     client = TestClient(test_app)
     response = client.get("/collections", headers={"Authorization": invalid_token})
-    assert (
-        response.status_code == expected_status
-    ), f"GET request should fail with token: {invalid_token}"
+    assert response.status_code == expected_status, (
+        f"GET request should fail with token: {invalid_token}"
+    )
 
     response = client.options("/collections", headers={"Authorization": invalid_token})
-    assert (
-        response.status_code == 200
-    ), f"OPTIONS request should succeed with token: {invalid_token}"
+    assert response.status_code == 200, (
+        f"OPTIONS request should succeed with token: {invalid_token}"
+    )
 
 
 def test_options_requests_with_cors_headers(source_api_server):
@@ -339,9 +339,9 @@ def test_options_requests_with_cors_headers(source_api_server):
     }
 
     response = client.options("/collections", headers=cors_headers)
-    assert (
-        response.status_code == 200
-    ), "OPTIONS request with CORS headers should succeed"
+    assert response.status_code == 200, (
+        "OPTIONS request with CORS headers should succeed"
+    )
 
 
 @pytest.mark.parametrize(

tests/test_filters_jinja2.py

@@ -188,9 +188,9 @@ async def test_search_post(
         "filter-lang": "cql2-json",
     }
 
-    assert (
-        proxied_body == expected_output
-    ), "POST query should combine filter expressions."
+    assert proxied_body == expected_output, (
+        "POST query should combine filter expressions."
+    )
 
 
 @pytest.mark.parametrize(
@@ -241,9 +241,9 @@ async def test_search_get(
         ),
         "filter-lang": filter_lang,
     }
-    assert (
-        proxied_request.query_params == expected_output
-    ), "GET query should combine filter expressions."
+    assert proxied_request.query_params == expected_output, (
+        "GET query should combine filter expressions."
+    )
 
 
 @pytest.mark.parametrize(
@@ -444,9 +444,9 @@ async def test_collections_list(
         ),
         "filter-lang": filter_lang,
     }
-    assert (
-        proxied_request.query_params == expected_output
-    ), "Collections query should combine filter expressions."
+    assert proxied_request.query_params == expected_output, (
+        "Collections query should combine filter expressions."
+    )
 
 
 @pytest.mark.parametrize(

tests/test_lifespan.py

@@ -89,13 +89,16 @@ def test_lifespan_reusable():
     """Ensure the public lifespan handler runs health and conformance checks."""
     upstream_url = "https://example.com"
     oidc_discovery_url = "https://example.com/.well-known/openid-configuration"
-    with patch(
-        "stac_auth_proxy.lifespan.check_server_health",
-        new=AsyncMock(),
-    ) as mock_health, patch(
-        "stac_auth_proxy.lifespan.check_conformance",
-        new=AsyncMock(),
-    ) as mock_conf:
+    with (
+        patch(
+            "stac_auth_proxy.lifespan.check_server_health",
+            new=AsyncMock(),
+        ) as mock_health,
+        patch(
+            "stac_auth_proxy.lifespan.check_conformance",
+            new=AsyncMock(),
+        ) as mock_conf,
+    ):
         app = FastAPI(
             lifespan=build_lifespan(
                 upstream_url=upstream_url,

tests/test_openapi.py

@@ -146,20 +146,20 @@ def test_oidc_in_openapi_spec_public_endpoints(
             security = config.get("security")
 
             if method == "options":
-                assert (
-                    not security
-                ), f"OPTIONS {path} requests should not require authentication"
+                assert not security, (
+                    f"OPTIONS {path} requests should not require authentication"
+                )
                 continue
 
             if security:
-                assert (
-                    path not in expected_required_auth
-                ), f"Path {path} should not require authentication"
+                assert path not in expected_required_auth, (
+                    f"Path {path} should not require authentication"
+                )
                 continue
 
-            assert (
-                path in expected_required_auth
-            ), f"Path {path} should require authentication"
+            assert path in expected_required_auth, (
+                f"Path {path} should require authentication"
+            )
             assert any(
                 method.casefold() == m.casefold() for m in expected_required_auth[path]
             )

tests/test_reverse_proxy.py

@@ -279,6 +279,6 @@ async def test_nginx_headers_behavior(scope_overrides, headers, expected_forward
     # Check that the Forwarded header contains expected values
     forwarded = result_headers["Forwarded"]
     for key, expected_value in expected_forwarded.items():
-        assert (
-            f"{key}={expected_value}" in forwarded
-        ), f"Expected {key}={expected_value} in {forwarded}"
+        assert f"{key}={expected_value}" in forwarded, (
+            f"Expected {key}={expected_value} in {forwarded}"
+        )