Extend tests to support new components_json field.

lsst-ts · Dec 19, 2024 · bae1e06 · bae1e06
1 parent 08cd263
commit bae1e06
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/narrativelog/testutils.py b/src/narrativelog/testutils.py
@@ -63,6 +63,11 @@
 TEST_PRIMARY_HARDWARE_COMPONENTS = [
     f"primary_hardware_component{n}" for n in range(10)
 ]
+TEST_COMPONENTS_JSON = {
+    "systems": TEST_SYSTEMS,
+    "subsystems": TEST_SUBSYSTEMS,
+    "components": TEST_COMPONENTS,
+}
 
 # Type annotation aliases
 MessageDictT = dict[str, typing.Any]
@@ -396,6 +401,8 @@ def random_message() -> MessageDictT:
         primary_hardware_components=random_strings(
             TEST_PRIMARY_HARDWARE_COMPONENTS
         ),
+        # Added 2024-12-16
+        components_json=TEST_COMPONENTS_JSON,
         # Added 2023-10-24
         category=random_str(nchar=CATEGORY_LEN),
         time_lost_type=random.choice(["fault", "weather"]),
@@ -507,6 +514,7 @@ async def create_test_database(
             del pruned_message["components"]
             del pruned_message["primary_software_components"]
             del pruned_message["primary_hardware_components"]
+            del pruned_message["components_json"]
 
             # Insert the message
             result_message = await connection.execute(
@@ -529,6 +537,7 @@ async def create_test_database(
                     primary_hardware_components=message[
                         "primary_hardware_components"
                     ],
+                    components_json=message["components_json"],
                     message_id=data_message.id,
                 )
                 .returning(literal_column("*"))

diff --git a/tests/test_add_message.py b/tests/test_add_message.py
@@ -7,6 +7,7 @@
 
 from narrativelog.testutils import (
     TEST_COMPONENTS,
+    TEST_COMPONENTS_JSON,
     TEST_CSCS,
     TEST_PRIMARY_HARDWARE_COMPONENTS,
     TEST_PRIMARY_SOFTWARE_COMPONENTS,
@@ -90,6 +91,7 @@ async def test_add_message(self) -> None:
             add_args_full["primary_hardware_components"] = random_strings(
                 TEST_PRIMARY_HARDWARE_COMPONENTS
             )
+            add_args_full["components_json"] = TEST_COMPONENTS_JSON
             add_args_full["category"] = "test"
             add_args_full["time_lost_type"] = random.choice(
                 ["fault", "weather"]