Use snapshot test helper in NAM (#115894)

Co-authored-by: Maciej Bieniek <478555+bieniu@users.noreply.github.com>
pull/115904/head
Maciej Bieniek 2024-04-20 18:43:33 +02:00 committed by GitHub
parent de1312f7e4
commit 5e345b7129
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 8 deletions

View File

@ -24,7 +24,11 @@ from homeassistant.util.dt import utcnow
from . import INCOMPLETE_NAM_DATA, init_integration
from tests.common import async_fire_time_changed, load_json_object_fixture
from tests.common import (
async_fire_time_changed,
load_json_object_fixture,
snapshot_platform,
)
async def test_sensor(
@ -41,13 +45,7 @@ async def test_sensor(
with patch("homeassistant.components.nam.PLATFORMS", [Platform.SENSOR]):
entry = await init_integration(hass)
entity_entries = er.async_entries_for_config_entry(entity_registry, entry.entry_id)
assert entity_entries
for entity_entry in entity_entries:
assert entity_entry == snapshot(name=f"{entity_entry.entity_id}-entry")
assert (state := hass.states.get(entity_entry.entity_id))
assert state == snapshot(name=f"{entity_entry.entity_id}-state")
await snapshot_platform(hass, entity_registry, snapshot, entry.entry_id)
async def test_sensor_disabled(