mirror of
https://github.com/home-assistant/core.git
synced 2025-12-26 05:57:01 +00:00
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
"""Test Tuya switch platform."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from syrupy.assertion import SnapshotAssertion
|
|
from tuya_sharing import CustomerDevice, Manager
|
|
|
|
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
|
|
from homeassistant.components.tuya import DOMAIN
|
|
from homeassistant.const import Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er, issue_registry as ir
|
|
|
|
from . import initialize_entry
|
|
|
|
from tests.common import MockConfigEntry, snapshot_platform
|
|
|
|
|
|
@patch("homeassistant.components.tuya.PLATFORMS", [Platform.SWITCH])
|
|
async def test_platform_setup_and_discovery(
|
|
hass: HomeAssistant,
|
|
mock_manager: Manager,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_devices: list[CustomerDevice],
|
|
entity_registry: er.EntityRegistry,
|
|
snapshot: SnapshotAssertion,
|
|
) -> None:
|
|
"""Test platform setup and discovery."""
|
|
await initialize_entry(hass, mock_manager, mock_config_entry, mock_devices)
|
|
|
|
await snapshot_platform(hass, entity_registry, snapshot, mock_config_entry.entry_id)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("preexisting_entity", "disabled_by", "expected_entity", "expected_issue"),
|
|
[
|
|
(True, None, True, True),
|
|
(True, er.RegistryEntryDisabler.USER, False, False),
|
|
(False, None, False, False),
|
|
],
|
|
)
|
|
@pytest.mark.parametrize(
|
|
"mock_device_code",
|
|
["sfkzq_rzklytdei8i8vo37"],
|
|
)
|
|
async def test_sfkzq_deprecated_switch(
|
|
hass: HomeAssistant,
|
|
mock_manager: Manager,
|
|
mock_config_entry: MockConfigEntry,
|
|
mock_device: CustomerDevice,
|
|
issue_registry: ir.IssueRegistry,
|
|
entity_registry: er.EntityRegistry,
|
|
preexisting_entity: bool,
|
|
disabled_by: er.RegistryEntryDisabler,
|
|
expected_entity: bool,
|
|
expected_issue: bool,
|
|
) -> None:
|
|
"""Test switch deprecation issue."""
|
|
original_entity_id = "switch.balkonbewasserung_switch"
|
|
entity_unique_id = "tuya.73ov8i8iedtylkzrqzkfsswitch"
|
|
if preexisting_entity:
|
|
suggested_id = original_entity_id.replace(f"{SWITCH_DOMAIN}.", "")
|
|
entity_registry.async_get_or_create(
|
|
SWITCH_DOMAIN,
|
|
DOMAIN,
|
|
entity_unique_id,
|
|
suggested_object_id=suggested_id,
|
|
disabled_by=disabled_by,
|
|
)
|
|
|
|
await initialize_entry(hass, mock_manager, mock_config_entry, mock_device)
|
|
|
|
assert (
|
|
entity_registry.async_get(original_entity_id) is not None
|
|
) is expected_entity
|
|
assert (
|
|
issue_registry.async_get_issue(
|
|
domain=DOMAIN,
|
|
issue_id=f"deprecated_entity_{entity_unique_id}",
|
|
)
|
|
is not None
|
|
) is expected_issue
|