66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
"""Common code for tests."""
|
|
|
|
from typing import Callable, NamedTuple, Tuple
|
|
|
|
from mock import MagicMock
|
|
from pyvera import VeraController, VeraDevice, VeraScene
|
|
|
|
from homeassistant.components.vera import CONF_CONTROLLER, DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
|
|
class ComponentData(NamedTuple):
|
|
"""Component data."""
|
|
|
|
controller: VeraController
|
|
|
|
|
|
class ComponentFactory:
|
|
"""Factory class."""
|
|
|
|
def __init__(self, init_controller_mock):
|
|
"""Initialize component factory."""
|
|
self.init_controller_mock = init_controller_mock
|
|
|
|
async def configure_component(
|
|
self,
|
|
hass: HomeAssistant,
|
|
devices: Tuple[VeraDevice] = (),
|
|
scenes: Tuple[VeraScene] = (),
|
|
setup_callback: Callable[[VeraController], None] = None,
|
|
) -> ComponentData:
|
|
"""Configure the component with specific mock data."""
|
|
controller_url = "http://127.0.0.1:123"
|
|
|
|
hass_config = {
|
|
DOMAIN: {CONF_CONTROLLER: controller_url},
|
|
}
|
|
|
|
controller = MagicMock(spec=VeraController) # type: VeraController
|
|
controller.base_url = controller_url
|
|
controller.register = MagicMock()
|
|
controller.get_devices = MagicMock(return_value=devices or ())
|
|
controller.get_scenes = MagicMock(return_value=scenes or ())
|
|
|
|
for vera_obj in controller.get_devices() + controller.get_scenes():
|
|
vera_obj.vera_controller = controller
|
|
|
|
controller.get_devices.reset_mock()
|
|
controller.get_scenes.reset_mock()
|
|
|
|
if setup_callback:
|
|
setup_callback(controller, hass_config)
|
|
|
|
def init_controller(base_url: str) -> list:
|
|
nonlocal controller
|
|
return [controller, True]
|
|
|
|
self.init_controller_mock.side_effect = init_controller
|
|
|
|
# Setup Home Assistant.
|
|
assert await async_setup_component(hass, DOMAIN, hass_config)
|
|
await hass.async_block_till_done()
|
|
|
|
return ComponentData(controller=controller)
|