evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Do not access hass.data in unifi test (#75348)

Browse source 
* Do not access hass.data in test

* Process review feedback
This commit is contained in:
Jelte Zeilstra 2022-07-23 12:42:43 +02:00 • committed by GitHub
parent b60a59270c
commit 5e10716dd8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/components/unifi/test_update.py
View file

@ -5,7 +5,7 @@ from aiounifi.controller import MESSAGE_DEVICE
from aiounifi.websocket import STATE_DISCONNECTED, STATE_RUNNING from aiounifi.websocket import STATE_DISCONNECTED, STATE_RUNNING
from yarl import URL 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 ( from homeassistant.components.update import (
ATTR_IN_PROGRESS, ATTR_IN_PROGRESS,
ATTR_INSTALLED_VERSION, ATTR_INSTALLED_VERSION,
@ -19,6 +19,7 @@ from homeassistant.const import (
ATTR_DEVICE_CLASS, ATTR_DEVICE_CLASS,
ATTR_ENTITY_ID, ATTR_ENTITY_ID,
ATTR_SUPPORTED_FEATURES, ATTR_SUPPORTED_FEATURES,
CONF_HOST,
STATE_OFF, STATE_OFF,
STATE_ON, STATE_ON,
STATE_UNAVAILABLE, STATE_UNAVAILABLE,
@ -166,8 +167,7 @@ async def test_install(hass, aioclient_mock):
device_state = hass.states.get("update.device_1") device_state = hass.states.get("update.device_1")
assert device_state.state == STATE_ON assert device_state.state == STATE_ON
controller = hass.data[UNIFI_DOMAIN][config_entry.entry_id] url = f"https://{config_entry.data[CONF_HOST]}:1234/api/s/{config_entry.data[CONF_SITE_ID]}/cmd/devmgr"
url = f"https://{controller.host}:1234/api/s/{controller.site}/cmd/devmgr"
aioclient_mock.clear_requests() aioclient_mock.clear_requests()
aioclient_mock.post(url) aioclient_mock.post(url)