Extend tests to support new components_json field.

lsst-ts · Dec 18, 2024 · e0f2dff · e0f2dff
1 parent 66b8f7d
commit e0f2dff
Show file tree

Hide file tree

Showing 2 changed files with 7 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]

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"]