mirror of
https://github.com/home-assistant/core.git
synced 2025-06-07 14:47:12 +00:00
4f8363a5c2
* Bump pySmartThings to 3.1.0 * Bump pySmartThings to 3.2.0 * Add availability to SmartThings devices * Add availability to SmartThings devices * Add availability to SmartThings devices
346 lines
11 KiB
Python
346 lines
11 KiB
Python
"""Test for the SmartThings sensors platform."""
|
|
|
|
from unittest.mock import AsyncMock
|
|
|
|
from pysmartthings import Attribute, Capability
|
|
from pysmartthings.models import HealthStatus
|
|
import pytest
|
|
from syrupy import SnapshotAssertion
|
|
|
|
from homeassistant.components import automation, script
|
|
from homeassistant.components.automation import automations_with_entity
|
|
from homeassistant.components.script import scripts_with_entity
|
|
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
|
|
from homeassistant.components.smartthings.const import DOMAIN, MAIN
|
|
from homeassistant.const import STATE_UNAVAILABLE, STATE_UNKNOWN, Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er, issue_registry as ir
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
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.SENSOR)
|
|
|
|
|
|
@pytest.mark.parametrize("device_fixture", ["da_ac_rac_000001"])
|
|
async def test_state_update(
|
|
hass: HomeAssistant,
|
|
devices: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
) -> None:
|
|
"""Test state update."""
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
assert hass.states.get("sensor.ac_office_granit_temperature").state == "25"
|
|
|
|
await trigger_update(
|
|
hass,
|
|
devices,
|
|
"96a5ef74-5832-a84b-f1f7-ca799957065d",
|
|
Capability.TEMPERATURE_MEASUREMENT,
|
|
Attribute.TEMPERATURE,
|
|
20,
|
|
)
|
|
|
|
assert hass.states.get("sensor.ac_office_granit_temperature").state == "20"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
(
|
|
"device_fixture",
|
|
"unique_id",
|
|
"suggested_object_id",
|
|
"issue_string",
|
|
"entity_id",
|
|
"expected_state",
|
|
),
|
|
[
|
|
(
|
|
"vd_stv_2017_k",
|
|
f"4588d2d9-a8cf-40f4-9a0b-ed5dfbaccda1_{MAIN}_{Capability.MEDIA_PLAYBACK}_{Attribute.PLAYBACK_STATUS}_{Attribute.PLAYBACK_STATUS}",
|
|
"tv_samsung_8_series_49_media_playback_status",
|
|
"media_player",
|
|
"sensor.tv_samsung_8_series_49_media_playback_status",
|
|
STATE_UNKNOWN,
|
|
),
|
|
(
|
|
"vd_stv_2017_k",
|
|
f"4588d2d9-a8cf-40f4-9a0b-ed5dfbaccda1_{MAIN}_{Capability.AUDIO_VOLUME}_{Attribute.VOLUME}_{Attribute.VOLUME}",
|
|
"tv_samsung_8_series_49_volume",
|
|
"media_player",
|
|
"sensor.tv_samsung_8_series_49_volume",
|
|
"13",
|
|
),
|
|
(
|
|
"vd_stv_2017_k",
|
|
f"4588d2d9-a8cf-40f4-9a0b-ed5dfbaccda1_{MAIN}_{Capability.MEDIA_INPUT_SOURCE}_{Attribute.INPUT_SOURCE}_{Attribute.INPUT_SOURCE}",
|
|
"tv_samsung_8_series_49_media_input_source",
|
|
"media_player",
|
|
"sensor.tv_samsung_8_series_49_media_input_source",
|
|
"hdmi1",
|
|
),
|
|
(
|
|
"im_speaker_ai_0001",
|
|
f"c9276e43-fe3c-88c3-1dcc-2eb79e292b8c_{MAIN}_{Capability.MEDIA_PLAYBACK_REPEAT}_{Attribute.PLAYBACK_REPEAT_MODE}_{Attribute.PLAYBACK_REPEAT_MODE}",
|
|
"galaxy_home_mini_media_playback_repeat",
|
|
"media_player",
|
|
"sensor.galaxy_home_mini_media_playback_repeat",
|
|
"off",
|
|
),
|
|
(
|
|
"im_speaker_ai_0001",
|
|
f"c9276e43-fe3c-88c3-1dcc-2eb79e292b8c_{MAIN}_{Capability.MEDIA_PLAYBACK_SHUFFLE}_{Attribute.PLAYBACK_SHUFFLE}_{Attribute.PLAYBACK_SHUFFLE}",
|
|
"galaxy_home_mini_media_playback_shuffle",
|
|
"media_player",
|
|
"sensor.galaxy_home_mini_media_playback_shuffle",
|
|
"disabled",
|
|
),
|
|
],
|
|
)
|
|
async def test_create_issue_with_items(
|
|
hass: HomeAssistant,
|
|
devices: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
entity_registry: er.EntityRegistry,
|
|
issue_registry: ir.IssueRegistry,
|
|
unique_id: str,
|
|
suggested_object_id: str,
|
|
issue_string: str,
|
|
entity_id: str,
|
|
expected_state: str,
|
|
) -> None:
|
|
"""Test we create an issue when an automation or script is using a deprecated entity."""
|
|
issue_id = f"deprecated_{issue_string}_{entity_id}"
|
|
|
|
entity_entry = entity_registry.async_get_or_create(
|
|
SENSOR_DOMAIN,
|
|
DOMAIN,
|
|
unique_id,
|
|
suggested_object_id=suggested_object_id,
|
|
original_name=suggested_object_id,
|
|
)
|
|
|
|
assert await async_setup_component(
|
|
hass,
|
|
automation.DOMAIN,
|
|
{
|
|
automation.DOMAIN: {
|
|
"id": "test",
|
|
"alias": "test",
|
|
"trigger": {"platform": "state", "entity_id": entity_id},
|
|
"action": {
|
|
"action": "automation.turn_on",
|
|
"target": {
|
|
"entity_id": "automation.test",
|
|
},
|
|
},
|
|
}
|
|
},
|
|
)
|
|
assert await async_setup_component(
|
|
hass,
|
|
script.DOMAIN,
|
|
{
|
|
script.DOMAIN: {
|
|
"test": {
|
|
"sequence": [
|
|
{
|
|
"condition": "state",
|
|
"entity_id": entity_id,
|
|
"state": "on",
|
|
},
|
|
],
|
|
}
|
|
}
|
|
},
|
|
)
|
|
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
assert hass.states.get(entity_id).state == expected_state
|
|
|
|
assert automations_with_entity(hass, entity_id)[0] == "automation.test"
|
|
assert scripts_with_entity(hass, entity_id)[0] == "script.test"
|
|
|
|
assert len(issue_registry.issues) == 1
|
|
issue = issue_registry.async_get_issue(DOMAIN, issue_id)
|
|
assert issue is not None
|
|
assert issue.translation_key == f"deprecated_{issue_string}_scripts"
|
|
assert issue.translation_placeholders == {
|
|
"entity_id": entity_id,
|
|
"entity_name": suggested_object_id,
|
|
"items": "- [test](/config/automation/edit/test)\n- [test](/config/script/edit/test)",
|
|
}
|
|
|
|
entity_registry.async_update_entity(
|
|
entity_entry.entity_id,
|
|
disabled_by=er.RegistryEntryDisabler.USER,
|
|
)
|
|
|
|
await hass.config_entries.async_reload(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
# Assert the issue is no longer present
|
|
assert not issue_registry.async_get_issue(DOMAIN, issue_id)
|
|
assert len(issue_registry.issues) == 0
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
(
|
|
"device_fixture",
|
|
"unique_id",
|
|
"suggested_object_id",
|
|
"issue_string",
|
|
"entity_id",
|
|
"expected_state",
|
|
),
|
|
[
|
|
(
|
|
"vd_stv_2017_k",
|
|
f"4588d2d9-a8cf-40f4-9a0b-ed5dfbaccda1_{MAIN}_{Capability.MEDIA_PLAYBACK}_{Attribute.PLAYBACK_STATUS}_{Attribute.PLAYBACK_STATUS}",
|
|
"tv_samsung_8_series_49_media_playback_status",
|
|
"media_player",
|
|
"sensor.tv_samsung_8_series_49_media_playback_status",
|
|
STATE_UNKNOWN,
|
|
),
|
|
(
|
|
"vd_stv_2017_k",
|
|
f"4588d2d9-a8cf-40f4-9a0b-ed5dfbaccda1_{MAIN}_{Capability.AUDIO_VOLUME}_{Attribute.VOLUME}_{Attribute.VOLUME}",
|
|
"tv_samsung_8_series_49_volume",
|
|
"media_player",
|
|
"sensor.tv_samsung_8_series_49_volume",
|
|
"13",
|
|
),
|
|
(
|
|
"vd_stv_2017_k",
|
|
f"4588d2d9-a8cf-40f4-9a0b-ed5dfbaccda1_{MAIN}_{Capability.MEDIA_INPUT_SOURCE}_{Attribute.INPUT_SOURCE}_{Attribute.INPUT_SOURCE}",
|
|
"tv_samsung_8_series_49_media_input_source",
|
|
"media_player",
|
|
"sensor.tv_samsung_8_series_49_media_input_source",
|
|
"hdmi1",
|
|
),
|
|
(
|
|
"im_speaker_ai_0001",
|
|
f"c9276e43-fe3c-88c3-1dcc-2eb79e292b8c_{MAIN}_{Capability.MEDIA_PLAYBACK_REPEAT}_{Attribute.PLAYBACK_REPEAT_MODE}_{Attribute.PLAYBACK_REPEAT_MODE}",
|
|
"galaxy_home_mini_media_playback_repeat",
|
|
"media_player",
|
|
"sensor.galaxy_home_mini_media_playback_repeat",
|
|
"off",
|
|
),
|
|
(
|
|
"im_speaker_ai_0001",
|
|
f"c9276e43-fe3c-88c3-1dcc-2eb79e292b8c_{MAIN}_{Capability.MEDIA_PLAYBACK_SHUFFLE}_{Attribute.PLAYBACK_SHUFFLE}_{Attribute.PLAYBACK_SHUFFLE}",
|
|
"galaxy_home_mini_media_playback_shuffle",
|
|
"media_player",
|
|
"sensor.galaxy_home_mini_media_playback_shuffle",
|
|
"disabled",
|
|
),
|
|
],
|
|
)
|
|
async def test_create_issue(
|
|
hass: HomeAssistant,
|
|
devices: AsyncMock,
|
|
mock_config_entry: MockConfigEntry,
|
|
entity_registry: er.EntityRegistry,
|
|
issue_registry: ir.IssueRegistry,
|
|
unique_id: str,
|
|
suggested_object_id: str,
|
|
issue_string: str,
|
|
entity_id: str,
|
|
expected_state: str,
|
|
) -> None:
|
|
"""Test we create an issue when an automation or script is using a deprecated entity."""
|
|
issue_id = f"deprecated_{issue_string}_{entity_id}"
|
|
|
|
entity_entry = entity_registry.async_get_or_create(
|
|
SENSOR_DOMAIN,
|
|
DOMAIN,
|
|
unique_id,
|
|
suggested_object_id=suggested_object_id,
|
|
original_name=suggested_object_id,
|
|
)
|
|
|
|
await setup_integration(hass, mock_config_entry)
|
|
|
|
assert hass.states.get(entity_id).state == expected_state
|
|
|
|
assert len(issue_registry.issues) == 1
|
|
issue = issue_registry.async_get_issue(DOMAIN, issue_id)
|
|
assert issue is not None
|
|
assert issue.translation_key == f"deprecated_{issue_string}"
|
|
assert issue.translation_placeholders == {
|
|
"entity_id": entity_id,
|
|
"entity_name": suggested_object_id,
|
|
}
|
|
|
|
entity_registry.async_update_entity(
|
|
entity_entry.entity_id,
|
|
disabled_by=er.RegistryEntryDisabler.USER,
|
|
)
|
|
|
|
await hass.config_entries.async_reload(mock_config_entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
# Assert the issue is no longer present
|
|
assert not issue_registry.async_get_issue(DOMAIN, issue_id)
|
|
assert len(issue_registry.issues) == 0
|
|
|
|
|
|
@pytest.mark.parametrize("device_fixture", ["da_ac_rac_000001"])
|
|
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("sensor.ac_office_granit_temperature").state == "25"
|
|
|
|
await trigger_health_update(
|
|
hass, devices, "96a5ef74-5832-a84b-f1f7-ca799957065d", HealthStatus.OFFLINE
|
|
)
|
|
|
|
assert (
|
|
hass.states.get("sensor.ac_office_granit_temperature").state
|
|
== STATE_UNAVAILABLE
|
|
)
|
|
|
|
await trigger_health_update(
|
|
hass, devices, "96a5ef74-5832-a84b-f1f7-ca799957065d", HealthStatus.ONLINE
|
|
)
|
|
|
|
assert hass.states.get("sensor.ac_office_granit_temperature").state == "25"
|
|
|
|
|
|
@pytest.mark.parametrize("device_fixture", ["da_ac_rac_000001"])
|
|
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("sensor.ac_office_granit_temperature").state
|
|
== STATE_UNAVAILABLE
|
|
)
|