Rename mikrotik coordinator module (#117488)
This commit is contained in:
parent
73ed49e4b7
commit
6c892b227b
7 changed files with 15 additions and 9 deletions
|
@ -800,7 +800,7 @@ omit =
|
|||
homeassistant/components/microbees/sensor.py
|
||||
homeassistant/components/microbees/switch.py
|
||||
homeassistant/components/microsoft/tts.py
|
||||
homeassistant/components/mikrotik/hub.py
|
||||
homeassistant/components/mikrotik/coordinator.py
|
||||
homeassistant/components/mill/climate.py
|
||||
homeassistant/components/mill/sensor.py
|
||||
homeassistant/components/minio/minio_helper.py
|
||||
|
|
|
@ -7,8 +7,8 @@ from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
|
|||
from homeassistant.helpers import config_validation as cv, device_registry as dr
|
||||
|
||||
from .const import ATTR_MANUFACTURER, DOMAIN
|
||||
from .coordinator import MikrotikDataUpdateCoordinator, get_api
|
||||
from .errors import CannotConnect, LoginError
|
||||
from .hub import MikrotikDataUpdateCoordinator, get_api
|
||||
|
||||
CONFIG_SCHEMA = cv.removed(DOMAIN, raise_if_present=False)
|
||||
|
||||
|
|
|
@ -31,8 +31,8 @@ from .const import (
|
|||
DEFAULT_NAME,
|
||||
DOMAIN,
|
||||
)
|
||||
from .coordinator import get_api
|
||||
from .errors import CannotConnect, LoginError
|
||||
from .hub import get_api
|
||||
|
||||
|
||||
class MikrotikFlowHandler(ConfigFlow, domain=DOMAIN):
|
||||
|
|
|
@ -17,7 +17,7 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
|||
import homeassistant.util.dt as dt_util
|
||||
|
||||
from .const import DOMAIN
|
||||
from .hub import Device, MikrotikDataUpdateCoordinator
|
||||
from .coordinator import Device, MikrotikDataUpdateCoordinator
|
||||
|
||||
|
||||
async def async_setup_entry(
|
||||
|
|
|
@ -210,7 +210,7 @@ async def setup_mikrotik_entry(hass: HomeAssistant, **kwargs: Any) -> None:
|
|||
|
||||
with (
|
||||
patch("librouteros.connect"),
|
||||
patch.object(mikrotik.hub.MikrotikData, "command", new=mock_command),
|
||||
patch.object(mikrotik.coordinator.MikrotikData, "command", new=mock_command),
|
||||
):
|
||||
await hass.config_entries.async_setup(config_entry.entry_id)
|
||||
await hass.async_block_till_done()
|
||||
|
|
|
@ -82,7 +82,7 @@ async def test_device_trackers(
|
|||
device_2 = hass.states.get("device_tracker.device_2")
|
||||
assert device_2 is None
|
||||
|
||||
with patch.object(mikrotik.hub.MikrotikData, "command", new=mock_command):
|
||||
with patch.object(mikrotik.coordinator.MikrotikData, "command", new=mock_command):
|
||||
# test device_2 is added after connecting to wireless network
|
||||
WIRELESS_DATA.append(DEVICE_2_WIRELESS)
|
||||
|
||||
|
@ -150,7 +150,9 @@ async def test_arp_ping_success(
|
|||
) -> None:
|
||||
"""Test arp ping devices to confirm they are connected."""
|
||||
|
||||
with patch.object(mikrotik.hub.MikrotikData, "do_arp_ping", return_value=True):
|
||||
with patch.object(
|
||||
mikrotik.coordinator.MikrotikData, "do_arp_ping", return_value=True
|
||||
):
|
||||
await setup_mikrotik_entry(hass, arp_ping=True, force_dhcp=True)
|
||||
|
||||
# test wired device_2 show as home if arp ping returns True
|
||||
|
@ -163,7 +165,9 @@ async def test_arp_ping_timeout(
|
|||
hass: HomeAssistant, mock_device_registry_devices
|
||||
) -> None:
|
||||
"""Test arp ping timeout so devices are shown away."""
|
||||
with patch.object(mikrotik.hub.MikrotikData, "do_arp_ping", return_value=False):
|
||||
with patch.object(
|
||||
mikrotik.coordinator.MikrotikData, "do_arp_ping", return_value=False
|
||||
):
|
||||
await setup_mikrotik_entry(hass, arp_ping=True, force_dhcp=True)
|
||||
|
||||
# test wired device_2 show as not_home if arp ping times out
|
||||
|
@ -262,7 +266,9 @@ async def test_update_failed(hass: HomeAssistant, mock_device_registry_devices)
|
|||
await setup_mikrotik_entry(hass)
|
||||
|
||||
with patch.object(
|
||||
mikrotik.hub.MikrotikData, "command", side_effect=mikrotik.errors.CannotConnect
|
||||
mikrotik.coordinator.MikrotikData,
|
||||
"command",
|
||||
side_effect=mikrotik.errors.CannotConnect,
|
||||
):
|
||||
async_fire_time_changed(hass, utcnow() + timedelta(seconds=10))
|
||||
await hass.async_block_till_done(wait_background_tasks=True)
|
||||
|
|
Loading…
Add table
Reference in a new issue