Use snapshot test helper in GIOS (#115893)

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

View File

@ -17,7 +17,7 @@ from homeassistant.util.dt import utcnow
from . import init_integration
from tests.common import async_fire_time_changed, load_fixture
from tests.common import async_fire_time_changed, load_fixture, snapshot_platform
async def test_sensor(
@ -27,13 +27,7 @@ async def test_sensor(
with patch("homeassistant.components.gios.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_availability(hass: HomeAssistant) -> None: