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

Add restart device to UniFi button platform (#97642)

Browse source 
* Add restart device to UniFi Button platform

* Add tests for button platform

* Small corrections
This commit is contained in:
Robert Svensson 2023-08-05 20:07:20 +02:00 • committed by GitHub
parent 76c443777d
commit e43ad1c6a0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 210 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/components/unifi/test_controller.py
View file

@ -9,6 +9,7 @@ import aiounifi
from aiounifi.websocket import WebsocketState
import pytest
from homeassistant.components.button import DOMAIN as BUTTON_DOMAIN
from homeassistant.components.device_tracker import DOMAIN as TRACKER_DOMAIN
from homeassistant.components.image import DOMAIN as IMAGE_DOMAIN
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
@ -222,10 +223,11 @@ async def test_controller_setup(
entry = controller.config_entry
assert len(forward_entry_setup.mock_calls) == len(PLATFORMS)
assert forward_entry_setup.mock_calls[0][1] == (entry, TRACKER_DOMAIN)
assert forward_entry_setup.mock_calls[1][1] == (entry, IMAGE_DOMAIN)
assert forward_entry_setup.mock_calls[2][1] == (entry, SENSOR_DOMAIN)
assert forward_entry_setup.mock_calls[3][1] == (entry, SWITCH_DOMAIN)
assert forward_entry_setup.mock_calls[0][1] == (entry, BUTTON_DOMAIN)
assert forward_entry_setup.mock_calls[1][1] == (entry, TRACKER_DOMAIN)
assert forward_entry_setup.mock_calls[2][1] == (entry, IMAGE_DOMAIN)
assert forward_entry_setup.mock_calls[3][1] == (entry, SENSOR_DOMAIN)
assert forward_entry_setup.mock_calls[4][1] == (entry, SWITCH_DOMAIN)
assert controller.host == ENTRY_CONFIG[CONF_HOST]
assert controller.site == ENTRY_CONFIG[CONF_SITE_ID]