Add testing

Author: mj23000

tests/test_events.py

@@ -2,6 +2,11 @@
 
 import pytest
 
+from homeassistant_api.models import (
+    ConfigEntryChange,
+    ConfigEntryDisabler,
+    ConfigEntryState,
+)
 from homeassistant_api.websocket import WebsocketClient
 
 
@@ -28,3 +33,40 @@ def test_listen_trigger(websocket_client: WebsocketClient) -> None:
             assert datetime.fromisoformat(
                 trigger["trigger"]["now"]
             ).timestamp() == pytest.approx(future.timestamp(), abs=1)
+
+
+def test_listen_config_entries(websocket_client: WebsocketClient) -> None:
+    with websocket_client.listen_config_entries() as flows:
+        for i, flow in zip(range(5), flows):
+            # The first "events" are currently available entries
+            if i == 0:
+                # Assumes that the first entry (sun.sun?) is enabled
+                assert flow[0].type is None
+                assert flow[0].entry.disabled_by is None
+                assert flow[0].entry.state == ConfigEntryState.LOADED
+
+                # Trigger an "updated" event
+                websocket_client.disable_config_entry(flow[0].entry.entry_id)
+
+            if i == 1:
+                assert flow[0].type == ConfigEntryChange.UPDATED
+                assert flow[0].entry.disabled_by == ConfigEntryDisabler.USER
+                assert flow[0].entry.state == ConfigEntryState.UNLOAD_IN_PROGRESS
+
+            if i == 2:
+                assert flow[0].type == ConfigEntryChange.UPDATED
+                assert flow[0].entry.disabled_by == ConfigEntryDisabler.USER
+                assert flow[0].entry.state == ConfigEntryState.NOT_LOADED
+
+                # Restore original state
+                websocket_client.enable_config_entry(flow[0].entry.entry_id)
+
+            if i == 3:
+                assert flow[0].type == ConfigEntryChange.UPDATED
+                assert flow[0].entry.disabled_by is None
+                assert flow[0].entry.state == ConfigEntryState.SETUP_IN_PROGRESS
+
+            if i == 4:
+                assert flow[0].type == ConfigEntryChange.UPDATED
+                assert flow[0].entry.disabled_by is None
+                assert flow[0].entry.state == ConfigEntryState.LOADED