home-assistant
/
core
mirror of https://github.com/home-assistant/core
0
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
core/tests/components/qnap_qsw/util.py

565 lines
14 KiB
Python
Raw Permalink Blame History

"""Tests for the QNAP QSW integration."""
from unittest.mock import patch
from aioqsw.const import (
API_ANOMALY,
API_BUILD_NUMBER,
API_CHIP_ID,
API_CI_BRANCH,
API_CI_COMMIT,
API_CI_PIPELINE,
API_COMMIT_CPSS,
API_COMMIT_ISS,
API_DATE,
API_DESCRIPTION,
API_DOWNLOAD_URL,
API_ERROR_CODE,
API_ERROR_MESSAGE,
API_FAN1_SPEED,
API_FAN2_SPEED,
API_FCS_ERRORS,
API_FULL_DUPLEX,
API_KEY,
API_LINK,
API_MAC_ADDR,
API_MAX_PORT_CHANNELS,
API_MAX_PORTS_PER_PORT_CHANNEL,
API_MAX_SWITCH_TEMP,
API_MESSAGE,
API_MODEL,
API_NEWER,
API_NUMBER,
API_PORT_NUM,
API_PRODUCT,
API_PUB_DATE,
API_RESULT,
API_RX_ERRORS,
API_RX_OCTETS,
API_SERIAL,
API_SPEED,
API_START_INDEX,
API_SWITCH_TEMP,
API_TRUNK_NUM,
API_TX_OCTETS,
API_UPTIME,
API_VAL,
API_VERSION,
)
from homeassistant.components.qnap_qsw import DOMAIN
from homeassistant.const import CONF_PASSWORD, CONF_URL, CONF_USERNAME
from homeassistant.core import HomeAssistant
from tests.common import MockConfigEntry
CONFIG = {
CONF_URL: "http://192.168.1.100",
CONF_USERNAME: "username",
CONF_PASSWORD: "password",
}
LIVE_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: "None",
}
SYSTEM_BOARD_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: {
API_MAC_ADDR: "MAC",
API_SERIAL: "SERIAL",
API_CHIP_ID: "ALLEYCAT3",
API_MODEL: "M408",
API_PORT_NUM: 12,
API_PRODUCT: "QSW-M408-4C",
API_TRUNK_NUM: 0,
},
}
FIRMWARE_CONDITION_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: {
API_ANOMALY: False,
API_MESSAGE: "",
},
}
FIRMWARE_INFO_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: {
API_VERSION: "1.2.0",
API_DATE: "20220128",
API_PUB_DATE: "Fri, 28 Jan 2022 01:17:39 +0800",
API_BUILD_NUMBER: "20220128",
API_NUMBER: "29649",
API_CI_COMMIT: "b2eb4c8ffb549995aeb4f9c4e645c6d882997c17",
API_CI_BRANCH: "m408/codesigning",
API_CI_PIPELINE: "9898",
API_COMMIT_CPSS: "",
API_COMMIT_ISS: "448a3208e5ea744c393b2580f4b9733add9c2faa",
},
}
FIRMWARE_UPDATE_CHECK_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: {
API_VERSION: "1.3.0",
API_NUMBER: "527638",
API_BUILD_NUMBER: "20221123",
API_DATE: "Wed, 07, Dec 2022 16:30:00 +0800",
API_DESCRIPTION: "",
API_DOWNLOAD_URL: [
"https://download.qnap.com/Storage/Networking/QSW408FW/QSW-M408AC3-FW.v1.3.0_S20221123_527638.img",
"https://eu1.qnap.com/Storage/Networking/QSW408FW/QSW-M408AC3-FW.v1.3.0_S20221123_527638.img",
"https://us1.qnap.com/Storage/Networking/QSW408FW/QSW-M408AC3-FW.v1.3.0_S20221123_527638.img",
],
API_NEWER: True,
},
}
LACP_INFO_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: {
API_START_INDEX: 28,
API_MAX_PORT_CHANNELS: 8,
API_MAX_PORTS_PER_PORT_CHANNEL: 8,
},
}
PORTS_STATISTICS_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: [
{
API_KEY: "1",
API_VAL: {
API_RX_OCTETS: 20000,
API_RX_ERRORS: 20,
API_TX_OCTETS: 10000,
API_FCS_ERRORS: 10,
},
},
{
API_KEY: "2",
API_VAL: {
API_RX_OCTETS: 2000,
API_RX_ERRORS: 2,
API_TX_OCTETS: 1000,
API_FCS_ERRORS: 1,
},
},
{
API_KEY: "3",
API_VAL: {
API_RX_OCTETS: 200,
API_RX_ERRORS: 0,
API_TX_OCTETS: 100,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "4",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "5",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "6",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "7",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "8",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "9",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "10",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "11",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "12",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "29",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "30",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "31",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "32",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "33",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
{
API_KEY: "34",
API_VAL: {
API_RX_OCTETS: 0,
API_RX_ERRORS: 0,
API_TX_OCTETS: 0,
API_FCS_ERRORS: 0,
},
},
],
}
PORTS_STATUS_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: [
{
API_KEY: "1",
API_VAL: {
API_LINK: True,
API_FULL_DUPLEX: True,
API_SPEED: "10000",
},
},
{
API_KEY: "2",
API_VAL: {
API_LINK: True,
API_FULL_DUPLEX: True,
API_SPEED: "1000",
},
},
{
API_KEY: "3",
API_VAL: {
API_LINK: True,
API_FULL_DUPLEX: False,
API_SPEED: "100",
},
},
{
API_KEY: "4",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "1000",
},
},
{
API_KEY: "5",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "1000",
},
},
{
API_KEY: "6",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "1000",
},
},
{
API_KEY: "7",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "1000",
},
},
{
API_KEY: "8",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "1000",
},
},
{
API_KEY: "9",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "1000",
},
},
{
API_KEY: "10",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "1000",
},
},
{
API_KEY: "11",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "1000",
},
},
{
API_KEY: "12",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "1000",
},
},
{
API_KEY: "29",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "0",
},
},
{
API_KEY: "30",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "0",
},
},
{
API_KEY: "31",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "0",
},
},
{
API_KEY: "32",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "0",
},
},
{
API_KEY: "33",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "0",
},
},
{
API_KEY: "34",
API_VAL: {
API_LINK: False,
API_FULL_DUPLEX: False,
API_SPEED: "0",
},
},
],
}
SYSTEM_COMMAND_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: "None",
}
SYSTEM_SENSOR_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: {
API_FAN1_SPEED: 1991,
API_FAN2_SPEED: -2,
API_MAX_SWITCH_TEMP: 85,
API_SWITCH_TEMP: 31,
},
}
SYSTEM_TIME_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: {
API_UPTIME: 91,
},
}
USERS_LOGIN_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: "TOKEN",
}
USERS_VERIFICATION_MOCK = {
API_ERROR_CODE: 200,
API_ERROR_MESSAGE: "OK",
API_RESULT: "None",
}
def init_config_entry(
hass: HomeAssistant,
) -> MockConfigEntry:
"""Set up the QNAP QSW entry in Home Assistant."""
config_entry = MockConfigEntry(
data=CONFIG,
domain=DOMAIN,
unique_id="qsw_unique_id",
)
config_entry.add_to_hass(hass)
return config_entry
async def async_init_integration(
hass: HomeAssistant,
config_entry: MockConfigEntry | None = None,
) -> None:
"""Set up the QNAP QSW integration in Home Assistant."""
if config_entry is None:
config_entry = init_config_entry(hass)
with (
patch(
"homeassistant.components.qnap_qsw.QnapQswApi.get_firmware_condition",
return_value=FIRMWARE_CONDITION_MOCK,
),
patch(
"homeassistant.components.qnap_qsw.QnapQswApi.get_firmware_info",
return_value=FIRMWARE_INFO_MOCK,
),
patch(
"homeassistant.components.qnap_qsw.QnapQswApi.get_firmware_update_check",
return_value=FIRMWARE_UPDATE_CHECK_MOCK,
),
patch(
"homeassistant.components.qnap_qsw.QnapQswApi.get_lacp_info",
return_value=LACP_INFO_MOCK,
),
patch(
"homeassistant.components.qnap_qsw.QnapQswApi.get_ports_statistics",
return_value=PORTS_STATISTICS_MOCK,
),
patch(
"homeassistant.components.qnap_qsw.QnapQswApi.get_ports_status",
return_value=PORTS_STATUS_MOCK,
),
patch(
"homeassistant.components.qnap_qsw.QnapQswApi.get_system_board",
return_value=SYSTEM_BOARD_MOCK,
),
patch(
"homeassistant.components.qnap_qsw.QnapQswApi.get_system_sensor",
return_value=SYSTEM_SENSOR_MOCK,
),
patch(
"homeassistant.components.qnap_qsw.QnapQswApi.get_system_time",
return_value=SYSTEM_TIME_MOCK,
),
patch(
"homeassistant.components.qnap_qsw.QnapQswApi.get_users_verification",
return_value=USERS_VERIFICATION_MOCK,
),
patch(
"homeassistant.components.qnap_qsw.QnapQswApi.post_users_login",
return_value=USERS_LOGIN_MOCK,
),
):
await hass.config_entries.async_setup(config_entry.entry_id)
await hass.async_block_till_done()
Reference in New Issue View Git Blame Copy Permalink