fix tests

Author: jennifer-bowser

tests/unit/test_schemas.py

@@ -1,10 +1,12 @@
 """Test schema validation functionality"""
 
+import os
 import re
 
 import pytest
 
 from anyvlm.schemas.vlm import (
+    BeaconHandover,
     HandoverType,
     ResponseField,
     ResponseSummary,
@@ -16,7 +18,22 @@
 
 @pytest.fixture(scope="module")
 def valid_handover_id() -> str:
-    return HandoverType().id
+    return os.environ.get("HANDOVER_TYPE_ID")  # type: ignore
+
+
+@pytest.fixture(scope="module")
+def beacon_handovers(valid_handover_id: str) -> list[BeaconHandover]:
+    handover_type = HandoverType(
+        id=valid_handover_id,
+        label=os.environ.get("HANDOVER_TYPE_LABEL"),  # type: ignore
+    )
+
+    return [
+        BeaconHandover(
+            handoverType=handover_type,
+            url=os.environ.get("BEACON_HANDOVER_URL"),  # type: ignore
+        )
+    ]
 
 
 @pytest.fixture(scope="module")
@@ -25,7 +42,7 @@ def response_summary() -> ResponseSummary:
 
 
 @pytest.fixture(scope="module")
-def responses_with_invalid_resultset_ids(valid_handover_id) -> list[ResponseField]:
+def responses_with_invalid_resultset_ids(valid_handover_id: str) -> list[ResponseField]:
     return [
         ResponseField(
             resultSets=[
@@ -54,7 +71,11 @@ def responses_with_invalid_resultset_ids(valid_handover_id) -> list[ResponseFiel
     ]
 
 
-def test_valid_resultset_id(response_summary, valid_handover_id):
+def test_valid_resultset_id(
+    valid_handover_id: str,
+    beacon_handovers: list[BeaconHandover],
+    response_summary: ResponseSummary,
+):
     response = ResponseField(
         resultSets=[
             ResultSet(
@@ -65,18 +86,30 @@ def test_valid_resultset_id(response_summary, valid_handover_id):
     )
 
     # Should NOT raise an error
-    vlm_response = VlmResponse(responseSummary=response_summary, response=response)
+    vlm_response = VlmResponse(
+        beaconHandovers=beacon_handovers,
+        responseSummary=response_summary,
+        response=response,
+    )
 
     assert (
         vlm_response.response.resultSets[0].id
         == f"{valid_handover_id} {Zygosity.HOMOZYGOUS}"
     )
 
 
-def test_invalid_resultset_ids(response_summary, responses_with_invalid_resultset_ids):
+def test_invalid_resultset_ids(
+    response_summary: ResponseSummary,
+    responses_with_invalid_resultset_ids: list[ResponseField],
+    beacon_handovers: list[BeaconHandover],
+):
     for response in responses_with_invalid_resultset_ids:
         with pytest.raises(
             ValueError,
             match=re.escape(VlmResponse.resultset_id_error_message_base),
         ):
-            VlmResponse(responseSummary=response_summary, response=response)
+            VlmResponse(
+                beaconHandovers=beacon_handovers,
+                responseSummary=response_summary,
+                response=response,
+            )