mirror of https://github.com/home-assistant/core
108 lines
3.2 KiB
Python
108 lines
3.2 KiB
Python
"""Tests for Fritz!Tools update platform."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from syrupy.assertion import SnapshotAssertion
|
|
|
|
from homeassistant.components.fritz.const import DOMAIN
|
|
from homeassistant.const import Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from .const import (
|
|
MOCK_FB_SERVICES,
|
|
MOCK_FIRMWARE_AVAILABLE,
|
|
MOCK_FIRMWARE_RELEASE_URL,
|
|
MOCK_USER_DATA,
|
|
)
|
|
|
|
from tests.common import MockConfigEntry, snapshot_platform
|
|
|
|
AVAILABLE_UPDATE = {
|
|
"UserInterface1": {
|
|
"GetInfo": {
|
|
"NewX_AVM-DE_Version": MOCK_FIRMWARE_AVAILABLE,
|
|
"NewX_AVM-DE_InfoURL": MOCK_FIRMWARE_RELEASE_URL,
|
|
},
|
|
}
|
|
}
|
|
|
|
|
|
@pytest.mark.usefixtures("entity_registry_enabled_by_default")
|
|
async def test_update_entities_initialized(
|
|
hass: HomeAssistant,
|
|
entity_registry: er.EntityRegistry,
|
|
fc_class_mock,
|
|
fh_class_mock,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test update entities."""
|
|
|
|
entry = MockConfigEntry(domain=DOMAIN, data=MOCK_USER_DATA)
|
|
entry.add_to_hass(hass)
|
|
|
|
with patch("homeassistant.components.fritz.PLATFORMS", [Platform.UPDATE]):
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
await snapshot_platform(hass, entity_registry, snapshot, entry.entry_id)
|
|
|
|
|
|
@pytest.mark.usefixtures("entity_registry_enabled_by_default")
|
|
async def test_update_available(
|
|
hass: HomeAssistant,
|
|
entity_registry: er.EntityRegistry,
|
|
fc_class_mock,
|
|
fh_class_mock,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test update entities."""
|
|
|
|
fc_class_mock().override_services({**MOCK_FB_SERVICES, **AVAILABLE_UPDATE})
|
|
|
|
entry = MockConfigEntry(domain=DOMAIN, data=MOCK_USER_DATA)
|
|
entry.add_to_hass(hass)
|
|
|
|
with patch("homeassistant.components.fritz.PLATFORMS", [Platform.UPDATE]):
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
await snapshot_platform(hass, entity_registry, snapshot, entry.entry_id)
|
|
|
|
|
|
@pytest.mark.usefixtures("entity_registry_enabled_by_default")
|
|
async def test_available_update_can_be_installed(
|
|
hass: HomeAssistant,
|
|
entity_registry: er.EntityRegistry,
|
|
fc_class_mock,
|
|
fh_class_mock,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test update entities."""
|
|
|
|
fc_class_mock().override_services({**MOCK_FB_SERVICES, **AVAILABLE_UPDATE})
|
|
|
|
with (
|
|
patch(
|
|
"homeassistant.components.fritz.coordinator.FritzBoxTools.async_trigger_firmware_update",
|
|
return_value=True,
|
|
) as mocked_update_call,
|
|
patch("homeassistant.components.fritz.PLATFORMS", [Platform.UPDATE]),
|
|
):
|
|
entry = MockConfigEntry(domain=DOMAIN, data=MOCK_USER_DATA)
|
|
entry.add_to_hass(hass)
|
|
|
|
assert await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
await snapshot_platform(hass, entity_registry, snapshot, entry.entry_id)
|
|
|
|
await hass.services.async_call(
|
|
"update",
|
|
"install",
|
|
{"entity_id": "update.mock_title_fritz_os"},
|
|
blocking=True,
|
|
)
|
|
mocked_update_call.assert_called_once()
|