evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use SnapshotAssertion in SFR button tests (#89633)

Browse source
This commit is contained in:
epenet 2023-03-13 15:23:00 +01:00 • committed by GitHub
parent 15506da332
commit 07b25939a2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 93 additions and 103 deletions
Download patch file Download diff file Expand all files Collapse all files

25
tests/components/sfr_box/test_button.py
View file

@ -4,6 +4,7 @@ from unittest.mock import patch
import pytest
from sfrbox_api.exceptions import SFRBoxError
from syrupy.assertion import SnapshotAssertion
from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN, SERVICE_PRESS
from homeassistant.config_entries import ConfigEntry
@ -12,9 +13,6 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers import device_registry as dr, entity_registry as er
from . import check_device_registry, check_entities
from .const import EXPECTED_ENTITIES
pytestmark = pytest.mark.usefixtures("system_get_info", "dsl_get_info")
@ -32,17 +30,30 @@ async def test_buttons(
config_entry_with_auth: ConfigEntry,
device_registry: dr.DeviceRegistry,
entity_registry: er.EntityRegistry,
snapshot: SnapshotAssertion,
) -> None:
"""Test for SFR Box buttons."""
await hass.config_entries.async_setup(config_entry_with_auth.entry_id)
await hass.async_block_till_done()
check_device_registry(device_registry, EXPECTED_ENTITIES["expected_device"])
device_entries = dr.async_entries_for_config_entry(
device_registry, config_entry_with_auth.entry_id
)
assert device_entries == snapshot
expected_entities = EXPECTED_ENTITIES[Platform.BUTTON]
assert len(entity_registry.entities) == len(expected_entities)
entity_entries = er.async_entries_for_config_entry(
entity_registry, config_entry_with_auth.entry_id
)
assert entity_entries == snapshot
check_entities(hass, entity_registry, expected_entities)
for entity in entity_entries:
assert hass.states.get(entity.entity_id) == snapshot(name=entity.entity_id)
async def test_reboot(hass: HomeAssistant, config_entry_with_auth: ConfigEntry) -> None:
"""Test for SFR Box reboot button."""
await hass.config_entries.async_setup(config_entry_with_auth.entry_id)
await hass.async_block_till_done()
# Reboot success
service_data = {ATTR_ENTITY_ID: "button.sfr_box_reboot"}