98 lines
3.2 KiB
Python
98 lines
3.2 KiB
Python
"""Test UniFi setup process."""
|
|
from unittest.mock import Mock, patch
|
|
|
|
from homeassistant.components import unifi
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from .test_controller import setup_unifi_integration
|
|
|
|
from tests.common import MockConfigEntry, mock_coro
|
|
|
|
|
|
async def test_setup_with_no_config(hass):
|
|
"""Test that we do not discover anything or try to set up a bridge."""
|
|
assert await async_setup_component(hass, unifi.DOMAIN, {}) is True
|
|
assert unifi.DOMAIN not in hass.data
|
|
assert hass.data[unifi.UNIFI_CONFIG] == []
|
|
|
|
|
|
async def test_setup_with_config(hass):
|
|
"""Test that we do not discover anything or try to set up a bridge."""
|
|
config = {
|
|
unifi.DOMAIN: {
|
|
unifi.CONF_CONTROLLERS: {
|
|
unifi.CONF_HOST: "1.2.3.4",
|
|
unifi.CONF_SITE_ID: "My site",
|
|
unifi.CONF_BLOCK_CLIENT: ["12:34:56:78:90:AB"],
|
|
unifi.CONF_DETECTION_TIME: 3,
|
|
unifi.CONF_SSID_FILTER: ["ssid"],
|
|
}
|
|
}
|
|
}
|
|
assert await async_setup_component(hass, unifi.DOMAIN, config) is True
|
|
assert unifi.DOMAIN not in hass.data
|
|
assert hass.data[unifi.UNIFI_CONFIG] == [
|
|
{
|
|
unifi.CONF_HOST: "1.2.3.4",
|
|
unifi.CONF_SITE_ID: "My site",
|
|
unifi.CONF_BLOCK_CLIENT: ["12:34:56:78:90:AB"],
|
|
unifi.CONF_DETECTION_TIME: 3,
|
|
unifi.CONF_SSID_FILTER: ["ssid"],
|
|
}
|
|
]
|
|
|
|
|
|
async def test_successful_config_entry(hass):
|
|
"""Test that configured options for a host are loaded via config entry."""
|
|
await setup_unifi_integration(hass)
|
|
assert hass.data[unifi.DOMAIN]
|
|
|
|
|
|
async def test_controller_fail_setup(hass):
|
|
"""Test that a failed setup still stores controller."""
|
|
with patch.object(unifi, "UniFiController") as mock_cntrlr:
|
|
mock_cntrlr.return_value.async_setup.return_value = mock_coro(False)
|
|
await setup_unifi_integration(hass)
|
|
|
|
assert hass.data[unifi.DOMAIN] == {}
|
|
|
|
|
|
async def test_controller_no_mac(hass):
|
|
"""Test that configured options for a host are loaded via config entry."""
|
|
entry = MockConfigEntry(
|
|
domain=unifi.DOMAIN,
|
|
data={
|
|
"controller": {
|
|
"host": "0.0.0.0",
|
|
"username": "user",
|
|
"password": "pass",
|
|
"port": 80,
|
|
"site": "default",
|
|
"verify_ssl": True,
|
|
},
|
|
"poe_control": True,
|
|
},
|
|
)
|
|
entry.add_to_hass(hass)
|
|
mock_registry = Mock()
|
|
with patch.object(unifi, "UniFiController") as mock_controller, patch(
|
|
"homeassistant.helpers.device_registry.async_get_registry",
|
|
return_value=mock_coro(mock_registry),
|
|
):
|
|
mock_controller.return_value.async_setup.return_value = mock_coro(True)
|
|
mock_controller.return_value.mac = None
|
|
assert await unifi.async_setup_entry(hass, entry) is True
|
|
|
|
assert len(mock_controller.mock_calls) == 2
|
|
|
|
assert len(mock_registry.mock_calls) == 0
|
|
|
|
|
|
async def test_unload_entry(hass):
|
|
"""Test being able to unload an entry."""
|
|
controller = await setup_unifi_integration(hass)
|
|
assert hass.data[unifi.DOMAIN]
|
|
|
|
assert await unifi.async_unload_entry(hass, controller.config_entry)
|
|
assert not hass.data[unifi.DOMAIN]
|