mirror of
https://github.com/home-assistant/core.git
synced 2025-07-28 23:57:06 +00:00
30 lines
633 B
Python
30 lines
633 B
Python
"""
|
|
custom_components.switch.test
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Provides a mock switch platform.
|
|
|
|
Call init before using it in your tests to ensure clean test data.
|
|
"""
|
|
from homeassistant.const import STATE_ON, STATE_OFF
|
|
from tests.helpers import MockToggleDevice
|
|
|
|
|
|
DEVICES = []
|
|
|
|
|
|
def init(empty=False):
|
|
""" (re-)initalizes the platform with devices. """
|
|
global DEVICES
|
|
|
|
DEVICES = [] if empty else [
|
|
MockToggleDevice('AC', STATE_ON),
|
|
MockToggleDevice('AC', STATE_OFF),
|
|
MockToggleDevice(None, STATE_OFF)
|
|
]
|
|
|
|
|
|
def get_switches(hass, config):
|
|
""" Returns mock devices. """
|
|
return DEVICES
|