mirror of
https://github.com/home-assistant/core.git
synced 2025-07-23 21:27:38 +00:00
Add D-link tests (#86825)
* Fix D-Link config flow auth * Add tests to D-Link * pyupgrade
This commit is contained in:
parent
adb0d85511
commit
b69576d6de
@ -205,10 +205,7 @@ omit =
|
|||||||
homeassistant/components/discord/notify.py
|
homeassistant/components/discord/notify.py
|
||||||
homeassistant/components/dlib_face_detect/image_processing.py
|
homeassistant/components/dlib_face_detect/image_processing.py
|
||||||
homeassistant/components/dlib_face_identify/image_processing.py
|
homeassistant/components/dlib_face_identify/image_processing.py
|
||||||
homeassistant/components/dlink/__init__.py
|
|
||||||
homeassistant/components/dlink/data.py
|
homeassistant/components/dlink/data.py
|
||||||
homeassistant/components/dlink/entity.py
|
|
||||||
homeassistant/components/dlink/switch.py
|
|
||||||
homeassistant/components/dominos/*
|
homeassistant/components/dominos/*
|
||||||
homeassistant/components/doods/*
|
homeassistant/components/doods/*
|
||||||
homeassistant/components/doorbird/__init__.py
|
homeassistant/components/doorbird/__init__.py
|
||||||
|
@ -1,5 +1,6 @@
|
|||||||
"""Configure pytest for D-Link tests."""
|
"""Configure pytest for D-Link tests."""
|
||||||
|
|
||||||
|
from collections.abc import Awaitable, Callable, Generator
|
||||||
from copy import deepcopy
|
from copy import deepcopy
|
||||||
from unittest.mock import MagicMock, patch
|
from unittest.mock import MagicMock, patch
|
||||||
|
|
||||||
@ -10,6 +11,7 @@ from homeassistant.components.dlink.const import CONF_USE_LEGACY_PROTOCOL, DOMAI
|
|||||||
from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PASSWORD, CONF_USERNAME
|
from homeassistant.const import CONF_HOST, CONF_NAME, CONF_PASSWORD, CONF_USERNAME
|
||||||
from homeassistant.core import HomeAssistant
|
from homeassistant.core import HomeAssistant
|
||||||
from homeassistant.helpers.device_registry import format_mac
|
from homeassistant.helpers.device_registry import format_mac
|
||||||
|
from homeassistant.setup import async_setup_component
|
||||||
|
|
||||||
from tests.common import MockConfigEntry
|
from tests.common import MockConfigEntry
|
||||||
|
|
||||||
@ -40,6 +42,8 @@ CONF_DHCP_FLOW_NEW_IP = dhcp.DhcpServiceInfo(
|
|||||||
hostname="dsp-w215",
|
hostname="dsp-w215",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
ComponentSetup = Callable[[], Awaitable[None]]
|
||||||
|
|
||||||
|
|
||||||
def create_entry(hass: HomeAssistant) -> MockConfigEntry:
|
def create_entry(hass: HomeAssistant) -> MockConfigEntry:
|
||||||
"""Create fixture for adding config entry in Home Assistant."""
|
"""Create fixture for adding config entry in Home Assistant."""
|
||||||
@ -67,24 +71,86 @@ def mocked_plug() -> MagicMock:
|
|||||||
"""Create mocked plug device."""
|
"""Create mocked plug device."""
|
||||||
mocked_plug = MagicMock()
|
mocked_plug = MagicMock()
|
||||||
mocked_plug.state = "OFF"
|
mocked_plug.state = "OFF"
|
||||||
mocked_plug.temperature = 0
|
mocked_plug.temperature = "33"
|
||||||
mocked_plug.current_consumption = "N/A"
|
mocked_plug.current_consumption = "50"
|
||||||
mocked_plug.total_consumption = "N/A"
|
mocked_plug.total_consumption = "1040"
|
||||||
mocked_plug.authenticated = ("0123456789ABCDEF0123456789ABCDEF", "ABCDefGHiJ")
|
mocked_plug.authenticated = None
|
||||||
|
mocked_plug.use_legacy_protocol = False
|
||||||
|
mocked_plug.model_name = "DSP-W215"
|
||||||
return mocked_plug
|
return mocked_plug
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def mocked_plug_no_auth(mocked_plug: MagicMock) -> MagicMock:
|
def mocked_plug_legacy() -> MagicMock:
|
||||||
"""Create mocked unauthenticated plug device."""
|
"""Create mocked legacy plug device."""
|
||||||
mocked_plug = deepcopy(mocked_plug)
|
mocked_plug = MagicMock()
|
||||||
mocked_plug.authenticated = None
|
mocked_plug.state = "OFF"
|
||||||
|
mocked_plug.temperature = "N/A"
|
||||||
|
mocked_plug.current_consumption = "N/A"
|
||||||
|
mocked_plug.total_consumption = "N/A"
|
||||||
|
mocked_plug.authenticated = ("0123456789ABCDEF0123456789ABCDEF", "ABCDefGHiJ")
|
||||||
|
mocked_plug.use_legacy_protocol = True
|
||||||
|
mocked_plug.model_name = "DSP-W215"
|
||||||
return mocked_plug
|
return mocked_plug
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def mocked_plug_legacy_no_auth(mocked_plug_legacy: MagicMock) -> MagicMock:
|
||||||
|
"""Create mocked legacy unauthenticated plug device."""
|
||||||
|
mocked_plug_legacy = deepcopy(mocked_plug_legacy)
|
||||||
|
mocked_plug_legacy.authenticated = None
|
||||||
|
return mocked_plug_legacy
|
||||||
|
|
||||||
|
|
||||||
def patch_config_flow(mocked_plug: MagicMock):
|
def patch_config_flow(mocked_plug: MagicMock):
|
||||||
"""Patch D-Link Smart Plug config flow."""
|
"""Patch D-Link Smart Plug config flow."""
|
||||||
return patch(
|
return patch(
|
||||||
"homeassistant.components.dlink.config_flow.SmartPlug",
|
"homeassistant.components.dlink.config_flow.SmartPlug",
|
||||||
return_value=mocked_plug,
|
return_value=mocked_plug,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def patch_setup(mocked_plug: MagicMock):
|
||||||
|
"""Patch D-Link Smart Plug object."""
|
||||||
|
return patch(
|
||||||
|
"homeassistant.components.dlink.SmartPlug",
|
||||||
|
return_value=mocked_plug,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
async def mock_setup_integration(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
mocked_plug: MagicMock,
|
||||||
|
) -> None:
|
||||||
|
"""Set up the D-Link integration in Home Assistant."""
|
||||||
|
with patch_setup(mocked_plug):
|
||||||
|
assert await async_setup_component(hass, DOMAIN, {})
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
async def setup_integration(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
config_entry_with_uid: MockConfigEntry,
|
||||||
|
mocked_plug: MagicMock,
|
||||||
|
) -> Generator[ComponentSetup, None, None]:
|
||||||
|
"""Set up the D-Link integration in Home Assistant."""
|
||||||
|
|
||||||
|
async def func() -> None:
|
||||||
|
await mock_setup_integration(hass, mocked_plug)
|
||||||
|
|
||||||
|
return func
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
async def setup_integration_legacy(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
config_entry_with_uid: MockConfigEntry,
|
||||||
|
mocked_plug_legacy: MagicMock,
|
||||||
|
) -> Generator[ComponentSetup, None, None]:
|
||||||
|
"""Set up the D-Link integration in Home Assistant with different data."""
|
||||||
|
|
||||||
|
async def func() -> None:
|
||||||
|
await mock_setup_integration(hass, mocked_plug_legacy)
|
||||||
|
|
||||||
|
return func
|
||||||
|
@ -53,10 +53,12 @@ async def test_flow_user_already_configured(
|
|||||||
|
|
||||||
|
|
||||||
async def test_flow_user_cannot_connect(
|
async def test_flow_user_cannot_connect(
|
||||||
hass: HomeAssistant, mocked_plug: MagicMock, mocked_plug_no_auth: MagicMock
|
hass: HomeAssistant,
|
||||||
|
mocked_plug_legacy: MagicMock,
|
||||||
|
mocked_plug_legacy_no_auth: MagicMock,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test user initialized flow with unreachable server."""
|
"""Test user initialized flow with unreachable server."""
|
||||||
with patch_config_flow(mocked_plug_no_auth):
|
with patch_config_flow(mocked_plug_legacy_no_auth):
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN, context={"source": SOURCE_USER}, data=CONF_DATA
|
DOMAIN, context={"source": SOURCE_USER}, data=CONF_DATA
|
||||||
)
|
)
|
||||||
@ -64,7 +66,7 @@ async def test_flow_user_cannot_connect(
|
|||||||
assert result["step_id"] == "user"
|
assert result["step_id"] == "user"
|
||||||
assert result["errors"]["base"] == "cannot_connect"
|
assert result["errors"]["base"] == "cannot_connect"
|
||||||
|
|
||||||
with patch_config_flow(mocked_plug), _patch_setup_entry():
|
with patch_config_flow(mocked_plug_legacy), _patch_setup_entry():
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
user_input=CONF_DATA,
|
user_input=CONF_DATA,
|
||||||
@ -127,16 +129,16 @@ async def test_dhcp(hass: HomeAssistant, mocked_plug: MagicMock) -> None:
|
|||||||
assert result["data"] == CONF_DATA
|
assert result["data"] == CONF_DATA
|
||||||
|
|
||||||
|
|
||||||
async def test_dhcp_failed_auth(
|
async def test_dhcp_failed_legacy_auth(
|
||||||
hass: HomeAssistant, mocked_plug: MagicMock, mocked_plug_no_auth: MagicMock
|
hass: HomeAssistant, mocked_plug: MagicMock, mocked_plug_legacy_no_auth: MagicMock
|
||||||
) -> None:
|
) -> None:
|
||||||
"""Test we can recovery from failed authentication during dhcp flow."""
|
"""Test we can recover from failed legacy authentication during dhcp flow."""
|
||||||
result = await hass.config_entries.flow.async_init(
|
result = await hass.config_entries.flow.async_init(
|
||||||
DOMAIN, context={"source": SOURCE_DHCP}, data=CONF_DHCP_FLOW
|
DOMAIN, context={"source": SOURCE_DHCP}, data=CONF_DHCP_FLOW
|
||||||
)
|
)
|
||||||
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
||||||
assert result["step_id"] == "confirm_discovery"
|
assert result["step_id"] == "confirm_discovery"
|
||||||
with patch_config_flow(mocked_plug_no_auth):
|
with patch_config_flow(mocked_plug_legacy_no_auth):
|
||||||
result = await hass.config_entries.flow.async_configure(
|
result = await hass.config_entries.flow.async_configure(
|
||||||
result["flow_id"],
|
result["flow_id"],
|
||||||
user_input=CONF_DHCP_DATA,
|
user_input=CONF_DHCP_DATA,
|
||||||
|
75
tests/components/dlink/test_init.py
Normal file
75
tests/components/dlink/test_init.py
Normal file
@ -0,0 +1,75 @@
|
|||||||
|
"""Test D-Link Smart Plug setup."""
|
||||||
|
from unittest.mock import MagicMock
|
||||||
|
|
||||||
|
from homeassistant.components.dlink.const import DOMAIN
|
||||||
|
from homeassistant.config_entries import ConfigEntryState
|
||||||
|
from homeassistant.core import HomeAssistant
|
||||||
|
from homeassistant.helpers import device_registry as dr
|
||||||
|
|
||||||
|
from .conftest import CONF_DATA, ComponentSetup, patch_setup
|
||||||
|
|
||||||
|
from tests.common import MockConfigEntry
|
||||||
|
|
||||||
|
|
||||||
|
async def test_setup_config_and_unload(
|
||||||
|
hass: HomeAssistant, setup_integration: ComponentSetup
|
||||||
|
) -> None:
|
||||||
|
"""Test setup and unload."""
|
||||||
|
await setup_integration()
|
||||||
|
|
||||||
|
entry = hass.config_entries.async_entries(DOMAIN)[0]
|
||||||
|
assert entry.state == ConfigEntryState.LOADED
|
||||||
|
assert len(hass.config_entries.async_entries(DOMAIN)) == 1
|
||||||
|
assert entry.data == CONF_DATA
|
||||||
|
|
||||||
|
assert await hass.config_entries.async_unload(entry.entry_id)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
assert entry.state is ConfigEntryState.NOT_LOADED
|
||||||
|
assert not hass.data.get(DOMAIN)
|
||||||
|
|
||||||
|
|
||||||
|
async def test_legacy_setup_config_and_unload(
|
||||||
|
hass: HomeAssistant, setup_integration_legacy: ComponentSetup
|
||||||
|
) -> None:
|
||||||
|
"""Test legacy setup and unload."""
|
||||||
|
await setup_integration_legacy()
|
||||||
|
|
||||||
|
entry = hass.config_entries.async_entries(DOMAIN)[0]
|
||||||
|
assert entry.state == ConfigEntryState.LOADED
|
||||||
|
assert len(hass.config_entries.async_entries(DOMAIN)) == 1
|
||||||
|
assert entry.data == CONF_DATA
|
||||||
|
|
||||||
|
assert await hass.config_entries.async_unload(entry.entry_id)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
assert entry.state is ConfigEntryState.NOT_LOADED
|
||||||
|
assert not hass.data.get(DOMAIN)
|
||||||
|
|
||||||
|
|
||||||
|
async def test_async_setup_entry_not_ready(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
config_entry_with_uid: MockConfigEntry,
|
||||||
|
mocked_plug_legacy_no_auth: MagicMock,
|
||||||
|
) -> None:
|
||||||
|
"""Test that it throws ConfigEntryNotReady when exception occurs during legacy setup."""
|
||||||
|
with patch_setup(mocked_plug_legacy_no_auth):
|
||||||
|
await hass.config_entries.async_setup(config_entry_with_uid.entry_id)
|
||||||
|
assert config_entry_with_uid.state == ConfigEntryState.SETUP_RETRY
|
||||||
|
|
||||||
|
|
||||||
|
async def test_device_info(
|
||||||
|
hass: HomeAssistant, setup_integration: ComponentSetup
|
||||||
|
) -> None:
|
||||||
|
"""Test device info."""
|
||||||
|
await setup_integration()
|
||||||
|
|
||||||
|
entry = hass.config_entries.async_entries(DOMAIN)[0]
|
||||||
|
device_registry = dr.async_get(hass)
|
||||||
|
device = device_registry.async_get_device({(DOMAIN, entry.entry_id)})
|
||||||
|
|
||||||
|
assert device.connections == {("mac", "aa:bb:cc:dd:ee:ff")}
|
||||||
|
assert device.identifiers == {(DOMAIN, entry.entry_id)}
|
||||||
|
assert device.manufacturer == "D-Link"
|
||||||
|
assert device.model == "DSP-W215"
|
||||||
|
assert device.name == "Mock Title"
|
77
tests/components/dlink/test_switch.py
Normal file
77
tests/components/dlink/test_switch.py
Normal file
@ -0,0 +1,77 @@
|
|||||||
|
"""Switch tests for the D-Link Smart Plug integration."""
|
||||||
|
from collections.abc import Awaitable, Callable
|
||||||
|
|
||||||
|
from homeassistant.components.dlink import DOMAIN
|
||||||
|
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
|
||||||
|
from homeassistant.const import (
|
||||||
|
ATTR_ENTITY_ID,
|
||||||
|
SERVICE_TURN_OFF,
|
||||||
|
SERVICE_TURN_ON,
|
||||||
|
STATE_OFF,
|
||||||
|
STATE_ON,
|
||||||
|
)
|
||||||
|
from homeassistant.core import HomeAssistant
|
||||||
|
from homeassistant.setup import async_setup_component
|
||||||
|
|
||||||
|
from .conftest import ComponentSetup
|
||||||
|
|
||||||
|
from tests.components.repairs import get_repairs
|
||||||
|
|
||||||
|
|
||||||
|
async def test_switch_state(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
hass_ws_client: Callable[[HomeAssistant], Awaitable[None]],
|
||||||
|
setup_integration: ComponentSetup,
|
||||||
|
) -> None:
|
||||||
|
"""Test we get the switch status."""
|
||||||
|
assert await async_setup_component(
|
||||||
|
hass,
|
||||||
|
SWITCH_DOMAIN,
|
||||||
|
{
|
||||||
|
SWITCH_DOMAIN: {
|
||||||
|
"platform": DOMAIN,
|
||||||
|
"host": "1.2.3.4",
|
||||||
|
"username": "admin",
|
||||||
|
"password": "123456",
|
||||||
|
"use_legacy_protocol": True,
|
||||||
|
}
|
||||||
|
},
|
||||||
|
)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
issues = await get_repairs(hass, hass_ws_client)
|
||||||
|
assert len(issues) == 1
|
||||||
|
assert issues[0]["issue_id"] == "deprecated_yaml"
|
||||||
|
|
||||||
|
await setup_integration()
|
||||||
|
|
||||||
|
entity_id = "switch.mock_title_switch"
|
||||||
|
state = hass.states.get(entity_id)
|
||||||
|
assert state.state == STATE_OFF
|
||||||
|
assert state.attributes["total_consumption"] == 1040.0
|
||||||
|
assert state.attributes["temperature"] == 33
|
||||||
|
await hass.services.async_call(
|
||||||
|
SWITCH_DOMAIN,
|
||||||
|
SERVICE_TURN_ON,
|
||||||
|
{ATTR_ENTITY_ID: [entity_id]},
|
||||||
|
blocking=True,
|
||||||
|
)
|
||||||
|
assert hass.states.get(entity_id).state == STATE_ON
|
||||||
|
await hass.services.async_call(
|
||||||
|
SWITCH_DOMAIN,
|
||||||
|
SERVICE_TURN_OFF,
|
||||||
|
{ATTR_ENTITY_ID: [entity_id]},
|
||||||
|
blocking=True,
|
||||||
|
)
|
||||||
|
assert hass.states.get(entity_id).state == STATE_OFF
|
||||||
|
|
||||||
|
|
||||||
|
async def test_switch_no_value(
|
||||||
|
hass: HomeAssistant, setup_integration_legacy: ComponentSetup
|
||||||
|
) -> None:
|
||||||
|
"""Test we handle 'N/A' being passed by the pypi package."""
|
||||||
|
await setup_integration_legacy()
|
||||||
|
|
||||||
|
state = hass.states.get("switch.mock_title_switch")
|
||||||
|
assert state.state == STATE_OFF
|
||||||
|
assert state.attributes["total_consumption"] is None
|
||||||
|
assert state.attributes["temperature"] is None
|
Loading…
x
Reference in New Issue
Block a user