Fix mypy and suppress type checks for mocked test

stac-utils · Apr 17, 2023 · 773a0e0 · 773a0e0
1 parent cfc4731
commit 773a0e0
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 27 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/src/stac_api_validator/validations.py b/src/stac_api_validator/validations.py
@@ -279,7 +279,7 @@ def is_geojson_type(maybe_type: Optional[str]) -> bool:
 def get_catalog(data_dict: Dict[str, Any], r_session: Session) -> Catalog:
     stac_io = StacIO.default()
     if r_session.headers and r_session.headers.get("Authorization"):
-        stac_io.headers = {"Authorization": r_session.headers["Authorization"]}
+        stac_io.headers = {"Authorization": str(r_session.headers["Authorization"])}
     catalog = Catalog.from_dict(data_dict)
     catalog._stac_io = stac_io
     return catalog
@@ -598,7 +598,7 @@ def validate_api(
         try:
             headers = {}
             if r_session.headers and r_session.headers.get("Authorization"):
-                headers["Authorization"] = r_session.headers["Authorization"]
+                headers["Authorization"] = str(r_session.headers["Authorization"])
             catalog = Client.open(root_url, headers=headers)
             catalog.validate()
             for child in catalog.get_children():

diff --git a/tests/test_main.py b/tests/test_main.py
@@ -3,6 +3,7 @@
 
 import pytest
 from click.testing import CliRunner
+from typeguard import suppress_type_checks
 
 from stac_api_validator import __main__
 
@@ -20,22 +21,24 @@ def test_main_fails(runner: CliRunner) -> None:
 
 def test_retrieve_called_with_auth_headers(runner: CliRunner) -> None:
     expected_auth_header = {"Authorization": "api-key fake-api-key-value"}
-    with unittest.mock.patch(
-        "stac_api_validator.validations.retrieve"
-    ) as retrieve_mock:
-        runner.invoke(
-            __main__.main,
-            args=[
-                "--root-url",
-                "https://invalid",
-                "--conformance",
-                "core",
-                "--auth-headers",
-                f"{expected_auth_header}",
-            ],
-        )
-        assert retrieve_mock.call_count == 1
-        r_session = retrieve_mock.call_args.args[-1]
-        assert (
-            r_session.headers["Authorization"] == expected_auth_header["Authorization"]
-        )
+    with suppress_type_checks():
+        with unittest.mock.patch(
+            "stac_api_validator.validations.retrieve"
+        ) as retrieve_mock:
+            runner.invoke(
+                __main__.main,
+                args=[
+                    "--root-url",
+                    "https://invalid",
+                    "--conformance",
+                    "core",
+                    "--auth-headers",
+                    f"{expected_auth_header}",
+                ],
+            )
+            assert retrieve_mock.call_count == 1
+            r_session = retrieve_mock.call_args.args[-1]
+            assert (
+                r_session.headers["Authorization"]
+                == expected_auth_header["Authorization"]
+            )