c71a7b901f
* Start gateway using new zigpy init. Update config entry data import. Use new zigpy startup. Fix config entry import without zha config section. Auto form Zigbee network. * Migrate config entry. * New ZHA config entry flow. Use lightweight probe() method for ZHA config entry validation when available. Failback to old behavior of setting up Zigpy app if radio lib does not provide probing. * Clean ZHA_GW_RADIO * Don't import ZHA device settings. * Update config flow tests. * Filter out empty manufacturer. * Replace port path with an by-id device name. * Rebase cleanup * Use correct mock. * Make lint happy again * Update tests/components/zha/test_config_flow.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * Update tests/components/zha/test_config_flow.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * Update tests/components/zha/test_config_flow.py Co-authored-by: Martin Hjelmare <marhje52@gmail.com> * Use executor pool for IO * Address comments. Use AsyncMock from tests. * Use core interface to test config flow. * Use core interface to test config_flow. * Address comments. Use core interface. * Update ZHA dependencies. * Schema guard * Use async_update_entry for migration. * Don't allow schema extra keys. Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
72 lines
2.9 KiB
Python
72 lines
2.9 KiB
Python
"""Tests for ZHA integration init."""
|
|
|
|
import pytest
|
|
from zigpy.config import CONF_DEVICE, CONF_DEVICE_PATH
|
|
|
|
from homeassistant.components.zha.core.const import (
|
|
CONF_BAUDRATE,
|
|
CONF_RADIO_TYPE,
|
|
CONF_USB_PATH,
|
|
DOMAIN,
|
|
)
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.async_mock import AsyncMock, patch
|
|
from tests.common import MockConfigEntry
|
|
|
|
DATA_RADIO_TYPE = "deconz"
|
|
DATA_PORT_PATH = "/dev/serial/by-id/FTDI_USB__-__Serial_Cable_12345678-if00-port0"
|
|
|
|
|
|
@pytest.fixture
|
|
def config_entry_v1(hass):
|
|
"""Config entry version 1 fixture."""
|
|
return MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data={CONF_RADIO_TYPE: DATA_RADIO_TYPE, CONF_USB_PATH: DATA_PORT_PATH},
|
|
version=1,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("config", ({}, {DOMAIN: {}}))
|
|
@patch("homeassistant.components.zha.async_setup_entry", AsyncMock(return_value=True))
|
|
async def test_migration_from_v1_no_baudrate(hass, config_entry_v1, config):
|
|
"""Test migration of config entry from v1."""
|
|
config_entry_v1.add_to_hass(hass)
|
|
assert await async_setup_component(hass, DOMAIN, config)
|
|
|
|
assert config_entry_v1.data[CONF_RADIO_TYPE] == DATA_RADIO_TYPE
|
|
assert CONF_DEVICE in config_entry_v1.data
|
|
assert config_entry_v1.data[CONF_DEVICE][CONF_DEVICE_PATH] == DATA_PORT_PATH
|
|
assert CONF_BAUDRATE not in config_entry_v1.data[CONF_DEVICE]
|
|
assert CONF_USB_PATH not in config_entry_v1.data
|
|
assert config_entry_v1.version == 2
|
|
|
|
|
|
@patch("homeassistant.components.zha.async_setup_entry", AsyncMock(return_value=True))
|
|
async def test_migration_from_v1_with_baudrate(hass, config_entry_v1):
|
|
"""Test migration of config entry from v1 with baudrate in config."""
|
|
config_entry_v1.add_to_hass(hass)
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {CONF_BAUDRATE: 115200}})
|
|
|
|
assert config_entry_v1.data[CONF_RADIO_TYPE] == DATA_RADIO_TYPE
|
|
assert CONF_DEVICE in config_entry_v1.data
|
|
assert config_entry_v1.data[CONF_DEVICE][CONF_DEVICE_PATH] == DATA_PORT_PATH
|
|
assert CONF_USB_PATH not in config_entry_v1.data
|
|
assert CONF_BAUDRATE in config_entry_v1.data[CONF_DEVICE]
|
|
assert config_entry_v1.data[CONF_DEVICE][CONF_BAUDRATE] == 115200
|
|
assert config_entry_v1.version == 2
|
|
|
|
|
|
@patch("homeassistant.components.zha.async_setup_entry", AsyncMock(return_value=True))
|
|
async def test_migration_from_v1_wrong_baudrate(hass, config_entry_v1):
|
|
"""Test migration of config entry from v1 with wrong baudrate."""
|
|
config_entry_v1.add_to_hass(hass)
|
|
assert await async_setup_component(hass, DOMAIN, {DOMAIN: {CONF_BAUDRATE: 115222}})
|
|
|
|
assert config_entry_v1.data[CONF_RADIO_TYPE] == DATA_RADIO_TYPE
|
|
assert CONF_DEVICE in config_entry_v1.data
|
|
assert config_entry_v1.data[CONF_DEVICE][CONF_DEVICE_PATH] == DATA_PORT_PATH
|
|
assert CONF_USB_PATH not in config_entry_v1.data
|
|
assert CONF_BAUDRATE not in config_entry_v1.data[CONF_DEVICE]
|
|
assert config_entry_v1.version == 2
|