mirror of https://github.com/home-assistant/core
89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
"""The tests for the litejet component."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from serial import SerialException
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components.litejet.const import CONF_DEFAULT_TRANSITION, DOMAIN
|
|
from homeassistant.const import CONF_PORT
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.data_entry_flow import FlowResultType
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
|
|
async def test_show_config_form(hass: HomeAssistant) -> None:
|
|
"""Test show configuration form."""
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["step_id"] == "user"
|
|
|
|
|
|
async def test_create_entry(hass: HomeAssistant, mock_litejet) -> None:
|
|
"""Test create entry from user input."""
|
|
test_data = {CONF_PORT: "/dev/test"}
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": config_entries.SOURCE_USER}, data=test_data
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result["title"] == "/dev/test"
|
|
assert result["data"] == test_data
|
|
|
|
|
|
async def test_flow_entry_already_exists(hass: HomeAssistant) -> None:
|
|
"""Test user input when a config entry already exists."""
|
|
first_entry = MockConfigEntry(
|
|
domain=DOMAIN,
|
|
data={CONF_PORT: "/dev/first"},
|
|
)
|
|
first_entry.add_to_hass(hass)
|
|
|
|
test_data = {CONF_PORT: "/dev/test"}
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": config_entries.SOURCE_USER}, data=test_data
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "single_instance_allowed"
|
|
|
|
|
|
async def test_flow_open_failed(hass: HomeAssistant) -> None:
|
|
"""Test user input when serial port open fails."""
|
|
test_data = {CONF_PORT: "/dev/test"}
|
|
|
|
with patch("pylitejet.LiteJet") as mock_pylitejet:
|
|
mock_pylitejet.side_effect = SerialException
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN, context={"source": config_entries.SOURCE_USER}, data=test_data
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"][CONF_PORT] == "open_failed"
|
|
|
|
|
|
async def test_options(hass: HomeAssistant) -> None:
|
|
"""Test updating options."""
|
|
entry = MockConfigEntry(domain=DOMAIN, data={CONF_PORT: "/dev/test"})
|
|
entry.add_to_hass(hass)
|
|
|
|
result = await hass.config_entries.options.async_init(entry.entry_id)
|
|
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["step_id"] == "init"
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={CONF_DEFAULT_TRANSITION: 12},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
|
assert result["data"] == {CONF_DEFAULT_TRANSITION: 12}
|