mirror of
https://github.com/home-assistant/core.git
synced 2025-04-29 19:57:52 +00:00
121 lines
4.5 KiB
Python
121 lines
4.5 KiB
Python
"""Test Mikrotik hub."""
|
|
from unittest.mock import patch
|
|
|
|
from homeassistant.components import mikrotik
|
|
|
|
from . import ARP_DATA, DHCP_DATA, MOCK_DATA, MOCK_OPTIONS, WIRELESS_DATA
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def setup_mikrotik_entry(hass, **kwargs):
|
|
"""Set up Mikrotik integration successfully."""
|
|
support_wireless = kwargs.get("support_wireless", True)
|
|
dhcp_data = kwargs.get("dhcp_data", DHCP_DATA)
|
|
wireless_data = kwargs.get("wireless_data", WIRELESS_DATA)
|
|
|
|
def mock_command(self, cmd, params=None):
|
|
if cmd == mikrotik.const.MIKROTIK_SERVICES[mikrotik.const.IS_WIRELESS]:
|
|
return support_wireless
|
|
if cmd == mikrotik.const.MIKROTIK_SERVICES[mikrotik.const.DHCP]:
|
|
return dhcp_data
|
|
if cmd == mikrotik.const.MIKROTIK_SERVICES[mikrotik.const.WIRELESS]:
|
|
return wireless_data
|
|
if cmd == mikrotik.const.MIKROTIK_SERVICES[mikrotik.const.ARP]:
|
|
return ARP_DATA
|
|
return {}
|
|
|
|
config_entry = MockConfigEntry(
|
|
domain=mikrotik.DOMAIN, data=MOCK_DATA, options=MOCK_OPTIONS
|
|
)
|
|
config_entry.add_to_hass(hass)
|
|
|
|
if "force_dhcp" in kwargs:
|
|
config_entry.options = {**config_entry.options, "force_dhcp": True}
|
|
|
|
if "arp_ping" in kwargs:
|
|
config_entry.options = {**config_entry.options, "arp_ping": True}
|
|
|
|
with patch("librouteros.connect"), patch.object(
|
|
mikrotik.hub.MikrotikData, "command", new=mock_command
|
|
):
|
|
await hass.config_entries.async_setup(config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
return hass.data[mikrotik.DOMAIN][config_entry.entry_id]
|
|
|
|
|
|
async def test_update_failed(hass):
|
|
"""Test failing to connect during update."""
|
|
|
|
hub = await setup_mikrotik_entry(hass)
|
|
|
|
with patch.object(
|
|
mikrotik.hub.MikrotikData, "command", side_effect=mikrotik.errors.CannotConnect
|
|
):
|
|
await hub.async_refresh()
|
|
|
|
assert not hub.last_update_success
|
|
|
|
|
|
async def test_hub_not_support_wireless(hass):
|
|
"""Test updating hub devices when hub doesn't support wireless interfaces."""
|
|
|
|
# test that the devices are constructed from dhcp data
|
|
|
|
hub = await setup_mikrotik_entry(hass, support_wireless=False)
|
|
|
|
assert hub.api.devices["00:00:00:00:00:01"]._params == DHCP_DATA[0]
|
|
assert hub.api.devices["00:00:00:00:00:01"]._wireless_params is None
|
|
assert hub.api.devices["00:00:00:00:00:02"]._params == DHCP_DATA[1]
|
|
assert hub.api.devices["00:00:00:00:00:02"]._wireless_params is None
|
|
|
|
|
|
async def test_hub_support_wireless(hass):
|
|
"""Test updating hub devices when hub support wireless interfaces."""
|
|
|
|
# test that the device list is from wireless data list
|
|
|
|
hub = await setup_mikrotik_entry(hass)
|
|
|
|
assert hub.api.support_wireless is True
|
|
assert hub.api.devices["00:00:00:00:00:01"]._params == DHCP_DATA[0]
|
|
assert hub.api.devices["00:00:00:00:00:01"]._wireless_params == WIRELESS_DATA[0]
|
|
|
|
# devices not in wireless list will not be added
|
|
assert "00:00:00:00:00:02" not in hub.api.devices
|
|
|
|
|
|
async def test_force_dhcp(hass):
|
|
"""Test updating hub devices with forced dhcp method."""
|
|
|
|
# test that the devices are constructed from dhcp data
|
|
|
|
hub = await setup_mikrotik_entry(hass, force_dhcp=True)
|
|
|
|
assert hub.api.support_wireless is True
|
|
assert hub.api.devices["00:00:00:00:00:01"]._params == DHCP_DATA[0]
|
|
assert hub.api.devices["00:00:00:00:00:01"]._wireless_params == WIRELESS_DATA[0]
|
|
|
|
# devices not in wireless list are added from dhcp
|
|
assert hub.api.devices["00:00:00:00:00:02"]._params == DHCP_DATA[1]
|
|
assert hub.api.devices["00:00:00:00:00:02"]._wireless_params is None
|
|
|
|
|
|
async def test_arp_ping(hass):
|
|
"""Test arp ping devices to confirm they are connected."""
|
|
|
|
# test device show as home if arp ping returns value
|
|
with patch.object(mikrotik.hub.MikrotikData, "do_arp_ping", return_value=True):
|
|
hub = await setup_mikrotik_entry(hass, arp_ping=True, force_dhcp=True)
|
|
|
|
assert hub.api.devices["00:00:00:00:00:01"].last_seen is not None
|
|
assert hub.api.devices["00:00:00:00:00:02"].last_seen is not None
|
|
|
|
# test device show as away if arp ping times out
|
|
with patch.object(mikrotik.hub.MikrotikData, "do_arp_ping", return_value=False):
|
|
hub = await setup_mikrotik_entry(hass, arp_ping=True, force_dhcp=True)
|
|
|
|
assert hub.api.devices["00:00:00:00:00:01"].last_seen is not None
|
|
# this device is not wireless so it will show as away
|
|
assert hub.api.devices["00:00:00:00:00:02"].last_seen is None
|