79d37fdf12
* Add bulb 6 multi color device state fixture * Add light test foundation * Add no cover comment for todo code * Update hs_color * Test turn on light * Test light turn off * Fix brightness comparison * Test setting same brightness * Test setting same rgb color * Test color temp update * Test setting same color temp * Add entity module to coverage calculation * Fix typing
97 lines
3.1 KiB
Python
97 lines
3.1 KiB
Python
"""Provide common Z-Wave JS fixtures."""
|
|
import json
|
|
from unittest.mock import DEFAULT, patch
|
|
|
|
import pytest
|
|
from zwave_js_server.model.driver import Driver
|
|
from zwave_js_server.model.node import Node
|
|
|
|
from homeassistant.helpers.device_registry import (
|
|
async_get_registry as async_get_device_registry,
|
|
)
|
|
|
|
from tests.common import MockConfigEntry, load_fixture
|
|
|
|
|
|
@pytest.fixture(name="device_registry")
|
|
async def device_registry_fixture(hass):
|
|
"""Return the device registry."""
|
|
return await async_get_device_registry(hass)
|
|
|
|
|
|
@pytest.fixture(name="controller_state", scope="session")
|
|
def controller_state_fixture():
|
|
"""Load the controller state fixture data."""
|
|
return json.loads(load_fixture("zwave_js/controller_state.json"))
|
|
|
|
|
|
@pytest.fixture(name="multisensor_6_state", scope="session")
|
|
def multisensor_6_state_fixture():
|
|
"""Load the multisensor 6 node state fixture data."""
|
|
return json.loads(load_fixture("zwave_js/multisensor_6_state.json"))
|
|
|
|
|
|
@pytest.fixture(name="hank_binary_switch_state", scope="session")
|
|
def binary_switch_state_fixture():
|
|
"""Load the hank binary switch node state fixture data."""
|
|
return json.loads(load_fixture("zwave_js/hank_binary_switch_state.json"))
|
|
|
|
|
|
@pytest.fixture(name="bulb_6_multi_color_state", scope="session")
|
|
def bulb_6_multi_color_state_fixture():
|
|
"""Load the bulb 6 multi-color node state fixture data."""
|
|
return json.loads(load_fixture("zwave_js/bulb_6_multi_color_state.json"))
|
|
|
|
|
|
@pytest.fixture(name="client")
|
|
def mock_client_fixture(controller_state):
|
|
"""Mock a client."""
|
|
with patch(
|
|
"homeassistant.components.zwave_js.ZwaveClient", autospec=True
|
|
) as client_class:
|
|
driver = Driver(client_class.return_value, controller_state)
|
|
client_class.return_value.driver = driver
|
|
yield client_class.return_value
|
|
|
|
|
|
@pytest.fixture(name="multisensor_6")
|
|
def multisensor_6_fixture(client, multisensor_6_state):
|
|
"""Mock a multisensor 6 node."""
|
|
node = Node(client, multisensor_6_state)
|
|
client.driver.controller.nodes[node.node_id] = node
|
|
return node
|
|
|
|
|
|
@pytest.fixture(name="hank_binary_switch")
|
|
def hank_binary_switch_fixture(client, hank_binary_switch_state):
|
|
"""Mock a binary switch node."""
|
|
node = Node(client, hank_binary_switch_state)
|
|
client.driver.controller.nodes[node.node_id] = node
|
|
return node
|
|
|
|
|
|
@pytest.fixture(name="bulb_6_multi_color")
|
|
def bulb_6_multi_color_fixture(client, bulb_6_multi_color_state):
|
|
"""Mock a bulb 6 multi-color node."""
|
|
node = Node(client, bulb_6_multi_color_state)
|
|
client.driver.controller.nodes[node.node_id] = node
|
|
return node
|
|
|
|
|
|
@pytest.fixture(name="integration")
|
|
async def integration_fixture(hass, client):
|
|
"""Set up the zwave_js integration."""
|
|
entry = MockConfigEntry(domain="zwave_js", data={"url": "ws://test.org"})
|
|
entry.add_to_hass(hass)
|
|
|
|
def initialize_client(async_on_initialized):
|
|
"""Init the client."""
|
|
hass.async_create_task(async_on_initialized())
|
|
return DEFAULT
|
|
|
|
client.register_on_initialized.side_effect = initialize_client
|
|
|
|
await hass.config_entries.async_setup(entry.entry_id)
|
|
await hass.async_block_till_done()
|
|
|
|
return entry
|