Add zone and arming protocol tests (#125)

Author: nickw444

nessclient_tests/test_event.py

@@ -151,6 +151,18 @@ def test_zone_in_alarm_with_zones(self):
             [ZoneUpdate_1_16.Zone.ZONE_3, ZoneUpdate_1_16.Zone.ZONE_5],
         )
 
+    def test_zone_detector_low_battery_with_zones(self):
+        pkt = make_packet(CommandType.USER_INTERFACE, "110500")
+        event = ZoneUpdate_1_16.decode(pkt)
+        self.assertEqual(
+            event.request_id,
+            ZoneUpdate_1_16.RequestID.ZONE_1_16_DETECTOR_LOW_BATTERY,
+        )
+        self.assertEqual(
+            event.included_zones,
+            [ZoneUpdate_1_16.Zone.ZONE_1, ZoneUpdate_1_16.Zone.ZONE_3],
+        )
+
     def test_zone_1_16_excluded_plus_auto(self):
         pkt = make_packet(CommandType.USER_INTERFACE, "190500")
         event = ZoneUpdate_1_16.decode(pkt)
@@ -195,6 +207,18 @@ def test_zone_in_delay_with_zones(self):
             [ZoneUpdate_17_32.Zone.ZONE_17, ZoneUpdate_17_32.Zone.ZONE_19],
         )
 
+    def test_zone_detector_low_battery_with_zones(self):
+        pkt = make_packet(CommandType.USER_INTERFACE, "310500")
+        event = ZoneUpdate_17_32.decode(pkt)
+        self.assertEqual(
+            event.request_id,
+            ZoneUpdate_17_32.RequestID.ZONE_17_32_DETECTOR_LOW_BATTERY,
+        )
+        self.assertEqual(
+            event.included_zones,
+            [ZoneUpdate_17_32.Zone.ZONE_17, ZoneUpdate_17_32.Zone.ZONE_19],
+        )
+
     def test_zone_excluded_plus_auto(self):
         pkt = make_packet(CommandType.USER_INTERFACE, "330500")
         event = ZoneUpdate_17_32.decode(pkt)

nessclient_tests/test_integration.py

@@ -36,6 +36,19 @@ async def test_zone_events_update_zone_state(
     assert alarm.zones[2].triggered is False
 
 
+@pytest.mark.asyncio
+async def test_zone_23_events_update_zone_state(
+    client: Client, connection: Connection, alarm: Alarm
+) -> None:
+    # Zone 23 unsealed at 23:45 on 10/5/2008
+    await _feed_ascii(client, connection, "87020361002301080510234500E3")
+    assert alarm.zones[22].triggered is True
+
+    # Zone 23 sealed one minute later
+    await _feed_ascii(client, connection, "87020361012301080510234600E1")
+    assert alarm.zones[22].triggered is False
+
+
 @pytest.mark.asyncio
 async def test_arming_events_update_alarm_state(
     client: Client, connection: Connection, alarm: Alarm
@@ -49,6 +62,36 @@ async def test_arming_events_update_alarm_state(
     assert alarm.arming_state == ArmingState.DISARMED
 
 
+@pytest.mark.asyncio
+async def test_armed_vacation_event_updates_alarm_state(
+    client: Client, connection: Connection, alarm: Alarm
+) -> None:
+    callback = Mock()
+    client.on_state_change(callback)
+
+    await _feed_ascii(client, connection, "87020361280101080510234500DD")
+    assert alarm._arming_mode == ArmingMode.ARMED_VACATION
+
+    assert callback.call_args_list == [
+        call(ArmingState.ARMING, ArmingMode.ARMED_VACATION),
+    ]
+
+
+@pytest.mark.asyncio
+async def test_armed_highest_event_updates_alarm_state(
+    client: Client, connection: Connection, alarm: Alarm
+) -> None:
+    callback = Mock()
+    client.on_state_change(callback)
+
+    await _feed_ascii(client, connection, "870203612E0101080510234500D0")
+    assert alarm._arming_mode == ArmingMode.ARMED_HIGHEST
+
+    assert callback.call_args_list == [
+        call(ArmingState.ARMING, ArmingMode.ARMED_HIGHEST),
+    ]
+
+
 @pytest.mark.asyncio
 async def test_arm_away_emits_ascii_payload(
     client: Client, connection: Connection