151 lines
4.2 KiB
Python
151 lines
4.2 KiB
Python
"""Test for the SmartThings event platform."""
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
from freezegun.api import FrozenDateTimeFactory
|
|
from pysmartthings import Attribute, Capability
|
|
from pysmartthings.models import HealthStatus
|
|
import pytest
|
|
from syrupy.assertion import SnapshotAssertion
|
|
|
|
from homeassistant.components.event import ATTR_EVENT_TYPES
|
|
from homeassistant.const import STATE_UNAVAILABLE, STATE_UNKNOWN, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from . import (
|
|
setup_integration,
|
|
snapshot_smartthings_entities,
|
|
trigger_health_update,
|
|
trigger_update,
|
|
)
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_all_entities(
|
|
hass: HomeAssistant,
|
|
snapshot: SnapshotAssertion,
|
|
devices: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
entity_registry: er.EntityRegistry,
|
|
) -> None:
|
|
"""Test all entities."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
snapshot_smartthings_entities(hass, entity_registry, snapshot, Platform.EVENT)
|
|
|
|
|
|
@pytest.mark.parametrize("device_fixture", ["heatit_zpushwall"])
|
|
async def test_state_update(
|
|
hass: HomeAssistant,
|
|
devices: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
freezer: FrozenDateTimeFactory,
|
|
) -> None:
|
|
"""Test state update."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
freezer.move_to("2023-10-21")
|
|
|
|
assert (
|
|
hass.states.get("event.livingroom_smart_switch_button1").state == STATE_UNKNOWN
|
|
)
|
|
|
|
await trigger_update(
|
|
hass,
|
|
devices,
|
|
"5e5b97f3-3094-44e6-abc0-f61283412d6a",
|
|
Capability.BUTTON,
|
|
Attribute.BUTTON,
|
|
"pushed",
|
|
component="button1",
|
|
)
|
|
|
|
assert (
|
|
hass.states.get("event.livingroom_smart_switch_button1").state
|
|
== "2023-10-21T00:00:00.000+00:00"
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("device_fixture", ["heatit_zpushwall"])
|
|
async def test_supported_button_values_update(
|
|
hass: HomeAssistant,
|
|
devices: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
freezer: FrozenDateTimeFactory,
|
|
) -> None:
|
|
"""Test supported button values update."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
freezer.move_to("2023-10-21")
|
|
|
|
assert (
|
|
hass.states.get("event.livingroom_smart_switch_button1").state == STATE_UNKNOWN
|
|
)
|
|
assert hass.states.get("event.livingroom_smart_switch_button1").attributes[
|
|
ATTR_EVENT_TYPES
|
|
] == ["pushed", "held", "down_hold"]
|
|
|
|
await trigger_update(
|
|
hass,
|
|
devices,
|
|
"5e5b97f3-3094-44e6-abc0-f61283412d6a",
|
|
Capability.BUTTON,
|
|
Attribute.SUPPORTED_BUTTON_VALUES,
|
|
["pushed", "held", "down_hold", "pushed_2x"],
|
|
component="button1",
|
|
)
|
|
|
|
assert (
|
|
hass.states.get("event.livingroom_smart_switch_button1").state == STATE_UNKNOWN
|
|
)
|
|
assert hass.states.get("event.livingroom_smart_switch_button1").attributes[
|
|
ATTR_EVENT_TYPES
|
|
] == ["pushed", "held", "down_hold", "pushed_2x"]
|
|
|
|
|
|
@pytest.mark.parametrize("device_fixture", ["heatit_zpushwall"])
|
|
async def test_availability(
|
|
hass: HomeAssistant,
|
|
devices: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test availability."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
assert (
|
|
hass.states.get("event.livingroom_smart_switch_button1").state == STATE_UNKNOWN
|
|
)
|
|
|
|
await trigger_health_update(
|
|
hass, devices, "5e5b97f3-3094-44e6-abc0-f61283412d6a", HealthStatus.OFFLINE
|
|
)
|
|
|
|
assert (
|
|
hass.states.get("event.livingroom_smart_switch_button1").state
|
|
== STATE_UNAVAILABLE
|
|
)
|
|
|
|
await trigger_health_update(
|
|
hass, devices, "5e5b97f3-3094-44e6-abc0-f61283412d6a", HealthStatus.ONLINE
|
|
)
|
|
|
|
assert (
|
|
hass.states.get("event.livingroom_smart_switch_button1").state == STATE_UNKNOWN
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("device_fixture", ["heatit_zpushwall"])
|
|
async def test_availability_at_start(
|
|
hass: HomeAssistant,
|
|
unavailable_device: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test unavailable at boot."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
assert (
|
|
hass.states.get("event.livingroom_smart_switch_button1").state
|
|
== STATE_UNAVAILABLE
|
|
)
|