home-assistant-core/test/mock_toggledevice_platform.py

"""
test.mock.switch_platform
~~~~~~~~~~~~~~~~~~~~~~~~~

Provides a mock switch platform.
"""
import homeassistant.components as components


class MockToggleDevice(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_DEVICES = False

DEVICES = [
    MockToggleDevice('AC', components.STATE_ON),
    MockToggleDevice('AC', components.STATE_OFF),
    MockToggleDevice(None, components.STATE_OFF)
]


def fake_no_switches(do_fake):
    """ Set the platform to act as if it has no devices. """
    global FAKE_NO_DEVICES

    FAKE_NO_DEVICES = do_fake


def get_switches(hass, config):
    """ Returns mock devices. """
    return [] if FAKE_NO_DEVICES else DEVICES

get_lights = get_switches