core/tests/components/smartthings/test_switch.py

100 lines
3.7 KiB
Python
Raw Normal View History

"""
Test for the SmartThings switch platform.
The only mocking required is of the underlying SmartThings API object so
real HTTP calls are not initiated during testing.
"""
from pysmartthings import Attribute, Capability
2019-07-31 19:25:30 +00:00
from homeassistant.components.smartthings.const import DOMAIN, SIGNAL_SMARTTHINGS_UPDATE
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
from homeassistant.config_entries import ConfigEntryState
from homeassistant.const import STATE_UNAVAILABLE
from homeassistant.helpers import device_registry as dr, entity_registry as er
from homeassistant.helpers.dispatcher import async_dispatcher_send
from .conftest import setup_platform
async def test_entity_and_device_attributes(hass, device_factory):
"""Test the attributes of the entity are correct."""
# Arrange
2019-07-31 19:25:30 +00:00
device = device_factory("Switch_1", [Capability.switch], {Attribute.switch: "on"})
entity_registry = er.async_get(hass)
device_registry = dr.async_get(hass)
# Act
await setup_platform(hass, SWITCH_DOMAIN, devices=[device])
# Assert
2019-07-31 19:25:30 +00:00
entry = entity_registry.async_get("switch.switch_1")
assert entry
assert entry.unique_id == device.device_id
entry = device_registry.async_get_device({(DOMAIN, device.device_id)})
assert entry
assert entry.configuration_url == "https://account.smartthings.com"
assert entry.identifiers == {(DOMAIN, device.device_id)}
assert entry.name == device.label
assert entry.model == device.device_type_name
2019-07-31 19:25:30 +00:00
assert entry.manufacturer == "Unavailable"
async def test_turn_off(hass, device_factory):
"""Test the switch turns of successfully."""
# Arrange
2019-07-31 19:25:30 +00:00
device = device_factory("Switch_1", [Capability.switch], {Attribute.switch: "on"})
await setup_platform(hass, SWITCH_DOMAIN, devices=[device])
# Act
await hass.services.async_call(
2019-07-31 19:25:30 +00:00
"switch", "turn_off", {"entity_id": "switch.switch_1"}, blocking=True
)
# Assert
2019-07-31 19:25:30 +00:00
state = hass.states.get("switch.switch_1")
assert state is not None
2019-07-31 19:25:30 +00:00
assert state.state == "off"
async def test_turn_on(hass, device_factory):
"""Test the switch turns of successfully."""
# Arrange
2019-07-31 19:25:30 +00:00
device = device_factory(
"Switch_1",
[Capability.switch, Capability.power_meter, Capability.energy_meter],
{Attribute.switch: "off", Attribute.power: 355, Attribute.energy: 11.422},
)
await setup_platform(hass, SWITCH_DOMAIN, devices=[device])
# Act
await hass.services.async_call(
2019-07-31 19:25:30 +00:00
"switch", "turn_on", {"entity_id": "switch.switch_1"}, blocking=True
)
# Assert
2019-07-31 19:25:30 +00:00
state = hass.states.get("switch.switch_1")
assert state is not None
2019-07-31 19:25:30 +00:00
assert state.state == "on"
async def test_update_from_signal(hass, device_factory):
"""Test the switch updates when receiving a signal."""
# Arrange
2019-07-31 19:25:30 +00:00
device = device_factory("Switch_1", [Capability.switch], {Attribute.switch: "off"})
await setup_platform(hass, SWITCH_DOMAIN, devices=[device])
await device.switch_on(True)
# Act
2019-07-31 19:25:30 +00:00
async_dispatcher_send(hass, SIGNAL_SMARTTHINGS_UPDATE, [device.device_id])
# Assert
await hass.async_block_till_done()
2019-07-31 19:25:30 +00:00
state = hass.states.get("switch.switch_1")
assert state is not None
2019-07-31 19:25:30 +00:00
assert state.state == "on"
async def test_unload_config_entry(hass, device_factory):
"""Test the switch is removed when the config entry is unloaded."""
# Arrange
2019-07-31 19:25:30 +00:00
device = device_factory("Switch 1", [Capability.switch], {Attribute.switch: "on"})
config_entry = await setup_platform(hass, SWITCH_DOMAIN, devices=[device])
config_entry.state = ConfigEntryState.LOADED
# Act
2019-07-31 19:25:30 +00:00
await hass.config_entries.async_forward_entry_unload(config_entry, "switch")
# Assert
assert hass.states.get("switch.switch_1").state == STATE_UNAVAILABLE