From 8f6c4f8b333c6c89676b227ce35682c94fdb6b8f Mon Sep 17 00:00:00 2001
From: Erik Montnemery <erik@montnemery.com>
Date: Sun, 17 Mar 2024 09:45:49 +0100
Subject: [PATCH] Add tests of homeworks binary sensor (#113577)

---
 .coveragerc                                   |  1 -
 .../snapshots/test_binary_sensor.ambr         | 40 +++++++++++
 .../homeworks/test_binary_sensor.py           | 69 +++++++++++++++++++
 3 files changed, 109 insertions(+), 1 deletion(-)
 create mode 100644 tests/components/homeworks/snapshots/test_binary_sensor.ambr
 create mode 100644 tests/components/homeworks/test_binary_sensor.py

diff --git a/.coveragerc b/.coveragerc
index 7a66f88ed87..e7de89daffb 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -550,7 +550,6 @@ omit =
     homeassistant/components/homematic/sensor.py
     homeassistant/components/homematic/switch.py
     homeassistant/components/homeworks/__init__.py
-    homeassistant/components/homeworks/binary_sensor.py
     homeassistant/components/homeworks/button.py
     homeassistant/components/homeworks/light.py
     homeassistant/components/horizon/media_player.py
diff --git a/tests/components/homeworks/snapshots/test_binary_sensor.ambr b/tests/components/homeworks/snapshots/test_binary_sensor.ambr
new file mode 100644
index 00000000000..effa4315861
--- /dev/null
+++ b/tests/components/homeworks/snapshots/test_binary_sensor.ambr
@@ -0,0 +1,40 @@
+# serializer version: 1
+# name: test_binary_sensor_attributes_state_update
+  StateSnapshot({
+    'attributes': ReadOnlyDict({
+      'friendly_name': 'Foyer Keypad Morning',
+      'homeworks_address': '[02:08:02:01]',
+    }),
+    'context': <ANY>,
+    'entity_id': 'binary_sensor.foyer_keypad_morning',
+    'last_changed': <ANY>,
+    'last_updated': <ANY>,
+    'state': 'unknown',
+  })
+# ---
+# name: test_binary_sensor_attributes_state_update.1
+  StateSnapshot({
+    'attributes': ReadOnlyDict({
+      'friendly_name': 'Foyer Keypad Morning',
+      'homeworks_address': '[02:08:02:01]',
+    }),
+    'context': <ANY>,
+    'entity_id': 'binary_sensor.foyer_keypad_morning',
+    'last_changed': <ANY>,
+    'last_updated': <ANY>,
+    'state': 'on',
+  })
+# ---
+# name: test_binary_sensor_attributes_state_update.2
+  StateSnapshot({
+    'attributes': ReadOnlyDict({
+      'friendly_name': 'Foyer Keypad Morning',
+      'homeworks_address': '[02:08:02:01]',
+    }),
+    'context': <ANY>,
+    'entity_id': 'binary_sensor.foyer_keypad_morning',
+    'last_changed': <ANY>,
+    'last_updated': <ANY>,
+    'state': 'off',
+  })
+# ---
diff --git a/tests/components/homeworks/test_binary_sensor.py b/tests/components/homeworks/test_binary_sensor.py
new file mode 100644
index 00000000000..cc9e580c23c
--- /dev/null
+++ b/tests/components/homeworks/test_binary_sensor.py
@@ -0,0 +1,69 @@
+"""Tests for the Lutron Homeworks Series 4 and 8 binary sensor."""
+from unittest.mock import ANY, MagicMock
+
+from freezegun.api import FrozenDateTimeFactory
+from pyhomeworks.pyhomeworks import HW_KEYPAD_LED_CHANGED
+from syrupy.assertion import SnapshotAssertion
+
+from homeassistant.components.binary_sensor import DOMAIN as BINARY_SENSOR_DOMAIN
+from homeassistant.components.homeworks import KEYPAD_LEDSTATE_POLL_COOLDOWN
+from homeassistant.const import STATE_OFF, STATE_ON, STATE_UNKNOWN
+from homeassistant.core import HomeAssistant
+
+from tests.common import MockConfigEntry, async_fire_time_changed
+
+
+async def test_binary_sensor_attributes_state_update(
+    hass: HomeAssistant,
+    mock_config_entry: MockConfigEntry,
+    mock_homeworks: MagicMock,
+    freezer: FrozenDateTimeFactory,
+    snapshot: SnapshotAssertion,
+) -> None:
+    """Test Homeworks binary sensor state changes."""
+    entity_id = "binary_sensor.foyer_keypad_morning"
+    mock_controller = MagicMock()
+    mock_homeworks.return_value = mock_controller
+
+    mock_config_entry.add_to_hass(hass)
+    await hass.config_entries.async_setup(mock_config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    mock_homeworks.assert_called_once_with("192.168.0.1", 1234, ANY)
+    hw_callback = mock_homeworks.mock_calls[0][1][2]
+
+    assert entity_id in hass.states.async_entity_ids(BINARY_SENSOR_DOMAIN)
+
+    state = hass.states.get(entity_id)
+    assert state.state == STATE_UNKNOWN
+    assert state == snapshot
+
+    freezer.tick(KEYPAD_LEDSTATE_POLL_COOLDOWN + 1)
+    async_fire_time_changed(hass)
+    await hass.async_block_till_done()
+    assert len(mock_controller._send.mock_calls) == 1
+    assert mock_controller._send.mock_calls[0][1] == ("RKLS, [02:08:02:01]",)
+
+    hw_callback(
+        HW_KEYPAD_LED_CHANGED,
+        [
+            "[02:08:02:01]",
+            [1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
+        ],
+    )
+    await hass.async_block_till_done()
+    state = hass.states.get(entity_id)
+    assert state.state == STATE_ON
+    assert state == snapshot
+
+    hw_callback(
+        HW_KEYPAD_LED_CHANGED,
+        [
+            "[02:08:02:01]",
+            [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
+        ],
+    )
+    await hass.async_block_till_done()
+    state = hass.states.get(entity_id)
+    assert state.state == STATE_OFF
+    assert state == snapshot