From 567584b1288607676ba2c4e0fd4630569ebf7c36 Mon Sep 17 00:00:00 2001
From: Michel Davit <michel.davit@orbitalize.com>
Date: Wed, 1 Oct 2025 14:26:20 +0200
Subject: [PATCH] [uss_qualifier] raise ScenarioDidNotStopError in fragments

---
 .../scenarios/astm/utm/dss/test_step_fragments.py        | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/monitoring/uss_qualifier/scenarios/astm/utm/dss/test_step_fragments.py b/monitoring/uss_qualifier/scenarios/astm/utm/dss/test_step_fragments.py
index 996018ab61..103e5509b5 100644
--- a/monitoring/uss_qualifier/scenarios/astm/utm/dss/test_step_fragments.py
+++ b/monitoring/uss_qualifier/scenarios/astm/utm/dss/test_step_fragments.py
@@ -350,8 +350,6 @@ def get_uss_availability(
     Returns:
         The state and version of the USS availability.
     """
-    availability = UssAvailabilityState.Unknown
-    version = ""
     with scenario.check(
         "USS Availability can be requested", dss.participant_id
     ) as check:
@@ -361,6 +359,7 @@ def get_uss_availability(
 
             availability = avail_response.status.availability
             version = avail_response.version
+            return availability, version
         except QueryError as e:
             scenario.record_queries(e.queries)
             avail_query = e.queries[0]
@@ -369,7 +368,7 @@ def get_uss_availability(
                 details=f"DSS responded code {avail_query.status_code}; {e}",
                 query_timestamps=[avail_query.request.timestamp],
             )
-    return availability, version
+            raise ScenarioDidNotStopError(check)
 
 
 def set_uss_availability(
@@ -387,7 +386,6 @@ def set_uss_availability(
     Returns:
         The new version of the USS availability.
     """
-    version = ""
     with scenario.check(
         "USS Availability can be updated", [dss.participant_id]
     ) as check:
@@ -398,6 +396,7 @@ def set_uss_availability(
                 version,
             )
             scenario.record_query(avail_query)
+            return version
         except QueryError as e:
             scenario.record_queries(e.queries)
             avail_query = e.queries[0]
@@ -406,7 +405,7 @@ def set_uss_availability(
                 details=f"DSS responded code {avail_query.status_code}; {e}",
                 query_timestamps=[avail_query.request.timestamp],
             )
-    return version
+            raise ScenarioDidNotStopError(check)
 
 
 def make_dss_report(