56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
"""
|
|
test.mock.switch_platform
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Provides a mock switch platform.
|
|
"""
|
|
import homeassistant.components as components
|
|
|
|
|
|
class MockSwitch(components.ToggleDevice):
|
|
""" Fake switch. """
|
|
def __init__(self, name, state):
|
|
self.name = name
|
|
self.state = state
|
|
|
|
def get_name(self):
|
|
""" Returns the name of the device if any. """
|
|
return self.name
|
|
|
|
def turn_on(self, **kwargs):
|
|
""" Turn the device on. """
|
|
self.state = components.STATE_ON
|
|
|
|
def turn_off(self, **kwargs):
|
|
""" Turn the device off. """
|
|
self.state = components.STATE_OFF
|
|
|
|
def is_on(self):
|
|
""" True if device is on. """
|
|
return self.state == components.STATE_ON
|
|
|
|
def get_state_attributes(self):
|
|
""" Returns optional state attributes. """
|
|
return {}
|
|
|
|
|
|
FAKE_NO_SWITCHES = False
|
|
|
|
SWITCHES = [
|
|
MockSwitch('AC', components.STATE_ON),
|
|
MockSwitch('AC', components.STATE_OFF),
|
|
MockSwitch(None, components.STATE_OFF)
|
|
]
|
|
|
|
|
|
def fake_no_switches(do_fake):
|
|
""" Set the platform to act as if it has no switches. """
|
|
global FAKE_NO_SWITCHES
|
|
|
|
FAKE_NO_SWITCHES = do_fake
|
|
|
|
|
|
def get_switches(hass, config):
|
|
""" Returns mock switches. """
|
|
return [] if FAKE_NO_SWITCHES else SWITCHES
|