mirror of
https://github.com/home-assistant/core.git
synced 2025-11-12 20:40:18 +00:00
Use registry fixtures in tests (s) (#118295)
This commit is contained in:
@@ -115,17 +115,15 @@ def _migration_connect(*args, **kwargs):
|
||||
)
|
||||
async def test_async_migrate_entries(
|
||||
hass: HomeAssistant,
|
||||
device_registry: dr.DeviceRegistry,
|
||||
entity_registry: er.EntityRegistry,
|
||||
mock_config_entry: MockConfigEntry,
|
||||
entity_def: dict,
|
||||
ent_data: EntityMigrationData,
|
||||
) -> None:
|
||||
"""Test migration to new entity names."""
|
||||
|
||||
mock_config_entry.add_to_hass(hass)
|
||||
|
||||
entity_registry = er.async_get(hass)
|
||||
device_registry = dr.async_get(hass)
|
||||
|
||||
device: dr.DeviceEntry = device_registry.async_get_or_create(
|
||||
config_entry_id=mock_config_entry.entry_id,
|
||||
connections={(dr.CONNECTION_NETWORK_MAC, MOCK_ADAPTER_MAC)},
|
||||
@@ -181,15 +179,13 @@ async def test_async_migrate_entries(
|
||||
|
||||
async def test_entity_migration_data(
|
||||
hass: HomeAssistant,
|
||||
device_registry: dr.DeviceRegistry,
|
||||
entity_registry: er.EntityRegistry,
|
||||
mock_config_entry: MockConfigEntry,
|
||||
) -> None:
|
||||
"""Test ENTITY_MIGRATION data guards."""
|
||||
|
||||
mock_config_entry.add_to_hass(hass)
|
||||
|
||||
entity_registry = er.async_get(hass)
|
||||
device_registry = dr.async_get(hass)
|
||||
|
||||
device: dr.DeviceEntry = device_registry.async_get_or_create(
|
||||
config_entry_id=mock_config_entry.entry_id,
|
||||
connections={(dr.CONNECTION_NETWORK_MAC, MOCK_ADAPTER_MAC)},
|
||||
|
||||
Reference in New Issue
Block a user