Files
core/tests/components/vicare/test_sensor.py
T

58 lines
2.2 KiB
Python

"""Test ViCare sensor entity."""
from unittest.mock import patch
import pytest
from syrupy.assertion import SnapshotAssertion
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from . import MODULE, setup_integration
from .conftest import Fixture, MockPyViCare
from tests.common import MockConfigEntry, snapshot_platform
@pytest.mark.usefixtures("entity_registry_enabled_by_default")
async def test_all_entities(
hass: HomeAssistant,
snapshot: SnapshotAssertion,
mock_config_entry: MockConfigEntry,
entity_registry: er.EntityRegistry,
) -> None:
"""Test all entities."""
fixtures: list[Fixture] = [
Fixture({"type:boiler"}, "vicare/Vitodens300W.json"),
Fixture({"type:heatpump"}, "vicare/Vitocal250A.json"),
Fixture({"type:heatpump"}, "vicare/Vitocal222G_Vitovent300W.json"),
Fixture({"type:ventilation"}, "vicare/ViAir300F.json"),
Fixture({"type:ventilation"}, "vicare/VitoPure.json"),
Fixture({"type:ess"}, "vicare/VitoChargeVX3.json"),
Fixture({None}, "vicare/VitoValor.json"),
Fixture({"type:climateSensor"}, "vicare/RoomSensor1.json"),
Fixture({"type:climateSensor"}, "vicare/RoomSensor2.json"),
Fixture({"type:radiator"}, "vicare/ZigbeeTRV.json"),
Fixture({"type:repeater"}, "vicare/ZigbeeRepeater.json"),
# FHT main and channel are the same physical zigbee node, so they share
# a gateway; this lets the channel link to the main via via_device.
Fixture({"type:fhtMain"}, "vicare/FHTMain.json", gateway_id="fht_gateway"),
Fixture(
{"type:fhtChannel"}, "vicare/FHTChannel.json", gateway_id="fht_gateway"
),
]
with (
patch(
"homeassistant.helpers.config_entry_oauth2_flow.OAuth2Session.async_ensure_token_valid",
),
patch(
f"{MODULE}._setup_vicare_api",
return_value=MockPyViCare(fixtures).as_vicare_data(),
),
patch(f"{MODULE}.PLATFORMS", [Platform.SENSOR]),
):
await setup_integration(hass, mock_config_entry)
await snapshot_platform(hass, entity_registry, snapshot, mock_config_entry.entry_id)