mirror of
https://github.com/home-assistant/core
synced 2024-09-25 00:41:32 +02:00
86 lines
2.2 KiB
Python
86 lines
2.2 KiB
Python
"""
|
|
tests.helper
|
|
~~~~~~~~~~~~~
|
|
|
|
Helper method for writing tests.
|
|
"""
|
|
import os
|
|
|
|
import homeassistant as ha
|
|
from homeassistant.helpers import ToggleDevice
|
|
from homeassistant.const import STATE_ON, STATE_OFF, DEVICE_DEFAULT_NAME
|
|
|
|
|
|
def get_test_home_assistant():
|
|
""" Returns a Home Assistant object pointing at test config dir. """
|
|
hass = ha.HomeAssistant()
|
|
hass.config_dir = os.path.join(os.path.dirname(__file__), "config")
|
|
|
|
return hass
|
|
|
|
|
|
def mock_service(hass, domain, service):
|
|
"""
|
|
Sets up a fake service.
|
|
Returns a list that logs all calls to fake service.
|
|
"""
|
|
calls = []
|
|
|
|
hass.services.register(
|
|
domain, service, lambda call: calls.append(call))
|
|
|
|
return calls
|
|
|
|
|
|
class MockModule(object):
|
|
""" Provides a fake module. """
|
|
|
|
def __init__(self, domain, dependencies=[], setup=None):
|
|
self.DOMAIN = domain
|
|
self.DEPENDENCIES = dependencies
|
|
# Setup a mock setup if none given.
|
|
self.setup = lambda hass, config: False if setup is None else setup
|
|
|
|
|
|
class MockToggleDevice(ToggleDevice):
|
|
""" Provides a mock toggle device. """
|
|
def __init__(self, name, state):
|
|
self._name = name or DEVICE_DEFAULT_NAME
|
|
self._state = state
|
|
self.calls = []
|
|
|
|
@property
|
|
def name(self):
|
|
""" Returns the name of the device if any. """
|
|
self.calls.append(('name', {}))
|
|
return self._name
|
|
|
|
@property
|
|
def state(self):
|
|
""" Returns the name of the device if any. """
|
|
self.calls.append(('state', {}))
|
|
return self._state
|
|
|
|
@property
|
|
def is_on(self):
|
|
""" True if device is on. """
|
|
self.calls.append(('is_on', {}))
|
|
return self._state == STATE_ON
|
|
|
|
def turn_on(self, **kwargs):
|
|
""" Turn the device on. """
|
|
self.calls.append(('turn_on', kwargs))
|
|
self._state = STATE_ON
|
|
|
|
def turn_off(self, **kwargs):
|
|
""" Turn the device off. """
|
|
self.calls.append(('turn_off', kwargs))
|
|
self._state = STATE_OFF
|
|
|
|
def last_call(self, method=None):
|
|
if method is None:
|
|
return self.calls[-1]
|
|
else:
|
|
return next(call for call in reversed(self.calls)
|
|
if call[0] == method)
|