hass-core/tests/components/tessie/test_update.py
Brett Adams 2afe3364ca
Add names to all Tessie entities (#106267)
* Add name back to device tracker and lock

* Add name to media player

* Add name to update

* Update snapshot

* Keep none name on Update

* Re-add update
2023-12-27 07:56:23 +01:00

42 lines
1.2 KiB
Python

"""Test the Tessie update platform."""
from unittest.mock import patch
from homeassistant.components.update import (
ATTR_IN_PROGRESS,
DOMAIN as UPDATE_DOMAIN,
SERVICE_INSTALL,
)
from homeassistant.const import ATTR_ENTITY_ID, STATE_ON
from homeassistant.core import HomeAssistant
from .common import setup_platform
async def test_updates(hass: HomeAssistant) -> None:
"""Tests that update entity is correct."""
assert len(hass.states.async_all("update")) == 0
await setup_platform(hass)
assert len(hass.states.async_all("update")) == 1
entity_id = "update.test_update"
state = hass.states.get(entity_id)
assert state.state == STATE_ON
assert state.attributes.get(ATTR_IN_PROGRESS) is False
with patch(
"homeassistant.components.tessie.update.schedule_software_update"
) as mock_update:
await hass.services.async_call(
UPDATE_DOMAIN,
SERVICE_INSTALL,
{ATTR_ENTITY_ID: [entity_id]},
blocking=True,
)
mock_update.assert_called_once()
state = hass.states.get(entity_id)
assert state.state == STATE_ON
assert state.attributes.get(ATTR_IN_PROGRESS) == 1