jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-12 20:40:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use registry fixtures in tests (z) (#118300)

Browse Source
This commit is contained in:
epenet
2024-05-28 18:25:49 +02:00
committed by GitHub
parent 106cb4cfb7
commit 0b2aac8f4c
26 changed files with 526 additions and 372 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
tests/components/zwave_js/test_api.py
View File

@@ -126,6 +126,7 @@ async def test_no_driver(
async def test_network_status(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
multisensor_6,
controller_state,
client,
@@ -158,8 +159,7 @@ async def test_network_status(
assert result["controller"]["inclusion_state"] == InclusionState.IDLE
# Try API call with device ID
dev_reg = dr.async_get(hass)
device = dev_reg.async_get_device(
device = device_registry.async_get_device(
identifiers={(DOMAIN, "3245146787-52")},
)
assert device
@@ -251,6 +251,7 @@ async def test_network_status(
async def test_subscribe_node_status(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
multisensor_6_state,
client,
integration,
@@ -265,8 +266,7 @@ async def test_subscribe_node_status(
driver = client.driver
driver.controller.nodes[node.node_id] = node
dev_reg = dr.async_get(hass)
device = dev_reg.async_get_or_create(
device = device_registry.async_get_or_create(
config_entry_id=entry.entry_id, identifiers={get_device_id(driver, node)}
)
@@ -461,6 +461,7 @@ async def test_node_metadata(
async def test_node_alerts(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
wallmote_central_scene,
integration,
hass_ws_client: WebSocketGenerator,
@@ -468,8 +469,7 @@ async def test_node_alerts(
"""Test the node comments websocket command."""
ws_client = await hass_ws_client(hass)
dev_reg = dr.async_get(hass)
device = dev_reg.async_get_device(identifiers={(DOMAIN, "3245146787-35")})
device = device_registry.async_get_device(identifiers={(DOMAIN, "3245146787-35")})
assert device
await ws_client.send_json(
@@ -1650,6 +1650,7 @@ async def test_cancel_inclusion_exclusion(
async def test_remove_node(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
integration,
client,
hass_ws_client: WebSocketGenerator,
@@ -1686,10 +1687,8 @@ async def test_remove_node(
msg = await ws_client.receive_json()
assert msg["event"]["event"] == "exclusion started"
dev_reg = dr.async_get(hass)
# Create device registry entry for mock node
device = dev_reg.async_get_or_create(
device = device_registry.async_get_or_create(
config_entry_id=entry.entry_id,
identifiers={(DOMAIN, "3245146787-67")},
name="Node 67",
@@ -1701,7 +1700,7 @@ async def test_remove_node(
assert msg["event"]["event"] == "node removed"
# Verify device was removed from device registry
device = dev_reg.async_get_device(
device = device_registry.async_get_device(
identifiers={(DOMAIN, "3245146787-67")},
)
assert device is None
@@ -1761,6 +1760,7 @@ async def test_remove_node(
async def test_replace_failed_node(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
nortek_thermostat,
integration,
client,
@@ -1772,10 +1772,8 @@ async def test_replace_failed_node(
entry = integration
ws_client = await hass_ws_client(hass)
dev_reg = dr.async_get(hass)
# Create device registry entry for mock node
device = dev_reg.async_get_or_create(
device = device_registry.async_get_or_create(
config_entry_id=entry.entry_id,
identifiers={(DOMAIN, "3245146787-67")},
name="Node 67",
@@ -1871,7 +1869,7 @@ async def test_replace_failed_node(
# Verify device was removed from device registry
assert (
dev_reg.async_get_device(
device_registry.async_get_device(
identifiers={(DOMAIN, "3245146787-67")},
)
is None
@@ -2110,6 +2108,7 @@ async def test_replace_failed_node(
async def test_remove_failed_node(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
nortek_thermostat,
integration,
client,
@@ -2153,10 +2152,8 @@ async def test_remove_failed_node(
msg = await ws_client.receive_json()
assert msg["success"]
dev_reg = dr.async_get(hass)
# Create device registry entry for mock node
device = dev_reg.async_get_or_create(
device = device_registry.async_get_or_create(
config_entry_id=entry.entry_id,
identifiers={(DOMAIN, "3245146787-67")},
name="Node 67",
@@ -2169,7 +2166,7 @@ async def test_remove_failed_node(
# Verify device was removed from device registry
assert (
dev_reg.async_get_device(
device_registry.async_get_device(
identifiers={(DOMAIN, "3245146787-67")},
)
is None
@@ -4674,6 +4671,7 @@ async def test_subscribe_node_statistics(
async def test_hard_reset_controller(
hass: HomeAssistant,
device_registry: dr.DeviceRegistry,
client,
integration,
listen_block,
@@ -4683,8 +4681,7 @@ async def test_hard_reset_controller(
entry = integration
ws_client = await hass_ws_client(hass)
dev_reg = dr.async_get(hass)
device = dev_reg.async_get_device(
device = device_registry.async_get_device(
identifiers={get_device_id(client.driver, client.driver.controller.nodes[1])}
)