Coverage for custom_components/supernotify/button.py: 100%

1"""Button platform: reset every runtime override in one press. 

2 

3The same as calling supernotify.reset_overrides with no `kind` - every scenario, recipient, 

4delivery and transport switched on or off at runtime is put back to its configured state. 

5""" 

6 

7from __future__ import annotations 

8 

9from typing import TYPE_CHECKING 

10 

11from homeassistant.components.button import ButtonEntity 

12from homeassistant.const import EntityCategory 

13 

14from .hass_api import ha_device_info 

15 

16if TYPE_CHECKING: 

17 from homeassistant.core import HomeAssistant 

18 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback 

19 

20 from . import SupernotifyConfigEntry 

21 from .engine import SupernotifyEngine 

22 

23 

24async def async_setup_entry( 

25 hass: HomeAssistant, 

26 entry: SupernotifyConfigEntry, 

27 async_add_entities: AddConfigEntryEntitiesCallback, 

28) -> None: 

29 """Add the reset overrides button.""" 

30 _ = hass 

31 async_add_entities([SupernotifyResetOverridesButton(entry.runtime_data, entry.entry_id)]) 

32 

33 

34class SupernotifyResetOverridesButton(ButtonEntity): 

35 """Put everything switched on or off at runtime back to its configured state.""" 

36 

37 _attr_has_entity_name = True 

38 _attr_entity_category = EntityCategory.CONFIG 

39 _attr_translation_key = "reset_overrides" 

40 _attr_unique_id = "reset_overrides" 

41 _attr_should_poll = False 

42 

43 def __init__(self, engine: SupernotifyEngine, entry_id: str) -> None: 

44 self._engine = engine 

45 self._attr_device_info = ha_device_info(entry_id) 

46 

47 async def async_press(self) -> None: 

48 self._engine.reset_overrides()