[uss-qualifier] DSS0030 port ISA-subscriptions interactions test

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: 371
 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_subscription_interactions.py

@@ -0,0 +1,240 @@
+from typing import Optional
+
+import arrow
+
+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")
+
+    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'
+        # (the generated isa_id ends with a few '0's)
+        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("Create a subscription within the ISA footprint", [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,
+            )
+
+        # Check the subscription
+        with self.check(
+            "Subscription for the ISA's area mentions the ISA",
+            [self._dss.participant_id],
+        ) as check:
+            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_deleted_isa.append(sub_in_subscriber.subscription_id)
+
+            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,
+                    ],
+                )
+
+        for subscriber_url, notification in deleted_isa.notifications.items():
+            # For checking the notifications, we ignore the request we made for the subscription that we created.
+            if self._isa.base_url not in subscriber_url:
+                with self.check("Notified subscriber", [subscriber_url]) as check:
+                    # TODO: Find a better way to identify a subscriber who couldn't be notified:
+                    #  as-is the subscriber url causes the test-suite to crash when it writes its report
+                    if not notification.success:
+                        check.record_failed(
+                            "Could not notify ISA subscriber",
+                            Severity.Medium,
+                            f"Attempting to notify subscriber for ISA {self._isa_id} at {subscriber_url} resulted in {notification.status_code}",
+                            query_timestamps=[notification.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._clean_any_sub()
+
+            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,
+            participant_id=self._dss_wrapper.participant_id,
+        )
+
+    def _clean_any_sub(self):
+        utils.delete_any_subscription(self, self._dss_wrapper, self._isa.footprint)
+
+    def cleanup(self):
+        self.begin_cleanup()
+
+        self._delete_isa_if_exists()
+        self._clean_any_sub()
+
+        self.end_cleanup()

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

@@ -12,6 +12,7 @@
 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,
@@ -77,17 +78,7 @@ def _setup_case(self):
         self.end_test_case()
 
     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)
+        utils.delete_any_subscription(self, self._dss_wrapper, self._isa.footprint)
 
     def _ensure_clean_workspace_step(self):
         self.begin_test_step("Ensure clean workspace")

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

@@ -1,10 +1,12 @@
-from typing import Optional
+from typing import Optional, List
 
 from monitoring.monitorlib.fetch import rid as fetch
+from monitoring.monitorlib.geo import LatLngPoint
 from monitoring.monitorlib.mutate import rid as mutate
 from monitoring.monitorlib.infrastructure import UTMClientSession
 from monitoring.monitorlib.rid import RIDVersion
 from monitoring.uss_qualifier.common_data_definitions import Severity
+from monitoring.uss_qualifier.scenarios.astm.netrid.dss_wrapper import DSSWrapper
 from monitoring.uss_qualifier.scenarios.scenario import GenericTestScenario
 
 
@@ -50,6 +52,7 @@ def delete_isa_if_exists(
                     f"Attempting to delete ISA {isa_id} from the {participant_id} DSS returned error {deleted.dss_query.status_code}",
                     query_timestamps=[deleted.dss_query.query.request.timestamp],
                 )
+
         for subscriber_url, notification in deleted.notifications.items():
             with scenario.check("Notified subscriber", [subscriber_url]) as check:
                 # TODO: Find a better way to identify a subscriber who couldn't be notified
@@ -60,3 +63,29 @@ def delete_isa_if_exists(
                         f"Attempting to notify subscriber for ISA {isa_id} at {subscriber_url} resulted in {notification.status_code}",
                         query_timestamps=[notification.query.request.timestamp],
                     )
+
+
+def delete_any_subscription(
+    scenario: GenericTestScenario,
+    dss_wrapper: DSSWrapper,
+    area: List[LatLngPoint],
+):
+    """
+    Deletes any subscription that is returned for the passed area.
+
+    Args:
+        scenario: the scenario instance that will provide the checks
+        dss_wrapper: the dss on which to delete subscriptions
+        area: the area for which subscriptions are to be deleted
+    """
+    with scenario.check(
+            "Successful subscription query", [dss_wrapper.participant_id]
+    ) as check:
+        fetched = dss_wrapper.search_subs(
+            check, [vertex.as_s2sphere() for vertex in area]
+        )
+    for sub_id in fetched.subscriptions.keys():
+        with scenario.check(
+            "Successful subscription deletion", [dss_wrapper.participant_id]
+        ) as check:
+            dss_wrapper.cleanup_sub(check, sub_id=sub_id)

monitoring/uss_qualifier/scenarios/astm/netrid/v19/dss/__init__.py

@@ -1,4 +1,5 @@
 from .isa_simple import ISASimple
 from .isa_validation import ISAValidation
+from .isa_subscription_interactions import ISASubscriptionInteractions
 from .subscription_validation import SubscriptionValidation
 from .crdb_access import CRDBAccess