112 lines
3.7 KiB
Python
112 lines
3.7 KiB
Python
"""Tests for Fritz!Tools update platform."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from aiohttp import ClientSession
|
|
|
|
from homeassistant.components.fritz.const import DOMAIN
|
|
from homeassistant.components.update import DOMAIN as UPDATE_DOMAIN
|
|
from homeassistant.config_entries import ConfigEntryState
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from .const import (
|
|
MOCK_FIRMWARE,
|
|
MOCK_FIRMWARE_AVAILABLE,
|
|
MOCK_FIRMWARE_RELEASE_URL,
|
|
MOCK_USER_DATA,
|
|
)
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_update_entities_initialized(
|
|
hass: HomeAssistant, hass_client: ClientSession, fc_class_mock, fh_class_mock
|
|
):
|
|
"""Test update entities."""
|
|
|
|
entry = MockConfigEntry(domain=DOMAIN, data=MOCK_USER_DATA)
|
|
entry.add_to_hass(hass)
|
|
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
assert entry.state == ConfigEntryState.LOADED
|
|
|
|
updates = hass.states.async_all(UPDATE_DOMAIN)
|
|
assert len(updates) == 1
|
|
|
|
|
|
async def test_update_available(
|
|
hass: HomeAssistant, hass_client: ClientSession, fc_class_mock, fh_class_mock
|
|
):
|
|
"""Test update entities."""
|
|
|
|
with patch(
|
|
"homeassistant.components.fritz.common.FritzBoxTools._update_device_info",
|
|
return_value=(True, MOCK_FIRMWARE_AVAILABLE, MOCK_FIRMWARE_RELEASE_URL),
|
|
):
|
|
entry = MockConfigEntry(domain=DOMAIN, data=MOCK_USER_DATA)
|
|
entry.add_to_hass(hass)
|
|
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
assert entry.state == ConfigEntryState.LOADED
|
|
|
|
update = hass.states.get("update.mock_title_fritz_os")
|
|
assert update is not None
|
|
assert update.state == "on"
|
|
assert update.attributes.get("installed_version") == MOCK_FIRMWARE
|
|
assert update.attributes.get("latest_version") == MOCK_FIRMWARE_AVAILABLE
|
|
assert update.attributes.get("release_url") == MOCK_FIRMWARE_RELEASE_URL
|
|
|
|
|
|
async def test_no_update_available(
|
|
hass: HomeAssistant, hass_client: ClientSession, fc_class_mock, fh_class_mock
|
|
):
|
|
"""Test update entities."""
|
|
|
|
entry = MockConfigEntry(domain=DOMAIN, data=MOCK_USER_DATA)
|
|
entry.add_to_hass(hass)
|
|
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
assert entry.state == ConfigEntryState.LOADED
|
|
|
|
update = hass.states.get("update.mock_title_fritz_os")
|
|
assert update is not None
|
|
assert update.state == "off"
|
|
assert update.attributes.get("installed_version") == MOCK_FIRMWARE
|
|
assert update.attributes.get("latest_version") == MOCK_FIRMWARE
|
|
|
|
|
|
async def test_available_update_can_be_installed(
|
|
hass: HomeAssistant, hass_client: ClientSession, fc_class_mock, fh_class_mock
|
|
):
|
|
"""Test update entities."""
|
|
|
|
with patch(
|
|
"homeassistant.components.fritz.common.FritzBoxTools._update_device_info",
|
|
return_value=(True, MOCK_FIRMWARE_AVAILABLE, MOCK_FIRMWARE_RELEASE_URL),
|
|
), patch(
|
|
"homeassistant.components.fritz.common.FritzBoxTools.async_trigger_firmware_update",
|
|
return_value=True,
|
|
) as mocked_update_call:
|
|
entry = MockConfigEntry(domain=DOMAIN, data=MOCK_USER_DATA)
|
|
entry.add_to_hass(hass)
|
|
|
|
assert await async_setup_component(hass, DOMAIN, {})
|
|
await hass.async_block_till_done()
|
|
assert entry.state == ConfigEntryState.LOADED
|
|
|
|
update = hass.states.get("update.mock_title_fritz_os")
|
|
assert update is not None
|
|
assert update.state == "on"
|
|
|
|
assert await hass.services.async_call(
|
|
"update",
|
|
"install",
|
|
{"entity_id": "update.mock_title_fritz_os"},
|
|
blocking=True,
|
|
)
|
|
assert mocked_update_call.assert_called_once
|