Use snapshot test helper in NAM (#115894)
Co-authored-by: Maciej Bieniek <478555+bieniu@users.noreply.github.com>pull/115904/head
parent
de1312f7e4
commit
5e345b7129
|
@ -24,7 +24,11 @@ from homeassistant.util.dt import utcnow
|
||||||
|
|
||||||
from . import INCOMPLETE_NAM_DATA, init_integration
|
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(
|
async def test_sensor(
|
||||||
|
@ -41,13 +45,7 @@ async def test_sensor(
|
||||||
with patch("homeassistant.components.nam.PLATFORMS", [Platform.SENSOR]):
|
with patch("homeassistant.components.nam.PLATFORMS", [Platform.SENSOR]):
|
||||||
entry = await init_integration(hass)
|
entry = await init_integration(hass)
|
||||||
|
|
||||||
entity_entries = er.async_entries_for_config_entry(entity_registry, entry.entry_id)
|
await snapshot_platform(hass, entity_registry, snapshot, 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")
|
|
||||||
|
|
||||||
|
|
||||||
async def test_sensor_disabled(
|
async def test_sensor_disabled(
|
||||||
|
|
Loading…
Reference in New Issue