DSS0030 ISA Expiry case

monitoring/prober/infrastructure.py

@@ -100,7 +100,7 @@ def wrapper_default_scope(*args, **kwargs):
 resource_type_code_descriptions: Dict[ResourceType, str] = {}
 
 
-# Next code: 369
+# Next code: 372
 def register_resource_type(code: int, description: str) -> ResourceType:
     """Register that the specified code refers to the described resource.
 

monitoring/uss_qualifier/scenarios/astm/netrid/common/dss/isa_expiry.py

@@ -0,0 +1,143 @@
+import datetime
+import time
+from typing import Optional
+
+import arrow
+from uas_standards.astm.f3411 import v19, v22a
+
+from monitoring.monitorlib.rid import RIDVersion
+from monitoring.prober.infrastructure import register_resource_type
+from monitoring.uss_qualifier.common_data_definitions import Severity
+from monitoring.uss_qualifier.resources.astm.f3411.dss import DSSInstanceResource
+from monitoring.uss_qualifier.resources.interuss.id_generator import IDGeneratorResource
+from monitoring.uss_qualifier.resources.netrid.service_area import ServiceAreaResource
+from monitoring.uss_qualifier.scenarios.astm.netrid.common.dss import utils
+from monitoring.uss_qualifier.scenarios.astm.netrid.dss_wrapper import DSSWrapper
+from monitoring.uss_qualifier.scenarios.scenario import GenericTestScenario
+
+
+class ISAExpiry(GenericTestScenario):
+    """Based on test_isa_expiry.py from the legacy prober tool."""
+
+    ISA_TYPE = register_resource_type(369, "ISA")
+
+    create_isa_path: str
+
+    write_scope: str
+
+    def __init__(
+        self,
+        dss: DSSInstanceResource,
+        id_generator: IDGeneratorResource,
+        isa: ServiceAreaResource,
+    ):
+        super().__init__()
+        self._dss = (
+            dss.dss_instance
+        )  # TODO: delete once _delete_isa_if_exists updated to use dss_wrapper
+        self._dss_wrapper = DSSWrapper(self, dss.dss_instance)
+        self._isa_id = id_generator.id_factory.make_id(ISAExpiry.ISA_TYPE)
+        self._isa_version: Optional[str] = None
+        self._isa = isa.specification
+
+        now = arrow.utcnow().datetime
+        self._isa_start_time = self._isa.shifted_time_start(now)
+        self._isa_end_time = self._isa.shifted_time_end(now)
+        self._isa_area = [vertex.as_s2sphere() for vertex in self._isa.footprint]
+
+    def run(self):
+        self.begin_test_scenario()
+
+        self._setup_case()
+
+        self.begin_test_case("ISA Expiry")
+        self.begin_test_step("ISA Expiry")
+
+        self._check_expiry_behaviors()
+
+        self.end_test_step()
+        self.end_test_case()
+        self.end_test_scenario()
+
+    def _check_expiry_behaviors(self):
+        """
+        Once an ISA is expired, it may still be queried directly using its ID,
+        but it should not appear in searches anymore.
+        """
+
+        start_time = datetime.datetime.utcnow()
+        end_time = start_time + datetime.timedelta(seconds=5)
+
+        # Create a short lived ISA of a few seconds
+        with self.check("Create short lived ISA", [self._dss.participant_id]) as check:
+            created_isa = self._dss_wrapper.put_isa_expect_response_code(
+                check=check,
+                expected_error_codes={200},
+                area_vertices=self._isa_area,
+                alt_lo=self._isa.altitude_min,
+                alt_hi=self._isa.altitude_max,
+                start_time=start_time,
+                end_time=end_time,
+                uss_base_url=self._isa.base_url,
+                isa_id=self._isa_id,
+                isa_version=None,
+            )
+
+        # Wait for it to expire
+        time.sleep(5)
+
+        # Search for ISAs: we should not find the expired one
+        with self.check(
+            "Expired ISAs are not part of search results", [self._dss.participant_id]
+        ) as check:
+            isas = self._dss_wrapper.search_isas_expect_response_code(
+                main_check=check,
+                expected_error_codes={200},
+                area=self._isa_area,
+            )
+            if self._isa_id in isas.isas.keys():
+                check.record_failed(
+                    summary=f"Expired ISA {self._isa_id} found in search results",
+                    severity=Severity.Medium,
+                    participants=[self._dss.participant_id],
+                    details=f"Searched for area {self._isa_area} with unspecified end and start time.",
+                    query_timestamps=[
+                        created_isa.dss_query.query.request.timestamp,
+                        isas.query.request.timestamp,
+                    ],
+                )
+
+        with self.check(
+            "An expired ISA can be queried by its ID", [self._dss.participant_id]
+        ) as check:
+            self._dss_wrapper.get_isa(check, self._isa_id)
+
+    def _setup_case(self):
+        self.begin_test_case("Setup")
+
+        def _ensure_clean_workspace_step():
+            self.begin_test_step("Ensure clean workspace")
+
+            self._delete_isa_if_exists()
+
+            self.end_test_step()
+
+        _ensure_clean_workspace_step()
+
+        self.end_test_case()
+
+    def _delete_isa_if_exists(self):
+        utils.delete_isa_if_exists(
+            self,
+            isa_id=self._isa_id,
+            rid_version=self._dss.rid_version,
+            session=self._dss.client,
+            server_id=self._dss_wrapper.participant_id,
+        )
+
+    def cleanup(self):
+        self.begin_cleanup()
+
+        self._delete_isa_if_exists()
+
+        self.end_cleanup()

monitoring/uss_qualifier/scenarios/astm/netrid/common/dss/isa_subscription_interactions.py

@@ -0,0 +1,241 @@
+from typing import Optional
+
+import arrow
+import loguru
+
+from monitoring.prober.infrastructure import register_resource_type
+from monitoring.uss_qualifier.common_data_definitions import Severity
+from monitoring.uss_qualifier.resources.astm.f3411.dss import DSSInstanceResource
+from monitoring.uss_qualifier.resources.interuss.id_generator import IDGeneratorResource
+from monitoring.uss_qualifier.resources.netrid.service_area import ServiceAreaResource
+from monitoring.uss_qualifier.scenarios.astm.netrid.common.dss import utils
+from monitoring.uss_qualifier.scenarios.astm.netrid.dss_wrapper import DSSWrapper
+from monitoring.uss_qualifier.scenarios.scenario import GenericTestScenario
+
+
+class ISASubscriptionInteractions(GenericTestScenario):
+    """Based on the test_subscription_isa_interactions.py from the legacy prober tool."""
+
+    ISA_TYPE = register_resource_type(370, "ISA")
+
+    create_isa_path: str
+
+    write_scope: str
+
+    def __init__(
+        self,
+        dss: DSSInstanceResource,
+        id_generator: IDGeneratorResource,
+        isa: ServiceAreaResource,
+    ):
+        super().__init__()
+        self._dss = (
+            dss.dss_instance
+        )  # TODO: delete once _delete_isa_if_exists updated to use dss_wrapper
+        self._dss_wrapper = DSSWrapper(self, dss.dss_instance)
+        self._isa_id = id_generator.id_factory.make_id(
+            ISASubscriptionInteractions.ISA_TYPE
+        )
+        # sub id is isa_id with last character replaced with '1'
+        self._sub_id = self._isa_id[:-1] + "1"
+        self._isa_version: Optional[str] = None
+        self._isa = isa.specification
+
+        now = arrow.utcnow().datetime
+        self._isa_start_time = self._isa.shifted_time_start(now)
+        self._isa_end_time = self._isa.shifted_time_end(now)
+        self._isa_area = [vertex.as_s2sphere() for vertex in self._isa.footprint]
+
+    def run(self):
+        self.begin_test_scenario()
+
+        self._setup_case()
+
+        self.begin_test_case("ISA Subscription Interactions")
+        self.begin_test_step("ISA Subscription Interactions")
+
+        self._check_subscription_behaviors()
+
+        self.end_test_step()
+        self.end_test_case()
+        self.end_test_scenario()
+
+    def _check_subscription_behaviors(self):
+        """
+        - Create an ISA.
+        - Create a subscription, response should include the pre-existing ISA.
+        - Modify the ISA, response should include the subscription.
+        - Delete the ISA, response should include the subscription.
+        - Delete the subscription.
+        """
+
+        # Create an ISA
+        with self.check("Create an ISA", [self._dss.participant_id]) as check:
+            created_isa = self._dss_wrapper.put_isa_expect_response_code(
+                check=check,
+                expected_error_codes={200},
+                area_vertices=self._isa_area,
+                alt_lo=self._isa.altitude_min,
+                alt_hi=self._isa.altitude_max,
+                start_time=self._isa_start_time,
+                end_time=self._isa_end_time,
+                uss_base_url=self._isa.base_url,
+                isa_id=self._isa_id,
+                isa_version=None,
+            )
+
+        # Create a subscription
+        with self.check(
+            "Subscription for the ISA's area mentions the ISA",
+            [self._dss.participant_id],
+        ) as check:
+            created_subscription = self._dss_wrapper.put_sub(
+                check=check,
+                area_vertices=self._isa_area,
+                alt_lo=self._isa.altitude_min,
+                alt_hi=self._isa.altitude_max,
+                start_time=self._isa_start_time,
+                end_time=self._isa_end_time,
+                uss_base_url=self._isa.base_url,
+                sub_id=self._sub_id,
+                sub_version=None,
+            )
+
+            if created_isa.dss_query.isa.id not in [
+                isa.id for isa in created_subscription.isas
+            ]:
+                check.record_failed(
+                    summary="Subscription response does not include the freshly created ISA",
+                    severity=Severity.High,
+                    participants=[self._dss.participant_id],
+                    details=f"The subscription created for the area {self._isa_area} is expected to contain the ISA created for this same area. The returned subscription did not mention it.",
+                    query_timestamps=[
+                        created_isa.dss_query.query.request.timestamp,
+                        created_subscription.query.request.timestamp,
+                    ],
+                )
+
+        # Modify the ISA
+        with self.check(
+            "Response to the mutation of the ISA contains subscription ID",
+            [self._dss.participant_id],
+        ) as check:
+            mutated_isa = self._dss_wrapper.put_isa_expect_response_code(
+                check=check,
+                expected_error_codes={200},
+                area_vertices=self._isa_area,
+                alt_lo=self._isa.altitude_min,
+                alt_hi=self._isa.altitude_max - 1,  # reduce max altitude by one meter
+                start_time=self._isa_start_time,
+                end_time=self._isa_end_time,
+                uss_base_url=self._isa.base_url,
+                isa_id=self._isa_id,
+                isa_version=created_isa.dss_query.isa.version,
+            )
+
+            subscriptions_to_isa = []
+            for returned_subscriber in mutated_isa.dss_query.subscribers:
+                for sub_in_subscriber in returned_subscriber.raw.subscriptions:
+                    subscriptions_to_isa.append(sub_in_subscriber.subscription_id)
+
+            if created_subscription.subscription.id not in subscriptions_to_isa:
+                check.record_failed(
+                    summary="ISA mutation response does not contain expected subscription ID",
+                    severity=Severity.High,
+                    participants=[self._dss.participant_id],
+                    details="Mutating an ISA to which a subscription was made, the DSS failed to return the subscription ID in the response.",
+                    query_timestamps=[
+                        created_isa.dss_query.query.request.timestamp,
+                        created_subscription.query.request.timestamp,
+                        mutated_isa.dss_query.query.request.timestamp,
+                    ],
+                )
+
+        # Delete the ISA
+        with self.check(
+            "Response to the deletion of the ISA contains subscription ID",
+            [self._dss.participant_id],
+        ) as check:
+            deleted_isa = self._dss_wrapper.del_isa_expect_response_code(
+                main_check=check,
+                expected_error_codes={200},
+                isa_id=mutated_isa.dss_query.isa.id,
+                isa_version=mutated_isa.dss_query.isa.version,
+            )
+
+            subscriptions_to_deleted_isa = []
+            for returned_subscriber in deleted_isa.dss_query.subscribers:
+                for sub_in_subscriber in returned_subscriber.raw.subscriptions:
+                    subscriptions_to_isa.append(sub_in_subscriber.subscription_id)
+
+            loguru.logger.info(
+                f"Deleted ISA had subscribers: {subscriptions_to_deleted_isa}"
+            )
+
+            if created_subscription.subscription.id not in subscriptions_to_deleted_isa:
+                check.record_failed(
+                    summary="ISA deletion response does not contain expected subscription ID",
+                    severity=Severity.High,
+                    participants=[self._dss.participant_id],
+                    details="Deleting an ISA to which a subscription was made, the DSS failed to return the subscription ID in the response.",
+                    query_timestamps=[
+                        created_isa.dss_query.query.request.timestamp,
+                        created_subscription.query.request.timestamp,
+                        deleted_isa.dss_query.query.request.timestamp,
+                    ],
+                )
+
+        # Delete the subscription
+        with self.check(
+            "Successful subscription deletion",
+            [self._dss.participant_id],
+        ) as check:
+            self._dss_wrapper.del_sub(
+                check=check,
+                sub_id=self._sub_id,
+                sub_version=created_subscription.subscription.version,
+            )
+
+    def _setup_case(self):
+        self.begin_test_case("Setup")
+
+        def _ensure_clean_workspace_step():
+            self.begin_test_step("Ensure clean workspace")
+
+            self._delete_isa_if_exists()
+
+            self.end_test_step()
+
+        _ensure_clean_workspace_step()
+
+        self.end_test_case()
+
+    def _delete_isa_if_exists(self):
+        utils.delete_isa_if_exists(
+            self,
+            isa_id=self._isa_id,
+            rid_version=self._dss.rid_version,
+            session=self._dss.client,
+            server_id=self._dss_wrapper.participant_id,
+        )
+
+    def _clean_any_sub(self):
+        with self.check(
+            "Successful subscription query", [self._dss.participant_id]
+        ) as check:
+            fetched = self._dss_wrapper.search_subs(
+                check, [vertex.as_s2sphere() for vertex in self._isa.footprint]
+            )
+        for sub_id in fetched.subscriptions.keys():
+            with self.check(
+                "Successful subscription deletion", [self._dss.participant_id]
+            ) as check:
+                self._dss_wrapper.cleanup_sub(check, sub_id=sub_id)
+
+    def cleanup(self):
+        self.begin_cleanup()
+
+        self._delete_isa_if_exists()
+        self._clean_any_sub()
+
+        self.end_cleanup()