148 lines
4.8 KiB
Python
148 lines
4.8 KiB
Python
"""Test the NMBS config flow."""
|
|
|
|
from typing import Any
|
|
from unittest.mock import AsyncMock
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components.nmbs.config_flow import CONF_EXCLUDE_VIAS
|
|
from homeassistant.components.nmbs.const import (
|
|
CONF_STATION_FROM,
|
|
CONF_STATION_TO,
|
|
DOMAIN,
|
|
)
|
|
from homeassistant.config_entries import SOURCE_USER
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.data_entry_flow import FlowResultType
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
DUMMY_DATA_IMPORT: dict[str, Any] = {
|
|
"STAT_BRUSSELS_NORTH": "Brussel-Noord/Bruxelles-Nord",
|
|
"STAT_BRUSSELS_CENTRAL": "Brussel-Centraal/Bruxelles-Central",
|
|
"STAT_BRUSSELS_SOUTH": "Brussel-Zuid/Bruxelles-Midi",
|
|
}
|
|
|
|
DUMMY_DATA_ALTERNATIVE_IMPORT: dict[str, Any] = {
|
|
"STAT_BRUSSELS_NORTH": "Brussels-North",
|
|
"STAT_BRUSSELS_CENTRAL": "Brussels-Central",
|
|
"STAT_BRUSSELS_SOUTH": "Brussels-South/Brussels-Midi",
|
|
}
|
|
|
|
DUMMY_DATA: dict[str, Any] = {
|
|
"STAT_BRUSSELS_NORTH": "BE.NMBS.008812005",
|
|
"STAT_BRUSSELS_CENTRAL": "BE.NMBS.008813003",
|
|
"STAT_BRUSSELS_SOUTH": "BE.NMBS.008814001",
|
|
}
|
|
|
|
|
|
async def test_full_flow(
|
|
hass: HomeAssistant, mock_nmbs_client: AsyncMock, mock_setup_entry: AsyncMock
|
|
) -> None:
|
|
"""Test the full flow."""
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": SOURCE_USER},
|
|
)
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"] == {}
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{
|
|
CONF_STATION_FROM: DUMMY_DATA["STAT_BRUSSELS_NORTH"],
|
|
CONF_STATION_TO: DUMMY_DATA["STAT_BRUSSELS_SOUTH"],
|
|
},
|
|
)
|
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
|
assert (
|
|
result["title"]
|
|
== "Train from Brussel-Noord/Bruxelles-Nord to Brussel-Zuid/Bruxelles-Midi"
|
|
)
|
|
assert result["data"] == {
|
|
CONF_STATION_FROM: DUMMY_DATA["STAT_BRUSSELS_NORTH"],
|
|
CONF_STATION_TO: DUMMY_DATA["STAT_BRUSSELS_SOUTH"],
|
|
}
|
|
assert (
|
|
result["result"].unique_id
|
|
== f"{DUMMY_DATA['STAT_BRUSSELS_NORTH']}_{DUMMY_DATA['STAT_BRUSSELS_SOUTH']}"
|
|
)
|
|
|
|
|
|
async def test_same_station(
|
|
hass: HomeAssistant, mock_nmbs_client: AsyncMock, mock_setup_entry: AsyncMock
|
|
) -> None:
|
|
"""Test selecting the same station."""
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": SOURCE_USER},
|
|
)
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"] == {}
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{
|
|
CONF_STATION_FROM: DUMMY_DATA["STAT_BRUSSELS_NORTH"],
|
|
CONF_STATION_TO: DUMMY_DATA["STAT_BRUSSELS_NORTH"],
|
|
},
|
|
)
|
|
assert result["type"] is FlowResultType.FORM
|
|
assert result["errors"] == {"base": "same_station"}
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
{
|
|
CONF_STATION_FROM: DUMMY_DATA["STAT_BRUSSELS_NORTH"],
|
|
CONF_STATION_TO: DUMMY_DATA["STAT_BRUSSELS_SOUTH"],
|
|
},
|
|
)
|
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
|
|
|
|
|
async def test_abort_if_exists(
|
|
hass: HomeAssistant, mock_nmbs_client: AsyncMock, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test aborting the flow if the entry already exists."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": SOURCE_USER},
|
|
data={
|
|
CONF_STATION_FROM: DUMMY_DATA["STAT_BRUSSELS_NORTH"],
|
|
CONF_STATION_TO: DUMMY_DATA["STAT_BRUSSELS_SOUTH"],
|
|
},
|
|
)
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "already_configured"
|
|
|
|
|
|
async def test_dont_abort_if_exists_when_vias_differs(
|
|
hass: HomeAssistant, mock_nmbs_client: AsyncMock, mock_config_entry: MockConfigEntry
|
|
) -> None:
|
|
"""Test aborting the flow if the entry already exists."""
|
|
mock_config_entry.add_to_hass(hass)
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": SOURCE_USER},
|
|
data={
|
|
CONF_STATION_FROM: DUMMY_DATA["STAT_BRUSSELS_NORTH"],
|
|
CONF_STATION_TO: DUMMY_DATA["STAT_BRUSSELS_SOUTH"],
|
|
CONF_EXCLUDE_VIAS: True,
|
|
},
|
|
)
|
|
assert result["type"] is FlowResultType.CREATE_ENTRY
|
|
|
|
|
|
async def test_unavailable_api(
|
|
hass: HomeAssistant, mock_nmbs_client: AsyncMock
|
|
) -> None:
|
|
"""Test starting a flow by user and api is unavailable."""
|
|
mock_nmbs_client.get_stations.return_value = None
|
|
result = await hass.config_entries.flow.async_init(
|
|
DOMAIN,
|
|
context={"source": config_entries.SOURCE_USER},
|
|
)
|
|
|
|
assert result["type"] is FlowResultType.ABORT
|
|
assert result["reason"] == "api_unavailable"
|