Merge pull request #133 from lsst-ts/tickets/DM-48037

Tickets/dm48037: Update the enabled_offline script
lsst-ts · Dec 9, 2024 · c2fc13e · c2fc13e
2 parents 7390656 + e5d9928
commit c2fc13e
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 5 deletions.
diff --git a/python/lsst/ts/IntegrationTests/configs/enabled_offline_state_transition_configs.py b/python/lsst/ts/IntegrationTests/configs/enabled_offline_state_transition_configs.py
@@ -58,19 +58,23 @@
 yaml_string = yaml.safe_load(
     """
     data:
+    - [DREAM, OFFLINE]
     - [DSM:2, OFFLINE]
     - [DSM:1, OFFLINE]
     - [DIMM:1, OFFLINE]
     - [DIMM:2, OFFLINE]
+    - [EPM:1, OFFLINE]
     - [ESS:1, OFFLINE]
-    - [ESS:101, OFFLINE]
-    - [ESS:102, OFFLINE]
-    - [ESS:103, OFFLINE]
     - [ESS:104, OFFLINE]
     - [ESS:105, OFFLINE]
     - [ESS:106, OFFLINE]
     - [ESS:107, OFFLINE]
     - [ESS:108, OFFLINE]
+    - [ESS:109, OFFLINE]
+    - [ESS:110, OFFLINE]
+    - [ESS:111 OFFLINE]
+    - [ESS:112 OFFLINE]
+    - [ESS:113 OFFLINE]
     - [ESS:201, OFFLINE]
     - [ESS:202, OFFLINE]
     - [ESS:203, OFFLINE]
@@ -91,8 +95,8 @@
     data:
     - [Test:42, OFFLINE]
     - [ScriptQueue:1, OFFLINE]
-    - [ScriptQueue:2, OFFLINE]
     - [ScriptQueue:3, OFFLINE]
+    - [ScriptQueue:2, OFFLINE]
     """
 )
 
@@ -139,3 +143,15 @@
 registry["gencam_enabled_offline"] = yaml.safe_dump(
     yaml_string, explicit_start=True, canonical=True
 )
+
+# ATBuilding
+yaml_string = yaml.safe_load(
+    """
+    data:
+    - [ATBuilding, OFFLINE]
+    """
+)
+
+registry["atbuilding_enabled_offline"] = yaml.safe_dump(
+    yaml_string, explicit_start=True, canonical=True
+)
diff --git a/python/lsst/ts/IntegrationTests/enabled_offline.py b/python/lsst/ts/IntegrationTests/enabled_offline.py
@@ -42,6 +42,7 @@ class EnabledOffline(BaseScript):
         registry["sched_ocps_enabled_offline"],
         "",
         "",
+        registry["atbuilding_enabled_offline"],
         registry["mtcs_enabled_offline"],
         registry["eas_enabled_offline"],
         registry["maintel_enabled_offline"],
@@ -54,6 +55,7 @@ class EnabledOffline(BaseScript):
         ("set_summary_state.py", BaseScript.is_standard),
         ("auxtel/offline_atcs.py", BaseScript.is_standard),
         ("auxtel/offline_latiss.py", BaseScript.is_standard),
+        ("set_summary_state.py", BaseScript.is_standard),
         ("maintel/offline_mtcs.py", BaseScript.is_standard),
         ("set_summary_state.py", BaseScript.is_standard),
         ("set_summary_state.py", BaseScript.is_standard),

diff --git a/tests/test_enabled_offline.py b/tests/test_enabled_offline.py
@@ -59,7 +59,7 @@ async def test_enabled_offline(self) -> None:
         # Assert script was added to ScriptQueue.
         self.assertEqual(len(self.controller.queue_list), num_scripts)
         # Assert scripts passed.
-        self.assertEqual(script_class.script_states, [8, 8, 8, 8, 8, 8, 8, 8, 8])
+        self.assertEqual(script_class.script_states, [8, 8, 8, 8, 8, 8, 8, 8, 8, 8])
 
     async def asyncTearDown(self) -> None:
         await self.controller.close()