Add power sensor to WiZ (#73260)

tests/components/wiz/test_sensor.py

@@ -1,11 +1,15 @@
 """Tests for the sensor platform."""
 
+from unittest.mock import AsyncMock
+
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import entity_registry as er
 
 from . import (
     FAKE_DUAL_HEAD_RGBWW_BULB,
     FAKE_MAC,
+    FAKE_SOCKET_WITH_POWER_MONITORING,
+    _mocked_wizlight,
     _patch_discovery,
     _patch_wizlight,
     async_push_update,
@@ -35,3 +39,29 @@ async def test_signal_strength(hass: HomeAssistant) -> None:
 
     await async_push_update(hass, bulb, {"mac": FAKE_MAC, "rssi": -50})
     assert hass.states.get(entity_id).state == "-50"
+
+
+async def test_power_monitoring(hass: HomeAssistant) -> None:
+    """Test power monitoring."""
+    socket = _mocked_wizlight(None, None, FAKE_SOCKET_WITH_POWER_MONITORING)
+    socket.power_monitoring = None
+    socket.get_power = AsyncMock(return_value=5.123)
+    _, entry = await async_setup_integration(
+        hass, wizlight=socket, bulb_type=FAKE_SOCKET_WITH_POWER_MONITORING
+    )
+    entity_id = "sensor.mock_title_current_power"
+    entity_registry = er.async_get(hass)
+    reg_entry = entity_registry.async_get(entity_id)
+    assert reg_entry.unique_id == f"{FAKE_MAC}_power"
+    updated_entity = entity_registry.async_update_entity(
+        entity_id=entity_id, disabled_by=None
+    )
+    assert not updated_entity.disabled
+
+    with _patch_discovery(), _patch_wizlight(device=socket):
+        await hass.config_entries.async_reload(entry.entry_id)
+        await hass.async_block_till_done()
+
+    assert hass.states.get(entity_id).state == "5.123"
+    await async_push_update(hass, socket, {"mac": FAKE_MAC, "pc": 800})
+    assert hass.states.get(entity_id).state == "0.8"