Factor out accessing planned_maintenances

Uninett · Jul 31, 2024 · e72201e · e72201e
1 parent 66a048a
commit e72201e
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/api/legacy_test.py b/tests/api/legacy_test.py
@@ -870,10 +870,9 @@ class TestZino1ServerProtocolPmListCommand:
     async def test_when_authenticated_should_list_all_pm_ids(
         self, authenticated_protocol, active_device_pm, active_portstate_pm
     ):
-        authenticated_protocol._state.planned_maintenances.planned_maintenances[active_device_pm.id] = active_device_pm
-        authenticated_protocol._state.planned_maintenances.planned_maintenances[active_portstate_pm.id] = (
-            active_portstate_pm
-        )
+        planned_maintenances = authenticated_protocol._state.planned_maintenances.planned_maintenances
+        planned_maintenances[active_device_pm.id] = active_device_pm
+        planned_maintenances[active_portstate_pm.id] = active_portstate_pm
 
         await authenticated_protocol.message_received("PM LIST")
         response = authenticated_protocol.transport.data_buffer.getvalue().decode("utf-8")