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

Update f-g* tests to use entity & device registry fixtures (#103841)

Browse source
This commit is contained in:
Jan-Philipp Benecke 2023-11-12 13:07:38 +01:00 • committed by GitHub
parent b9bc6ca070
commit 04a497343d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
56 changed files with 317 additions and 270 deletions
Download patch file Download diff file Expand all files Collapse all files

22
tests/components/freedompro/test_cover.py
View file

@ -36,6 +36,8 @@ from tests.common import async_fire_time_changed
)
async def test_cover_get_state(
hass: HomeAssistant,
entity_registry: er.EntityRegistry,
device_registry: dr.DeviceRegistry,
init_integration,
entity_id: str,
uid: str,
@ -44,10 +46,8 @@ async def test_cover_get_state(
) -> None:
"""Test states of the cover."""
init_integration
registry = er.async_get(hass)
registry_device = dr.async_get(hass)
device = registry_device.async_get_device(identifiers={("freedompro", uid)})
device = device_registry.async_get_device(identifiers={("freedompro", uid)})
assert device is not None
assert device.identifiers == {("freedompro", uid)}
assert device.manufacturer == "Freedompro"
@ -59,7 +59,7 @@ async def test_cover_get_state(
assert state.state == STATE_CLOSED
assert state.attributes.get("friendly_name") == name
entry = registry.async_get(entity_id)
entry = entity_registry.async_get(entity_id)
assert entry
assert entry.unique_id == uid
@ -76,7 +76,7 @@ async def test_cover_get_state(
assert state
assert state.attributes.get("friendly_name") == name
entry = registry.async_get(entity_id)
entry = entity_registry.async_get(entity_id)
assert entry
assert entry.unique_id == uid
@ -96,6 +96,7 @@ async def test_cover_get_state(
)
async def test_cover_set_position(
hass: HomeAssistant,
entity_registry: er.EntityRegistry,
init_integration,
entity_id: str,
uid: str,
@ -104,14 +105,13 @@ async def test_cover_set_position(
) -> None:
"""Test set position of the cover."""
init_integration
registry = er.async_get(hass)
state = hass.states.get(entity_id)
assert state
assert state.state == STATE_CLOSED
assert state.attributes.get("friendly_name") == name
entry = registry.async_get(entity_id)
entry = entity_registry.async_get(entity_id)
assert entry
assert entry.unique_id == uid
@ -151,6 +151,7 @@ async def test_cover_set_position(
)
async def test_cover_close(
hass: HomeAssistant,
entity_registry: er.EntityRegistry,
init_integration,
entity_id: str,
uid: str,
@ -159,7 +160,6 @@ async def test_cover_close(
) -> None:
"""Test close cover."""
init_integration
registry = er.async_get(hass)
states_response = get_states_response_for_uid(uid)
states_response[0]["state"]["position"] = 100
@ -176,7 +176,7 @@ async def test_cover_close(
assert state.state == STATE_OPEN
assert state.attributes.get("friendly_name") == name
entry = registry.async_get(entity_id)
entry = entity_registry.async_get(entity_id)
assert entry
assert entry.unique_id == uid
@ -214,6 +214,7 @@ async def test_cover_close(
)
async def test_cover_open(
hass: HomeAssistant,
entity_registry: er.EntityRegistry,
init_integration,
entity_id: str,
uid: str,
@ -222,14 +223,13 @@ async def test_cover_open(
) -> None:
"""Test open cover."""
init_integration
registry = er.async_get(hass)
state = hass.states.get(entity_id)
assert state
assert state.state == STATE_CLOSED
assert state.attributes.get("friendly_name") == name
entry = registry.async_get(entity_id)
entry = entity_registry.async_get(entity_id)
assert entry
assert entry.unique_id == uid