core/tests/components/cookidoo/test_sensor.py
Cyrill Raccaud 33a23ad1c6
Add diagnostic sensors for the active subscription of Cookidoo (#136485)
* add diagnostics for the active subcription

* fix mapping between api and ha states for subscription

* multiline lambda
2025-01-27 08:43:30 +01:00

45 lines
1.3 KiB
Python

"""Test for sensor platform of the Cookidoo integration."""
from collections.abc import Generator
from unittest.mock import patch
import pytest
from syrupy.assertion import SnapshotAssertion
from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er
from tests.common import MockConfigEntry, snapshot_platform
@pytest.fixture(autouse=True)
def sensor_only() -> Generator[None]:
"""Enable only the sensor platform."""
with patch(
"homeassistant.components.cookidoo.PLATFORMS",
[Platform.SENSOR],
):
yield
@pytest.mark.usefixtures("mock_cookidoo_client")
async def test_setup(
hass: HomeAssistant,
cookidoo_config_entry: MockConfigEntry,
snapshot: SnapshotAssertion,
entity_registry: er.EntityRegistry,
) -> None:
"""Snapshot test states of sensor platform."""
cookidoo_config_entry.add_to_hass(hass)
await hass.config_entries.async_setup(cookidoo_config_entry.entry_id)
await hass.async_block_till_done()
assert cookidoo_config_entry.state is ConfigEntryState.LOADED
await snapshot_platform(
hass, entity_registry, snapshot, cookidoo_config_entry.entry_id
)