Add a test to ensure auth headers arg works

Author: john-dupuy

src/stac_api_validator/__main__.py

@@ -66,7 +66,7 @@
 )
 @click.option(
     "--auth-headers",
-    help="Basic auth to attach to the request, e.g., {'Authorization': 'api-key=<api-key>'}",
+    help="Auth headers to attach to the request, e.g., {'Authorization': 'api-key <api-key>'}",
 )
 def main(
     log_level: str,

src/stac_api_validator/validations.py

@@ -745,7 +745,8 @@ def validate_browseable(
                     Context.BROWSEABLE,
                     params={
                         "ids": item.id,
-                        "collections": getattr(item, "collection", None) or getattr(item, "collection_id", None)
+                        "collections": getattr(item, "collection", None)
+                        or getattr(item, "collection_id", None),
                     },
                     r_session=r_session,
                 )

tests/test_main.py

@@ -1,4 +1,6 @@
 """Test cases for the __main__ module."""
+import unittest.mock
+
 import pytest
 from click.testing import CliRunner
 
@@ -14,3 +16,26 @@ def runner() -> CliRunner:
 def test_main_fails(runner: CliRunner) -> None:
     result = runner.invoke(__main__.main)
     assert result.exit_code == 2
+
+
+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"]
+        )