115 lines
3.8 KiB
Python
115 lines
3.8 KiB
Python
"""The tests for the ASUSWRT device tracker platform."""
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.components.asuswrt import (
|
|
CONF_DNSMASQ,
|
|
CONF_INTERFACE,
|
|
DATA_ASUSWRT,
|
|
DOMAIN,
|
|
)
|
|
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.common import mock_coro_func
|
|
|
|
|
|
async def test_password_or_pub_key_required(hass):
|
|
"""Test creating an AsusWRT scanner without a pass or pubkey."""
|
|
with patch("homeassistant.components.asuswrt.AsusWrt") as AsusWrt:
|
|
AsusWrt().connection.async_connect = mock_coro_func()
|
|
AsusWrt().is_connected = False
|
|
result = await async_setup_component(
|
|
hass,
|
|
DOMAIN,
|
|
{DOMAIN: {CONF_HOST: "fake_host", CONF_USERNAME: "fake_user"}},
|
|
)
|
|
assert not result
|
|
|
|
|
|
async def test_get_scanner_with_password_no_pubkey(hass):
|
|
"""Test creating an AsusWRT scanner with a password and no pubkey."""
|
|
with patch("homeassistant.components.asuswrt.AsusWrt") as AsusWrt:
|
|
AsusWrt().connection.async_connect = mock_coro_func()
|
|
AsusWrt().connection.async_get_connected_devices = mock_coro_func(
|
|
return_value={}
|
|
)
|
|
result = await async_setup_component(
|
|
hass,
|
|
DOMAIN,
|
|
{
|
|
DOMAIN: {
|
|
CONF_HOST: "fake_host",
|
|
CONF_USERNAME: "fake_user",
|
|
CONF_PASSWORD: "4321",
|
|
CONF_DNSMASQ: "/",
|
|
}
|
|
},
|
|
)
|
|
assert result
|
|
assert hass.data[DATA_ASUSWRT] is not None
|
|
|
|
|
|
async def test_specify_non_directory_path_for_dnsmasq(hass):
|
|
"""Test creating an AsusWRT scanner with a dnsmasq location which is not a valid directory."""
|
|
with patch("homeassistant.components.asuswrt.AsusWrt") as AsusWrt:
|
|
AsusWrt().connection.async_connect = mock_coro_func()
|
|
AsusWrt().is_connected = False
|
|
result = await async_setup_component(
|
|
hass,
|
|
DOMAIN,
|
|
{
|
|
DOMAIN: {
|
|
CONF_HOST: "fake_host",
|
|
CONF_USERNAME: "fake_user",
|
|
CONF_PASSWORD: "4321",
|
|
CONF_DNSMASQ: "?non_directory?",
|
|
}
|
|
},
|
|
)
|
|
assert not result
|
|
|
|
|
|
async def test_interface(hass):
|
|
"""Test creating an AsusWRT scanner using interface eth1."""
|
|
with patch("homeassistant.components.asuswrt.AsusWrt") as AsusWrt:
|
|
AsusWrt().connection.async_connect = mock_coro_func()
|
|
AsusWrt().connection.async_get_connected_devices = mock_coro_func(
|
|
return_value={}
|
|
)
|
|
result = await async_setup_component(
|
|
hass,
|
|
DOMAIN,
|
|
{
|
|
DOMAIN: {
|
|
CONF_HOST: "fake_host",
|
|
CONF_USERNAME: "fake_user",
|
|
CONF_PASSWORD: "4321",
|
|
CONF_DNSMASQ: "/",
|
|
CONF_INTERFACE: "eth1",
|
|
}
|
|
},
|
|
)
|
|
assert result
|
|
assert hass.data[DATA_ASUSWRT] is not None
|
|
|
|
|
|
async def test_no_interface(hass):
|
|
"""Test creating an AsusWRT scanner using no interface."""
|
|
with patch("homeassistant.components.asuswrt.AsusWrt") as AsusWrt:
|
|
AsusWrt().connection.async_connect = mock_coro_func()
|
|
AsusWrt().is_connected = False
|
|
result = await async_setup_component(
|
|
hass,
|
|
DOMAIN,
|
|
{
|
|
DOMAIN: {
|
|
CONF_HOST: "fake_host",
|
|
CONF_USERNAME: "fake_user",
|
|
CONF_PASSWORD: "4321",
|
|
CONF_DNSMASQ: "/",
|
|
CONF_INTERFACE: None,
|
|
}
|
|
},
|
|
)
|
|
assert not result
|