e577f047f7
* Adding tests for vera component. Fixing update bug in the vera climate platform. * Updating requrements file. * Moving vera stop to a job. Sorting imports. * Addressing simple PR feedback. * Splitting tests into platforms. * Mocking controller instead of using requests_mock. * Updating pyvera to use version that stops threads quickly. * Updating requirements files. * Mocking the pyvera module, not the API. * Addressing PR feedback. Handling start/stop of patch in fixture. Removing unecessary code. * Using generator
61 lines
2 KiB
Python
61 lines
2 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
|
|
|
|
ComponentData = NamedTuple("ComponentData", (("controller", VeraController),))
|
|
|
|
|
|
class ComponentFactory:
|
|
"""Factory class."""
|
|
|
|
def __init__(self, init_controller_mock):
|
|
"""Constructor."""
|
|
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)
|