35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
|
"""Tests for the Rova component."""
|
||
|
|
|
||
|
|
from unittest.mock import MagicMock
|
||
|
|
|
||
|
|
from syrupy import SnapshotAssertion
|
||
|
|
|
||
|
|
from homeassistant.const import Platform
|
||
|
|
from homeassistant.core import HomeAssistant
|
||
|
|
from homeassistant.helpers import entity_registry as er
|
||
|
|
|
||
|
|
from . import setup_with_selected_platforms
|
||
|
|
|
||
|
|
from tests.common import MockConfigEntry
|
||
|
|
|
||
|
|
|
||
|
|
async def test_all_entities(
|
||
|
|
hass: HomeAssistant,
|
||
|
|
snapshot: SnapshotAssertion,
|
||
|
|
mock_rova: MagicMock,
|
||
|
|
mock_config_entry: MockConfigEntry,
|
||
|
|
entity_registry: er.EntityRegistry,
|
||
|
|
) -> None:
|
||
|
|
"""Test all entities."""
|
||
|
|
await setup_with_selected_platforms(hass, mock_config_entry, [Platform.SENSOR])
|
||
|
|
entity_entries = er.async_entries_for_config_entry(
|
||
|
|
entity_registry, mock_config_entry.entry_id
|
||
|
|
)
|
||
|
|
|
||
|
|
assert entity_entries
|
||
|
|
for entity_entry in entity_entries:
|
||
|
|
assert hass.states.get(entity_entry.entity_id) == snapshot(
|
||
|
|
name=f"{entity_entry.entity_id}-state"
|
||
|
|
)
|
||
|
|
assert entity_entry == snapshot(name=f"{entity_entry.entity_id}-entry")
|