mirror of
https://github.com/home-assistant/core.git
synced 2025-07-23 21:27:38 +00:00
Bring back auto on off switches to lamarzocco (#119421)
* add auto on off switches
This commit is contained in:
parent
c8e9a3a8f4
commit
a515562a11
@ -139,6 +139,9 @@
|
|||||||
}
|
}
|
||||||
},
|
},
|
||||||
"switch": {
|
"switch": {
|
||||||
|
"auto_on_off": {
|
||||||
|
"name": "Auto on/off ({id})"
|
||||||
|
},
|
||||||
"steam_boiler": {
|
"steam_boiler": {
|
||||||
"name": "Steam boiler"
|
"name": "Steam boiler"
|
||||||
}
|
}
|
||||||
|
@ -13,7 +13,8 @@ from homeassistant.core import HomeAssistant
|
|||||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||||
|
|
||||||
from . import LaMarzoccoConfigEntry
|
from . import LaMarzoccoConfigEntry
|
||||||
from .entity import LaMarzoccoEntity, LaMarzoccoEntityDescription
|
from .coordinator import LaMarzoccoUpdateCoordinator
|
||||||
|
from .entity import LaMarzoccoBaseEntity, LaMarzoccoEntity, LaMarzoccoEntityDescription
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True, kw_only=True)
|
@dataclass(frozen=True, kw_only=True)
|
||||||
@ -52,12 +53,21 @@ async def async_setup_entry(
|
|||||||
"""Set up switch entities and services."""
|
"""Set up switch entities and services."""
|
||||||
|
|
||||||
coordinator = entry.runtime_data
|
coordinator = entry.runtime_data
|
||||||
async_add_entities(
|
|
||||||
|
entities: list[SwitchEntity] = []
|
||||||
|
entities.extend(
|
||||||
LaMarzoccoSwitchEntity(coordinator, description)
|
LaMarzoccoSwitchEntity(coordinator, description)
|
||||||
for description in ENTITIES
|
for description in ENTITIES
|
||||||
if description.supported_fn(coordinator)
|
if description.supported_fn(coordinator)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
entities.extend(
|
||||||
|
LaMarzoccoAutoOnOffSwitchEntity(coordinator, wake_up_sleep_entry_id)
|
||||||
|
for wake_up_sleep_entry_id in coordinator.device.config.wake_up_sleep_entries
|
||||||
|
)
|
||||||
|
|
||||||
|
async_add_entities(entities)
|
||||||
|
|
||||||
|
|
||||||
class LaMarzoccoSwitchEntity(LaMarzoccoEntity, SwitchEntity):
|
class LaMarzoccoSwitchEntity(LaMarzoccoEntity, SwitchEntity):
|
||||||
"""Switches representing espresso machine power, prebrew, and auto on/off."""
|
"""Switches representing espresso machine power, prebrew, and auto on/off."""
|
||||||
@ -78,3 +88,44 @@ class LaMarzoccoSwitchEntity(LaMarzoccoEntity, SwitchEntity):
|
|||||||
def is_on(self) -> bool:
|
def is_on(self) -> bool:
|
||||||
"""Return true if device is on."""
|
"""Return true if device is on."""
|
||||||
return self.entity_description.is_on_fn(self.coordinator.device.config)
|
return self.entity_description.is_on_fn(self.coordinator.device.config)
|
||||||
|
|
||||||
|
|
||||||
|
class LaMarzoccoAutoOnOffSwitchEntity(LaMarzoccoBaseEntity, SwitchEntity):
|
||||||
|
"""Switch representing espresso machine auto on/off."""
|
||||||
|
|
||||||
|
coordinator: LaMarzoccoUpdateCoordinator
|
||||||
|
_attr_translation_key = "auto_on_off"
|
||||||
|
|
||||||
|
def __init__(
|
||||||
|
self,
|
||||||
|
coordinator: LaMarzoccoUpdateCoordinator,
|
||||||
|
identifier: str,
|
||||||
|
) -> None:
|
||||||
|
"""Initialize the switch."""
|
||||||
|
super().__init__(coordinator, f"auto_on_off_{identifier}")
|
||||||
|
self._identifier = identifier
|
||||||
|
self._attr_translation_placeholders = {"id": identifier}
|
||||||
|
|
||||||
|
async def _async_enable(self, state: bool) -> None:
|
||||||
|
"""Enable or disable the auto on/off schedule."""
|
||||||
|
wake_up_sleep_entry = self.coordinator.device.config.wake_up_sleep_entries[
|
||||||
|
self._identifier
|
||||||
|
]
|
||||||
|
wake_up_sleep_entry.enabled = state
|
||||||
|
await self.coordinator.device.set_wake_up_sleep(wake_up_sleep_entry)
|
||||||
|
self.async_write_ha_state()
|
||||||
|
|
||||||
|
async def async_turn_on(self, **kwargs: Any) -> None:
|
||||||
|
"""Turn switch on."""
|
||||||
|
await self._async_enable(True)
|
||||||
|
|
||||||
|
async def async_turn_off(self, **kwargs: Any) -> None:
|
||||||
|
"""Turn switch off."""
|
||||||
|
await self._async_enable(False)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def is_on(self) -> bool:
|
||||||
|
"""Return true if switch is on."""
|
||||||
|
return self.coordinator.device.config.wake_up_sleep_entries[
|
||||||
|
self._identifier
|
||||||
|
].enabled
|
||||||
|
@ -1,4 +1,96 @@
|
|||||||
# serializer version: 1
|
# serializer version: 1
|
||||||
|
# name: test_auto_on_off_switches[entry.auto_on_off_Os2OswX]
|
||||||
|
EntityRegistryEntrySnapshot({
|
||||||
|
'aliases': set({
|
||||||
|
}),
|
||||||
|
'area_id': None,
|
||||||
|
'capabilities': None,
|
||||||
|
'config_entry_id': <ANY>,
|
||||||
|
'device_class': None,
|
||||||
|
'device_id': <ANY>,
|
||||||
|
'disabled_by': None,
|
||||||
|
'domain': 'switch',
|
||||||
|
'entity_category': None,
|
||||||
|
'entity_id': 'switch.gs01234_auto_on_off_os2oswx',
|
||||||
|
'has_entity_name': True,
|
||||||
|
'hidden_by': None,
|
||||||
|
'icon': None,
|
||||||
|
'id': <ANY>,
|
||||||
|
'labels': set({
|
||||||
|
}),
|
||||||
|
'name': None,
|
||||||
|
'options': dict({
|
||||||
|
}),
|
||||||
|
'original_device_class': None,
|
||||||
|
'original_icon': None,
|
||||||
|
'original_name': 'Auto on/off (Os2OswX)',
|
||||||
|
'platform': 'lamarzocco',
|
||||||
|
'previous_unique_id': None,
|
||||||
|
'supported_features': 0,
|
||||||
|
'translation_key': 'auto_on_off',
|
||||||
|
'unique_id': 'GS01234_auto_on_off_Os2OswX',
|
||||||
|
'unit_of_measurement': None,
|
||||||
|
})
|
||||||
|
# ---
|
||||||
|
# name: test_auto_on_off_switches[entry.auto_on_off_aXFz5bJ]
|
||||||
|
EntityRegistryEntrySnapshot({
|
||||||
|
'aliases': set({
|
||||||
|
}),
|
||||||
|
'area_id': None,
|
||||||
|
'capabilities': None,
|
||||||
|
'config_entry_id': <ANY>,
|
||||||
|
'device_class': None,
|
||||||
|
'device_id': <ANY>,
|
||||||
|
'disabled_by': None,
|
||||||
|
'domain': 'switch',
|
||||||
|
'entity_category': None,
|
||||||
|
'entity_id': 'switch.gs01234_auto_on_off_axfz5bj',
|
||||||
|
'has_entity_name': True,
|
||||||
|
'hidden_by': None,
|
||||||
|
'icon': None,
|
||||||
|
'id': <ANY>,
|
||||||
|
'labels': set({
|
||||||
|
}),
|
||||||
|
'name': None,
|
||||||
|
'options': dict({
|
||||||
|
}),
|
||||||
|
'original_device_class': None,
|
||||||
|
'original_icon': None,
|
||||||
|
'original_name': 'Auto on/off (aXFz5bJ)',
|
||||||
|
'platform': 'lamarzocco',
|
||||||
|
'previous_unique_id': None,
|
||||||
|
'supported_features': 0,
|
||||||
|
'translation_key': 'auto_on_off',
|
||||||
|
'unique_id': 'GS01234_auto_on_off_aXFz5bJ',
|
||||||
|
'unit_of_measurement': None,
|
||||||
|
})
|
||||||
|
# ---
|
||||||
|
# name: test_auto_on_off_switches[state.auto_on_off_Os2OswX]
|
||||||
|
StateSnapshot({
|
||||||
|
'attributes': ReadOnlyDict({
|
||||||
|
'friendly_name': 'GS01234 Auto on/off (Os2OswX)',
|
||||||
|
}),
|
||||||
|
'context': <ANY>,
|
||||||
|
'entity_id': 'switch.gs01234_auto_on_off_os2oswx',
|
||||||
|
'last_changed': <ANY>,
|
||||||
|
'last_reported': <ANY>,
|
||||||
|
'last_updated': <ANY>,
|
||||||
|
'state': 'on',
|
||||||
|
})
|
||||||
|
# ---
|
||||||
|
# name: test_auto_on_off_switches[state.auto_on_off_aXFz5bJ]
|
||||||
|
StateSnapshot({
|
||||||
|
'attributes': ReadOnlyDict({
|
||||||
|
'friendly_name': 'GS01234 Auto on/off (aXFz5bJ)',
|
||||||
|
}),
|
||||||
|
'context': <ANY>,
|
||||||
|
'entity_id': 'switch.gs01234_auto_on_off_axfz5bj',
|
||||||
|
'last_changed': <ANY>,
|
||||||
|
'last_reported': <ANY>,
|
||||||
|
'last_updated': <ANY>,
|
||||||
|
'state': 'on',
|
||||||
|
})
|
||||||
|
# ---
|
||||||
# name: test_device
|
# name: test_device
|
||||||
DeviceRegistryEntrySnapshot({
|
DeviceRegistryEntrySnapshot({
|
||||||
'area_id': None,
|
'area_id': None,
|
||||||
|
@ -14,7 +14,7 @@ from homeassistant.const import ATTR_ENTITY_ID
|
|||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.helpers import device_registry as dr, entity_registry as er
|
from homeassistant.helpers import device_registry as dr, entity_registry as er
|
||||||
|
|
||||||
from . import async_init_integration
|
from . import WAKE_UP_SLEEP_ENTRY_IDS, async_init_integration
|
||||||
|
|
||||||
from tests.common import MockConfigEntry
|
from tests.common import MockConfigEntry
|
||||||
|
|
||||||
@ -106,3 +106,55 @@ async def test_device(
|
|||||||
device = device_registry.async_get(entry.device_id)
|
device = device_registry.async_get(entry.device_id)
|
||||||
assert device
|
assert device
|
||||||
assert device == snapshot
|
assert device == snapshot
|
||||||
|
|
||||||
|
|
||||||
|
async def test_auto_on_off_switches(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
mock_lamarzocco: MagicMock,
|
||||||
|
mock_config_entry: MockConfigEntry,
|
||||||
|
entity_registry: er.EntityRegistry,
|
||||||
|
snapshot: SnapshotAssertion,
|
||||||
|
) -> None:
|
||||||
|
"""Test the auto on off/switches."""
|
||||||
|
|
||||||
|
await async_init_integration(hass, mock_config_entry)
|
||||||
|
|
||||||
|
serial_number = mock_lamarzocco.serial_number
|
||||||
|
|
||||||
|
for wake_up_sleep_entry_id in WAKE_UP_SLEEP_ENTRY_IDS:
|
||||||
|
state = hass.states.get(
|
||||||
|
f"switch.{serial_number}_auto_on_off_{wake_up_sleep_entry_id}"
|
||||||
|
)
|
||||||
|
assert state
|
||||||
|
assert state == snapshot(name=f"state.auto_on_off_{wake_up_sleep_entry_id}")
|
||||||
|
|
||||||
|
entry = entity_registry.async_get(state.entity_id)
|
||||||
|
assert entry
|
||||||
|
assert entry == snapshot(name=f"entry.auto_on_off_{wake_up_sleep_entry_id}")
|
||||||
|
|
||||||
|
await hass.services.async_call(
|
||||||
|
SWITCH_DOMAIN,
|
||||||
|
SERVICE_TURN_OFF,
|
||||||
|
{
|
||||||
|
ATTR_ENTITY_ID: f"switch.{serial_number}_auto_on_off_{wake_up_sleep_entry_id}",
|
||||||
|
},
|
||||||
|
blocking=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
wake_up_sleep_entry = mock_lamarzocco.config.wake_up_sleep_entries[
|
||||||
|
wake_up_sleep_entry_id
|
||||||
|
]
|
||||||
|
wake_up_sleep_entry.enabled = False
|
||||||
|
|
||||||
|
mock_lamarzocco.set_wake_up_sleep.assert_called_with(wake_up_sleep_entry)
|
||||||
|
|
||||||
|
await hass.services.async_call(
|
||||||
|
SWITCH_DOMAIN,
|
||||||
|
SERVICE_TURN_ON,
|
||||||
|
{
|
||||||
|
ATTR_ENTITY_ID: f"switch.{serial_number}_auto_on_off_{wake_up_sleep_entry_id}",
|
||||||
|
},
|
||||||
|
blocking=True,
|
||||||
|
)
|
||||||
|
wake_up_sleep_entry.enabled = True
|
||||||
|
mock_lamarzocco.set_wake_up_sleep.assert_called_with(wake_up_sleep_entry)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user