6ef53a2c5d
* Add support for config_flow Add support for switches Add support for binary sensors * Add config flow strings * Update .coveragerc * black-formatting * fixes for isort and flake * fixes for pylint * fixes for isort * fixes for isort * fixes for config_flow * Add devices to Device Registry * Updated comments * fixes for flake8 * Updated comments * Updated comments * Implement async_unload_entry * remove binary_sensor and switch implementation (will move to new PR) * Update .coveragerc Co-authored-by: Chris Talkington <chris@talkingtontech.com> * Update .coveragerc * Review config flow to store the configuration type * Add config_flow tests * Move CONF_NAMES to constants * Fix isort * Tweak to onewire logger * Tweak to onewire logger for sensor * Reset _LOGGER variable * Normalise header * Normalise header * Update to use references in config flow translations * Improve test coverage * fixes for isort * Update async_unload_entry * Update common strings * Update imports * Move connect attempt to executor * Update common strings * Remove OWFS from config_flow * Prevent duplicate config entries * Fix isort * Fix flake8 * Fix tests following removal of OWFS implementation * Fix flake8 * Adjust config from yaml to config-flow, and add ability to specify sysbus directory * Ajust unique_id for config entries * Fix test_config_flow * Fix invalid merge * Convert yaml to config_entry * Update sysbus tests * Tweaks to yaml import process, and add OWFS warning * Enable migration of OWFS platform config to config_entry * update the existing corresponding config entry on OWFS conversion * Remove CONFIG_SCHEMA * Move data_schema to constants * Remove log message * Remove duplicate warning * Update already_configured to use already_configured_device constant * Schedule get_entities on the executor * Update duplicate entry check for OWServer * Review TryCatch * Schedule os.path.isdir on the executor * rename owhost/owport * Update checks for empty * Fix incorrect patch * Move config_flow validation methods to new OneWireHub * Fix typo and pre-commit * Cleanup try/else * patch async_setup/async_setup_entry * cleanup implicit exit point * Fix invalid patch * cleanup implicit exit point * cleanup implicit exit point Co-authored-by: Chris Talkington <chris@talkingtontech.com>
333 lines
11 KiB
Python
333 lines
11 KiB
Python
"""Tests for 1-Wire config flow."""
|
|
from pyownet import protocol
|
|
|
|
from homeassistant.components.onewire.const import (
|
|
CONF_MOUNT_DIR,
|
|
CONF_TYPE_OWSERVER,
|
|
CONF_TYPE_SYSBUS,
|
|
DEFAULT_OWSERVER_PORT,
|
|
DEFAULT_SYSBUS_MOUNT_DIR,
|
|
DOMAIN,
|
|
)
|
|
from homeassistant.config_entries import SOURCE_IMPORT, SOURCE_USER
|
|
from homeassistant.const import CONF_HOST, CONF_PORT, CONF_TYPE
|
|
from homeassistant.data_entry_flow import (
|
|
RESULT_TYPE_ABORT,
|
|
RESULT_TYPE_CREATE_ENTRY,
|
|
RESULT_TYPE_FORM,
|
|
)
|
|
|
|
from . import setup_onewire_owserver_integration, setup_onewire_sysbus_integration
|
|
|
|
from tests.async_mock import patch
|
|
|
|
|
|
async def test_user_owserver(hass):
|
|
"""Test OWServer user flow."""
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": SOURCE_USER}
|
|
)
|
|
assert result["type"] == RESULT_TYPE_FORM
|
|
assert not result["errors"]
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_TYPE: CONF_TYPE_OWSERVER},
|
|
)
|
|
|
|
assert result["type"] == RESULT_TYPE_FORM
|
|
assert result["step_id"] == "owserver"
|
|
assert not result["errors"]
|
|
|
|
# Invalid server
|
|
with patch(
|
|
"homeassistant.components.onewire.onewirehub.protocol.proxy",
|
|
side_effect=protocol.ConnError,
|
|
):
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_HOST: "1.2.3.4", CONF_PORT: 1234},
|
|
)
|
|
|
|
assert result["type"] == RESULT_TYPE_FORM
|
|
assert result["step_id"] == "owserver"
|
|
assert result["errors"] == {"base": "cannot_connect"}
|
|
|
|
# Valid server
|
|
with patch("homeassistant.components.onewire.onewirehub.protocol.proxy",), patch(
|
|
"homeassistant.components.onewire.async_setup", return_value=True
|
|
) as mock_setup, patch(
|
|
"homeassistant.components.onewire.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry:
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_HOST: "1.2.3.4", CONF_PORT: 1234},
|
|
)
|
|
|
|
assert result["type"] == RESULT_TYPE_CREATE_ENTRY
|
|
assert result["title"] == "1.2.3.4"
|
|
assert result["data"] == {
|
|
CONF_TYPE: CONF_TYPE_OWSERVER,
|
|
CONF_HOST: "1.2.3.4",
|
|
CONF_PORT: 1234,
|
|
}
|
|
await hass.async_block_till_done()
|
|
assert len(mock_setup.mock_calls) == 1
|
|
assert len(mock_setup_entry.mock_calls) == 1
|
|
|
|
|
|
async def test_user_owserver_duplicate(hass):
|
|
"""Test OWServer flow."""
|
|
with patch(
|
|
"homeassistant.components.onewire.async_setup", return_value=True
|
|
) as mock_setup, patch(
|
|
"homeassistant.components.onewire.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry:
|
|
await setup_onewire_owserver_integration(hass)
|
|
assert len(hass.config_entries.async_entries(DOMAIN)) == 1
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": SOURCE_USER}
|
|
)
|
|
assert result["type"] == RESULT_TYPE_FORM
|
|
assert not result["errors"]
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_TYPE: CONF_TYPE_OWSERVER},
|
|
)
|
|
|
|
assert result["type"] == RESULT_TYPE_FORM
|
|
assert result["step_id"] == "owserver"
|
|
assert not result["errors"]
|
|
|
|
# Duplicate server
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_HOST: "1.2.3.4", CONF_PORT: 1234},
|
|
)
|
|
assert result["type"] == RESULT_TYPE_ABORT
|
|
assert result["reason"] == "already_configured"
|
|
await hass.async_block_till_done()
|
|
assert len(mock_setup.mock_calls) == 1
|
|
assert len(mock_setup_entry.mock_calls) == 1
|
|
|
|
|
|
async def test_user_sysbus(hass):
|
|
"""Test SysBus flow."""
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": SOURCE_USER}
|
|
)
|
|
assert result["type"] == RESULT_TYPE_FORM
|
|
assert not result["errors"]
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_TYPE: CONF_TYPE_SYSBUS},
|
|
)
|
|
|
|
assert result["type"] == RESULT_TYPE_FORM
|
|
assert result["step_id"] == "mount_dir"
|
|
assert not result["errors"]
|
|
|
|
# Invalid path
|
|
with patch(
|
|
"homeassistant.components.onewire.onewirehub.os.path.isdir",
|
|
return_value=False,
|
|
):
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_MOUNT_DIR: "/sys/bus/invalid_directory"},
|
|
)
|
|
|
|
assert result["type"] == RESULT_TYPE_FORM
|
|
assert result["step_id"] == "mount_dir"
|
|
assert result["errors"] == {"base": "invalid_path"}
|
|
|
|
# Valid path
|
|
with patch(
|
|
"homeassistant.components.onewire.onewirehub.os.path.isdir",
|
|
return_value=True,
|
|
), patch(
|
|
"homeassistant.components.onewire.async_setup", return_value=True
|
|
) as mock_setup, patch(
|
|
"homeassistant.components.onewire.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry:
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_MOUNT_DIR: "/sys/bus/directory"},
|
|
)
|
|
|
|
assert result["type"] == RESULT_TYPE_CREATE_ENTRY
|
|
assert result["title"] == "/sys/bus/directory"
|
|
assert result["data"] == {
|
|
CONF_TYPE: CONF_TYPE_SYSBUS,
|
|
CONF_MOUNT_DIR: "/sys/bus/directory",
|
|
}
|
|
await hass.async_block_till_done()
|
|
assert len(mock_setup.mock_calls) == 1
|
|
assert len(mock_setup_entry.mock_calls) == 1
|
|
|
|
|
|
async def test_user_sysbus_duplicate(hass):
|
|
"""Test SysBus duplicate flow."""
|
|
with patch(
|
|
"homeassistant.components.onewire.async_setup", return_value=True
|
|
) as mock_setup, patch(
|
|
"homeassistant.components.onewire.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry:
|
|
await setup_onewire_sysbus_integration(hass)
|
|
assert len(hass.config_entries.async_entries(DOMAIN)) == 1
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": SOURCE_USER}
|
|
)
|
|
assert result["type"] == RESULT_TYPE_FORM
|
|
assert not result["errors"]
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_TYPE: CONF_TYPE_SYSBUS},
|
|
)
|
|
|
|
assert result["type"] == RESULT_TYPE_FORM
|
|
assert result["step_id"] == "mount_dir"
|
|
assert not result["errors"]
|
|
|
|
# Valid path
|
|
with patch(
|
|
"homeassistant.components.onewire.onewirehub.os.path.isdir",
|
|
return_value=True,
|
|
):
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_MOUNT_DIR: DEFAULT_SYSBUS_MOUNT_DIR},
|
|
)
|
|
|
|
assert result["type"] == RESULT_TYPE_ABORT
|
|
assert result["reason"] == "already_configured"
|
|
await hass.async_block_till_done()
|
|
assert len(mock_setup.mock_calls) == 1
|
|
assert len(mock_setup_entry.mock_calls) == 1
|
|
|
|
|
|
async def test_import_sysbus(hass):
|
|
"""Test import step."""
|
|
|
|
with patch(
|
|
"homeassistant.components.onewire.onewirehub.os.path.isdir",
|
|
return_value=True,
|
|
), patch(
|
|
"homeassistant.components.onewire.async_setup", return_value=True
|
|
) as mock_setup, patch(
|
|
"homeassistant.components.onewire.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry:
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": SOURCE_IMPORT},
|
|
data={CONF_TYPE: CONF_TYPE_SYSBUS},
|
|
)
|
|
assert result["type"] == RESULT_TYPE_CREATE_ENTRY
|
|
assert result["title"] == DEFAULT_SYSBUS_MOUNT_DIR
|
|
assert result["data"] == {
|
|
CONF_TYPE: CONF_TYPE_SYSBUS,
|
|
CONF_MOUNT_DIR: DEFAULT_SYSBUS_MOUNT_DIR,
|
|
}
|
|
await hass.async_block_till_done()
|
|
assert len(mock_setup.mock_calls) == 1
|
|
assert len(mock_setup_entry.mock_calls) == 1
|
|
|
|
|
|
async def test_import_sysbus_with_mount_dir(hass):
|
|
"""Test import step."""
|
|
|
|
with patch(
|
|
"homeassistant.components.onewire.onewirehub.os.path.isdir",
|
|
return_value=True,
|
|
), patch(
|
|
"homeassistant.components.onewire.async_setup", return_value=True
|
|
) as mock_setup, patch(
|
|
"homeassistant.components.onewire.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry:
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": SOURCE_IMPORT},
|
|
data={
|
|
CONF_TYPE: CONF_TYPE_SYSBUS,
|
|
CONF_MOUNT_DIR: DEFAULT_SYSBUS_MOUNT_DIR,
|
|
},
|
|
)
|
|
assert result["type"] == RESULT_TYPE_CREATE_ENTRY
|
|
assert result["title"] == DEFAULT_SYSBUS_MOUNT_DIR
|
|
assert result["data"] == {
|
|
CONF_TYPE: CONF_TYPE_SYSBUS,
|
|
CONF_MOUNT_DIR: DEFAULT_SYSBUS_MOUNT_DIR,
|
|
}
|
|
await hass.async_block_till_done()
|
|
assert len(mock_setup.mock_calls) == 1
|
|
assert len(mock_setup_entry.mock_calls) == 1
|
|
|
|
|
|
async def test_import_owserver(hass):
|
|
"""Test import step."""
|
|
|
|
with patch("homeassistant.components.onewire.onewirehub.protocol.proxy",), patch(
|
|
"homeassistant.components.onewire.async_setup", return_value=True
|
|
) as mock_setup, patch(
|
|
"homeassistant.components.onewire.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry:
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": SOURCE_IMPORT},
|
|
data={
|
|
CONF_TYPE: CONF_TYPE_OWSERVER,
|
|
CONF_HOST: "1.2.3.4",
|
|
},
|
|
)
|
|
assert result["type"] == RESULT_TYPE_CREATE_ENTRY
|
|
assert result["title"] == "1.2.3.4"
|
|
assert result["data"] == {
|
|
CONF_TYPE: CONF_TYPE_OWSERVER,
|
|
CONF_HOST: "1.2.3.4",
|
|
CONF_PORT: DEFAULT_OWSERVER_PORT,
|
|
}
|
|
await hass.async_block_till_done()
|
|
assert len(mock_setup.mock_calls) == 1
|
|
assert len(mock_setup_entry.mock_calls) == 1
|
|
|
|
|
|
async def test_import_owserver_with_port(hass):
|
|
"""Test import step."""
|
|
|
|
with patch("homeassistant.components.onewire.onewirehub.protocol.proxy",), patch(
|
|
"homeassistant.components.onewire.async_setup", return_value=True
|
|
) as mock_setup, patch(
|
|
"homeassistant.components.onewire.async_setup_entry",
|
|
return_value=True,
|
|
) as mock_setup_entry:
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": SOURCE_IMPORT},
|
|
data={
|
|
CONF_TYPE: CONF_TYPE_OWSERVER,
|
|
CONF_HOST: "1.2.3.4",
|
|
CONF_PORT: "1234",
|
|
},
|
|
)
|
|
assert result["type"] == RESULT_TYPE_CREATE_ENTRY
|
|
assert result["title"] == "1.2.3.4"
|
|
assert result["data"] == {
|
|
CONF_TYPE: CONF_TYPE_OWSERVER,
|
|
CONF_HOST: "1.2.3.4",
|
|
CONF_PORT: "1234",
|
|
}
|
|
await hass.async_block_till_done()
|
|
assert len(mock_setup.mock_calls) == 1
|
|
assert len(mock_setup_entry.mock_calls) == 1
|