71a0a7fe00
* Use `setup_test_component_platform` helper for switch entity component tests instead of `hass.components` * Do not import fixtures * Re-add switch.py to testing_config as stub * Rename to mock_toggle_entities
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
"""Collection of helper methods.
|
|
|
|
All containing methods are legacy helpers that should not be used by new
|
|
components. Instead call the service directly.
|
|
"""
|
|
|
|
from homeassistant.components.switch import DOMAIN
|
|
from homeassistant.const import (
|
|
ATTR_ENTITY_ID,
|
|
ENTITY_MATCH_ALL,
|
|
SERVICE_TURN_OFF,
|
|
SERVICE_TURN_ON,
|
|
STATE_OFF,
|
|
STATE_ON,
|
|
)
|
|
from homeassistant.loader import bind_hass
|
|
|
|
from tests.common import MockToggleEntity
|
|
|
|
|
|
@bind_hass
|
|
def turn_on(hass, entity_id=ENTITY_MATCH_ALL):
|
|
"""Turn all or specified switch on."""
|
|
hass.add_job(async_turn_on, hass, entity_id)
|
|
|
|
|
|
async def async_turn_on(hass, entity_id=ENTITY_MATCH_ALL):
|
|
"""Turn all or specified switch on."""
|
|
data = {ATTR_ENTITY_ID: entity_id} if entity_id else None
|
|
await hass.services.async_call(DOMAIN, SERVICE_TURN_ON, data, blocking=True)
|
|
|
|
|
|
@bind_hass
|
|
def turn_off(hass, entity_id=ENTITY_MATCH_ALL):
|
|
"""Turn all or specified switch off."""
|
|
hass.add_job(async_turn_off, hass, entity_id)
|
|
|
|
|
|
async def async_turn_off(hass, entity_id=ENTITY_MATCH_ALL):
|
|
"""Turn all or specified switch off."""
|
|
data = {ATTR_ENTITY_ID: entity_id} if entity_id else None
|
|
await hass.services.async_call(DOMAIN, SERVICE_TURN_OFF, data, blocking=True)
|
|
|
|
|
|
def get_mock_toggle_entities() -> list[MockToggleEntity]:
|
|
"""Return a list of mock toggle entities."""
|
|
return [
|
|
MockToggleEntity("AC", STATE_ON),
|
|
MockToggleEntity("AC", STATE_OFF),
|
|
MockToggleEntity(None, STATE_OFF),
|
|
]
|