From fb1bb0d37473df4008ec6dce7a22c49bd34288a0 Mon Sep 17 00:00:00 2001
From: Erik Montnemery <erik@montnemery.com>
Date: Thu, 10 Aug 2023 18:24:39 +0200
Subject: [PATCH] Adjust switch_as_x tests which create devices (#98210)
---
tests/components/switch_as_x/test_init.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/tests/components/switch_as_x/test_init.py b/tests/components/switch_as_x/test_init.py
index fac744d0c0e..a0c0bfca825 100644
--- a/tests/components/switch_as_x/test_init.py
+++ b/tests/components/switch_as_x/test_init.py
@@ -143,6 +143,7 @@ async def test_device_registry_config_entry_1(
entity_registry = er.async_get(hass)
switch_config_entry = MockConfigEntry()
+ switch_config_entry.add_to_hass(hass)
device_entry = device_registry.async_get_or_create(
config_entry_id=switch_config_entry.entry_id,
@@ -170,7 +171,6 @@ async def test_device_registry_config_entry_1(
},
title="ABC",
)
-
switch_as_x_config_entry.add_to_hass(hass)
assert await hass.config_entries.async_setup(switch_as_x_config_entry.entry_id)
@@ -202,6 +202,7 @@ async def test_device_registry_config_entry_2(
entity_registry = er.async_get(hass)
switch_config_entry = MockConfigEntry()
+ switch_config_entry.add_to_hass(hass)
device_entry = device_registry.async_get_or_create(
config_entry_id=switch_config_entry.entry_id,
@@ -313,6 +314,7 @@ async def test_device(hass: HomeAssistant, target_domain: Platform) -> None:
entity_registry = er.async_get(hass)
test_config_entry = MockConfigEntry()
+ test_config_entry.add_to_hass(hass)
device_entry = device_registry.async_get_or_create(
config_entry_id=test_config_entry.entry_id,
@@ -504,6 +506,7 @@ async def test_entity_name(
device_registry = dr.async_get(hass)
switch_config_entry = MockConfigEntry()
+ switch_config_entry.add_to_hass(hass)
device_entry = device_registry.async_get_or_create(
config_entry_id=switch_config_entry.entry_id,
@@ -559,6 +562,7 @@ async def test_custom_name_1(
device_registry = dr.async_get(hass)
switch_config_entry = MockConfigEntry()
+ switch_config_entry.add_to_hass(hass)
device_entry = device_registry.async_get_or_create(
config_entry_id=switch_config_entry.entry_id,
@@ -622,6 +626,7 @@ async def test_custom_name_2(
device_registry = dr.async_get(hass)
switch_config_entry = MockConfigEntry()
+ switch_config_entry.add_to_hass(hass)
device_entry = device_registry.async_get_or_create(
config_entry_id=switch_config_entry.entry_id,
--
GitLab