83 lines
3.2 KiB
Python
83 lines
3.2 KiB
Python
"""Fixtures for Asuswrt component."""
|
|
|
|
from unittest.mock import Mock, patch
|
|
|
|
from aioasuswrt.asuswrt import AsusWrt as AsusWrtLegacy
|
|
from aioasuswrt.connection import TelnetConnection
|
|
import pytest
|
|
|
|
from .common import ASUSWRT_BASE, MOCK_MACS, ROUTER_MAC_ADDR, new_device
|
|
|
|
ASUSWRT_LEGACY_LIB = f"{ASUSWRT_BASE}.bridge.AsusWrtLegacy"
|
|
|
|
MOCK_BYTES_TOTAL = [60000000000, 50000000000]
|
|
MOCK_CURRENT_TRANSFER_RATES = [20000000, 10000000]
|
|
MOCK_LOAD_AVG = [1.1, 1.2, 1.3]
|
|
MOCK_TEMPERATURES = {"2.4GHz": 40.2, "CPU": 71.2}
|
|
|
|
|
|
@pytest.fixture(name="patch_setup_entry")
|
|
def mock_controller_patch_setup_entry():
|
|
"""Mock setting up a config entry."""
|
|
with patch(
|
|
f"{ASUSWRT_BASE}.async_setup_entry", return_value=True
|
|
) as setup_entry_mock:
|
|
yield setup_entry_mock
|
|
|
|
|
|
@pytest.fixture(name="mock_devices_legacy")
|
|
def mock_devices_legacy_fixture():
|
|
"""Mock a list of devices."""
|
|
return {
|
|
MOCK_MACS[0]: new_device(MOCK_MACS[0], "192.168.1.2", "Test"),
|
|
MOCK_MACS[1]: new_device(MOCK_MACS[1], "192.168.1.3", "TestTwo"),
|
|
}
|
|
|
|
|
|
@pytest.fixture(name="mock_available_temps")
|
|
def mock_available_temps_fixture():
|
|
"""Mock a list of available temperature sensors."""
|
|
return [True, False, True]
|
|
|
|
|
|
@pytest.fixture(name="connect_legacy")
|
|
def mock_controller_connect_legacy(mock_devices_legacy, mock_available_temps):
|
|
"""Mock a successful connection with legacy library."""
|
|
with patch(ASUSWRT_LEGACY_LIB, spec=AsusWrtLegacy) as service_mock:
|
|
service_mock.return_value.connection = Mock(spec=TelnetConnection)
|
|
service_mock.return_value.is_connected = True
|
|
service_mock.return_value.async_get_nvram.return_value = {
|
|
"label_mac": ROUTER_MAC_ADDR,
|
|
"model": "abcd",
|
|
"firmver": "efg",
|
|
"buildno": "123",
|
|
}
|
|
service_mock.return_value.async_get_connected_devices.return_value = (
|
|
mock_devices_legacy
|
|
)
|
|
service_mock.return_value.async_get_bytes_total.return_value = MOCK_BYTES_TOTAL
|
|
service_mock.return_value.async_get_current_transfer_rates.return_value = (
|
|
MOCK_CURRENT_TRANSFER_RATES
|
|
)
|
|
service_mock.return_value.async_get_loadavg.return_value = MOCK_LOAD_AVG
|
|
service_mock.return_value.async_get_temperature.return_value = MOCK_TEMPERATURES
|
|
service_mock.return_value.async_find_temperature_commands.return_value = (
|
|
mock_available_temps
|
|
)
|
|
yield service_mock
|
|
|
|
|
|
@pytest.fixture(name="connect_legacy_sens_fail")
|
|
def mock_controller_connect_legacy_sens_fail(connect_legacy):
|
|
"""Mock a successful connection using legacy library with sensors fail."""
|
|
connect_legacy.return_value.async_get_nvram.side_effect = OSError
|
|
connect_legacy.return_value.async_get_connected_devices.side_effect = OSError
|
|
connect_legacy.return_value.async_get_bytes_total.side_effect = OSError
|
|
connect_legacy.return_value.async_get_current_transfer_rates.side_effect = OSError
|
|
connect_legacy.return_value.async_get_loadavg.side_effect = OSError
|
|
connect_legacy.return_value.async_get_temperature.side_effect = OSError
|
|
connect_legacy.return_value.async_find_temperature_commands.return_value = [
|
|
True,
|
|
True,
|
|
True,
|
|
]
|