9b157f974d
* Reduce overhead to refire events - asyncio timers can fire early for a varity of reasons including poor clock resolution and performance. To solve this problem we re-arm async_track_point_in_utc_time and try again later when this happens. - On some platforms this means the async_track_point_in_utc_time can end up trying many times to prevent firing the timer early since as soon as it rearms it fires again and this repeats until we reach the appointed time. While there is not much we can do to prevent asyncio from firing the timer callback early, we can reduce the overhead when this happens by using avoiding creating datetime objects * tweak mocking * -vvv * fix time freeze being too broad in litterrobot * adjust
44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
"""Test the Litter-Robot button entity."""
|
|
from unittest.mock import MagicMock
|
|
|
|
from freezegun import freeze_time
|
|
|
|
from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN, SERVICE_PRESS
|
|
from homeassistant.const import ATTR_ENTITY_ID, ATTR_ICON, STATE_UNKNOWN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
from homeassistant.helpers.entity import EntityCategory
|
|
|
|
from .conftest import setup_integration
|
|
|
|
BUTTON_ENTITY = "button.test_reset_waste_drawer"
|
|
|
|
|
|
async def test_button(hass: HomeAssistant, mock_account: MagicMock) -> None:
|
|
"""Test the creation and values of the Litter-Robot button."""
|
|
await setup_integration(hass, mock_account, BUTTON_DOMAIN)
|
|
entity_registry = er.async_get(hass)
|
|
|
|
state = hass.states.get(BUTTON_ENTITY)
|
|
assert state
|
|
assert state.attributes.get(ATTR_ICON) == "mdi:delete-variant"
|
|
assert state.state == STATE_UNKNOWN
|
|
|
|
entry = entity_registry.async_get(BUTTON_ENTITY)
|
|
assert entry
|
|
assert entry.entity_category is EntityCategory.CONFIG
|
|
|
|
with freeze_time("2021-11-15 17:37:00", tz_offset=-7):
|
|
await hass.services.async_call(
|
|
BUTTON_DOMAIN,
|
|
SERVICE_PRESS,
|
|
{ATTR_ENTITY_ID: BUTTON_ENTITY},
|
|
blocking=True,
|
|
)
|
|
await hass.async_block_till_done()
|
|
assert mock_account.robots[0].reset_waste_drawer.call_count == 1
|
|
mock_account.robots[0].reset_waste_drawer.assert_called_with()
|
|
|
|
state = hass.states.get(BUTTON_ENTITY)
|
|
assert state
|
|
assert state.state == "2021-11-15T10:37:00+00:00"
|