Merge pull request #494 from JosyBan/493-unlock

493 unlock

README.md

@@ -46,6 +46,7 @@ It will create HA devices depending on what you have installed:
   - Service to start boost (provide boost amount in kWh as parameter)
   - Service to start smart boost (provide boost amount in kWh and desired finished time as paramters)
   - Service to stop boost
+  - Service to unlock the Zappi
 
 - Eddi
 

custom_components/myenergi/entity.py

@@ -70,6 +70,12 @@ async def stop_boost(self) -> None:
         await self.device.stop_boost()
         self.schedule_update_ha_state()
 
+    async def unlock(self) -> None:
+        _LOGGER.debug("unlock called")
+        """Unlock"""
+        await self.device.unlock()
+        self.schedule_update_ha_state()
+
 
 class MyenergiHub(CoordinatorEntity):
     def __init__(self, coordinator, config_entry, meta):

custom_components/myenergi/select.py

@@ -60,6 +60,11 @@ async def async_setup_entry(hass, entry, async_add_devices):
                 {},
                 "stop_boost",
             )
+            platform.async_register_entity_service(
+                "myenergi_unlock",
+                {},
+                "unlock",
+            )
             devices.append(ZappiChargeModeSelect(coordinator, device, entry))
         elif device.kind == "eddi":
             platform.async_register_entity_service(

custom_components/myenergi/services.yaml

@@ -71,3 +71,11 @@ myenergi_stop_boost:
       model: Zappi
     entity:
       domain: select
+myenergi_unlock:
+  name: Unlock
+  description: Unlock
+  target:
+    device:
+      model: Zappi
+    entity:
+      domain: select

tests/conftest.py

@@ -145,3 +145,10 @@ def mock_zappi_stop_boost():
     """Return a mocked Zappi object."""
     with patch("pymyenergi.client.Zappi.stop_boost") as stop_boost:
         yield stop_boost
+
+
+@pytest.fixture
+def mock_zappi_unlock():
+    """Return a mocked Zappi object."""
+    with patch("pymyenergi.client.Zappi.unlock") as unlock:
+        yield unlock

tests/test_services.py

@@ -90,3 +90,23 @@ async def test_stop_boost(
     assert mock_zappi_stop_boost.call_count == 0
     await hass.async_block_till_done()
     assert mock_zappi_stop_boost.call_count == 1
+
+
+async def test_unlock(
+    hass: HomeAssistant, mock_zappi_unlock: MagicMock
+) -> None:
+    """Verify device information includes expected details."""
+
+    await setup_mock_myenergi_config_entry(hass)
+
+    await hass.services.async_call(
+        "myenergi",
+        "myenergi_unlock",
+        {
+            ATTR_ENTITY_ID: TEST_ZAPPI_SELECT_CHARGE_MODE,
+        },
+        blocking=False,
+    )
+    assert mock_zappi_unlock.call_count == 0
+    await hass.async_block_till_done()
+    assert mock_zappi_unlock.call_count == 1