From 5e10716dd84fe5cb227368e7f0e42b0b2c3a66ea Mon Sep 17 00:00:00 2001 From: Jelte Zeilstra Date: Sat, 23 Jul 2022 12:42:43 +0200 Subject: [PATCH] Do not access hass.data in unifi test (#75348) * Do not access hass.data in test * Process review feedback --- tests/components/unifi/test_update.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/components/unifi/test_update.py b/tests/components/unifi/test_update.py index 7bbeb65497e..677491319a1 100644 --- a/tests/components/unifi/test_update.py +++ b/tests/components/unifi/test_update.py @@ -5,7 +5,7 @@ from aiounifi.controller import MESSAGE_DEVICE from aiounifi.websocket import STATE_DISCONNECTED, STATE_RUNNING from yarl import URL -from homeassistant.components.unifi.const import DOMAIN as UNIFI_DOMAIN +from homeassistant.components.unifi.const import CONF_SITE_ID from homeassistant.components.update import ( ATTR_IN_PROGRESS, ATTR_INSTALLED_VERSION, @@ -19,6 +19,7 @@ from homeassistant.const import ( ATTR_DEVICE_CLASS, ATTR_ENTITY_ID, ATTR_SUPPORTED_FEATURES, + CONF_HOST, STATE_OFF, STATE_ON, STATE_UNAVAILABLE, @@ -166,8 +167,7 @@ async def test_install(hass, aioclient_mock): device_state = hass.states.get("update.device_1") assert device_state.state == STATE_ON - controller = hass.data[UNIFI_DOMAIN][config_entry.entry_id] - url = f"https://{controller.host}:1234/api/s/{controller.site}/cmd/devmgr" + url = f"https://{config_entry.data[CONF_HOST]}:1234/api/s/{config_entry.data[CONF_SITE_ID]}/cmd/devmgr" aioclient_mock.clear_requests() aioclient_mock.post(url)