From 70899dfe22f205b82e6f01a30cf56aaafaa5c3b5 Mon Sep 17 00:00:00 2001 From: Bas Nijholt Date: Sun, 6 Nov 2022 15:11:17 -0800 Subject: [PATCH] Remove area_id from entity_registry.async_get_or_create, closes #356 --- tests/test_switch.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_switch.py b/tests/test_switch.py index 4739855e..312cf7c6 100644 --- a/tests/test_switch.py +++ b/tests/test_switch.py @@ -878,7 +878,10 @@ async def test_area(hass): area_registry.async_create("test_area") entity = entity_registry.async_get(hass).async_get_or_create( - LIGHT_DOMAIN, "demo", light.unique_id, area_id="test_area" + LIGHT_DOMAIN, "demo", light.unique_id + ) + entity = entity_registry.async_get(hass).async_update_entity( + entity.entity_id, area_id="test_area" ) _LOGGER.debug("test_area entity: %s", entity) await hass.services.async_call(