Address late review of vicare config flow (#63343)
* Review comments from #56691 * Adapt to review suggestions * Update homeassistant/components/vicare/strings.json Co-authored-by: Martin Hjelmare <marhje52@gmail.com> Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
This commit is contained in:
parent
10195dc700
commit
2969a7ab4f
11 changed files with 96 additions and 155 deletions
|
@ -10,13 +10,7 @@ from PyViCare.PyViCareDevice import Device
|
|||
import voluptuous as vol
|
||||
|
||||
from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
|
||||
from homeassistant.const import (
|
||||
CONF_CLIENT_ID,
|
||||
CONF_NAME,
|
||||
CONF_PASSWORD,
|
||||
CONF_SCAN_INTERVAL,
|
||||
CONF_USERNAME,
|
||||
)
|
||||
from homeassistant.const import CONF_CLIENT_ID, CONF_PASSWORD, CONF_USERNAME
|
||||
from homeassistant.core import HomeAssistant
|
||||
import homeassistant.helpers.config_validation as cv
|
||||
from homeassistant.helpers.storage import STORAGE_DIR
|
||||
|
@ -26,6 +20,7 @@ from .const import (
|
|||
CONF_CIRCUIT,
|
||||
CONF_HEATING_TYPE,
|
||||
DEFAULT_HEATING_TYPE,
|
||||
DEFAULT_SCAN_INTERVAL,
|
||||
DOMAIN,
|
||||
HEATING_TYPE_TO_CREATOR_METHOD,
|
||||
PLATFORMS,
|
||||
|
@ -46,28 +41,28 @@ class ViCareRequiredKeysMixin:
|
|||
|
||||
|
||||
CONFIG_SCHEMA = vol.Schema(
|
||||
{
|
||||
DOMAIN: vol.All(
|
||||
cv.deprecated(CONF_CIRCUIT),
|
||||
vol.Schema(
|
||||
{
|
||||
vol.Required(CONF_USERNAME): cv.string,
|
||||
vol.Required(CONF_PASSWORD): cv.string,
|
||||
vol.Required(CONF_CLIENT_ID): cv.string,
|
||||
vol.Optional(CONF_SCAN_INTERVAL, default=60): vol.All(
|
||||
cv.time_period, lambda value: value.total_seconds()
|
||||
),
|
||||
vol.Optional(
|
||||
CONF_CIRCUIT
|
||||
): int, # Ignored: All circuits are now supported. Will be removed when switching to Setup via UI.
|
||||
vol.Optional(CONF_NAME, default="ViCare"): cv.string,
|
||||
vol.Optional(
|
||||
CONF_HEATING_TYPE, default=DEFAULT_HEATING_TYPE.value
|
||||
): vol.In([e.value for e in HeatingType]),
|
||||
}
|
||||
),
|
||||
)
|
||||
},
|
||||
vol.All(
|
||||
cv.deprecated(DOMAIN),
|
||||
{
|
||||
DOMAIN: vol.All(
|
||||
cv.deprecated(CONF_CIRCUIT),
|
||||
cv.deprecated(DOMAIN),
|
||||
vol.Schema(
|
||||
{
|
||||
vol.Required(CONF_USERNAME): cv.string,
|
||||
vol.Required(CONF_PASSWORD): cv.string,
|
||||
vol.Required(CONF_CLIENT_ID): cv.string,
|
||||
vol.Optional(
|
||||
CONF_CIRCUIT
|
||||
): int, # Ignored: All circuits are now supported. Will be removed when switching to Setup via UI.
|
||||
vol.Optional(
|
||||
CONF_HEATING_TYPE, default=DEFAULT_HEATING_TYPE.value
|
||||
): vol.In([e.value for e in HeatingType]),
|
||||
}
|
||||
),
|
||||
)
|
||||
},
|
||||
),
|
||||
extra=vol.ALLOW_EXTRA,
|
||||
)
|
||||
|
||||
|
@ -106,7 +101,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|||
def vicare_login(hass, entry_data):
|
||||
"""Login via PyVicare API."""
|
||||
vicare_api = PyViCare()
|
||||
vicare_api.setCacheDuration(entry_data[CONF_SCAN_INTERVAL])
|
||||
vicare_api.setCacheDuration(DEFAULT_SCAN_INTERVAL)
|
||||
vicare_api.initWithCredentials(
|
||||
entry_data[CONF_USERNAME],
|
||||
entry_data[CONF_PASSWORD],
|
||||
|
|
|
@ -18,12 +18,17 @@ from homeassistant.components.binary_sensor import (
|
|||
BinarySensorEntityDescription,
|
||||
)
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.const import CONF_NAME
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||
|
||||
from . import ViCareRequiredKeysMixin
|
||||
from .const import DOMAIN, VICARE_API, VICARE_CIRCUITS, VICARE_DEVICE_CONFIG
|
||||
from .const import (
|
||||
DOMAIN,
|
||||
VICARE_API,
|
||||
VICARE_CIRCUITS,
|
||||
VICARE_DEVICE_CONFIG,
|
||||
VICARE_NAME,
|
||||
)
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
|
@ -88,7 +93,7 @@ def _build_entity(name, vicare_api, device_config, sensor):
|
|||
|
||||
|
||||
async def _entities_from_descriptions(
|
||||
hass, name, all_devices, sensor_descriptions, iterables, config_entry
|
||||
hass, name, entities, sensor_descriptions, iterables, config_entry
|
||||
):
|
||||
"""Create entities from descriptions and list of burners/circuits."""
|
||||
for description in sensor_descriptions:
|
||||
|
@ -104,19 +109,19 @@ async def _entities_from_descriptions(
|
|||
description,
|
||||
)
|
||||
if entity is not None:
|
||||
all_devices.append(entity)
|
||||
entities.append(entity)
|
||||
|
||||
|
||||
async def async_setup_entry(
|
||||
hass: HomeAssistant,
|
||||
config_entry: ConfigEntry,
|
||||
async_add_devices: AddEntitiesCallback,
|
||||
async_add_entities: AddEntitiesCallback,
|
||||
) -> None:
|
||||
"""Create the ViCare binary sensor devices."""
|
||||
name = config_entry.data[CONF_NAME]
|
||||
name = VICARE_NAME
|
||||
api = hass.data[DOMAIN][config_entry.entry_id][VICARE_API]
|
||||
|
||||
all_devices = []
|
||||
entities = []
|
||||
|
||||
for description in CIRCUIT_SENSORS:
|
||||
for circuit in hass.data[DOMAIN][config_entry.entry_id][VICARE_CIRCUITS]:
|
||||
|
@ -131,23 +136,23 @@ async def async_setup_entry(
|
|||
description,
|
||||
)
|
||||
if entity is not None:
|
||||
all_devices.append(entity)
|
||||
entities.append(entity)
|
||||
|
||||
try:
|
||||
await _entities_from_descriptions(
|
||||
hass, name, all_devices, BURNER_SENSORS, api.burners, config_entry
|
||||
hass, name, entities, BURNER_SENSORS, api.burners, config_entry
|
||||
)
|
||||
except PyViCareNotSupportedFeatureError:
|
||||
_LOGGER.info("No burners found")
|
||||
|
||||
try:
|
||||
await _entities_from_descriptions(
|
||||
hass, name, all_devices, COMPRESSOR_SENSORS, api.compressors, config_entry
|
||||
hass, name, entities, COMPRESSOR_SENSORS, api.compressors, config_entry
|
||||
)
|
||||
except PyViCareNotSupportedFeatureError:
|
||||
_LOGGER.info("No compressors found")
|
||||
|
||||
async_add_devices(all_devices)
|
||||
async_add_entities(entities)
|
||||
|
||||
|
||||
class ViCareBinarySensor(BinarySensorEntity):
|
||||
|
|
|
@ -23,12 +23,7 @@ from homeassistant.components.climate.const import (
|
|||
SUPPORT_TARGET_TEMPERATURE,
|
||||
)
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.const import (
|
||||
ATTR_TEMPERATURE,
|
||||
CONF_NAME,
|
||||
PRECISION_WHOLE,
|
||||
TEMP_CELSIUS,
|
||||
)
|
||||
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_WHOLE, TEMP_CELSIUS
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers import entity_platform
|
||||
import homeassistant.helpers.config_validation as cv
|
||||
|
@ -40,6 +35,7 @@ from .const import (
|
|||
VICARE_API,
|
||||
VICARE_CIRCUITS,
|
||||
VICARE_DEVICE_CONFIG,
|
||||
VICARE_NAME,
|
||||
)
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
@ -109,12 +105,12 @@ def _build_entity(name, vicare_api, circuit, device_config, heating_type):
|
|||
async def async_setup_entry(
|
||||
hass: HomeAssistant,
|
||||
config_entry: ConfigEntry,
|
||||
async_add_devices: AddEntitiesCallback,
|
||||
async_add_entities: AddEntitiesCallback,
|
||||
) -> None:
|
||||
"""Set up the ViCare climate platform."""
|
||||
name = config_entry.data[CONF_NAME]
|
||||
name = VICARE_NAME
|
||||
|
||||
all_devices = []
|
||||
entities = []
|
||||
|
||||
for circuit in hass.data[DOMAIN][config_entry.entry_id][VICARE_CIRCUITS]:
|
||||
suffix = ""
|
||||
|
@ -128,7 +124,7 @@ async def async_setup_entry(
|
|||
config_entry.data[CONF_HEATING_TYPE],
|
||||
)
|
||||
if entity is not None:
|
||||
all_devices.append(entity)
|
||||
entities.append(entity)
|
||||
|
||||
platform = entity_platform.async_get_current_platform()
|
||||
|
||||
|
@ -138,7 +134,7 @@ async def async_setup_entry(
|
|||
"set_vicare_mode",
|
||||
)
|
||||
|
||||
async_add_devices(all_devices)
|
||||
async_add_entities(entities)
|
||||
|
||||
|
||||
class ViCareClimate(ClimateEntity):
|
||||
|
|
|
@ -9,13 +9,7 @@ import voluptuous as vol
|
|||
|
||||
from homeassistant import config_entries
|
||||
from homeassistant.components import dhcp
|
||||
from homeassistant.const import (
|
||||
CONF_CLIENT_ID,
|
||||
CONF_NAME,
|
||||
CONF_PASSWORD,
|
||||
CONF_SCAN_INTERVAL,
|
||||
CONF_USERNAME,
|
||||
)
|
||||
from homeassistant.const import CONF_CLIENT_ID, CONF_PASSWORD, CONF_USERNAME
|
||||
from homeassistant.data_entry_flow import FlowResult
|
||||
import homeassistant.helpers.config_validation as cv
|
||||
from homeassistant.helpers.device_registry import format_mac
|
||||
|
@ -25,8 +19,8 @@ from .const import (
|
|||
CONF_CIRCUIT,
|
||||
CONF_HEATING_TYPE,
|
||||
DEFAULT_HEATING_TYPE,
|
||||
DEFAULT_SCAN_INTERVAL,
|
||||
DOMAIN,
|
||||
VICARE_NAME,
|
||||
HeatingType,
|
||||
)
|
||||
|
||||
|
@ -50,10 +44,6 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
|
|||
vol.Required(CONF_HEATING_TYPE, default=DEFAULT_HEATING_TYPE.value): vol.In(
|
||||
[e.value for e in HeatingType]
|
||||
),
|
||||
vol.Optional(CONF_NAME, default="ViCare"): cv.string,
|
||||
vol.Optional(CONF_SCAN_INTERVAL, default=DEFAULT_SCAN_INTERVAL): vol.All(
|
||||
vol.Coerce(int), vol.Range(min=30)
|
||||
),
|
||||
}
|
||||
errors: dict[str, str] = {}
|
||||
|
||||
|
@ -62,12 +52,10 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
|
|||
await self.hass.async_add_executor_job(
|
||||
vicare_login, self.hass, user_input
|
||||
)
|
||||
return self.async_create_entry(
|
||||
title=user_input[CONF_NAME], data=user_input
|
||||
)
|
||||
except PyViCareInvalidCredentialsError as ex:
|
||||
_LOGGER.debug("Could not log in to ViCare, %s", ex)
|
||||
except PyViCareInvalidCredentialsError:
|
||||
errors["base"] = "invalid_auth"
|
||||
else:
|
||||
return self.async_create_entry(title=VICARE_NAME, data=user_input)
|
||||
|
||||
return self.async_show_form(
|
||||
step_id="user",
|
||||
|
@ -90,22 +78,14 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
|
|||
|
||||
async def async_step_import(self, import_info):
|
||||
"""Handle a flow initiated by a YAML config import."""
|
||||
|
||||
await self.async_set_unique_id("Configuration.yaml")
|
||||
self._abort_if_unique_id_configured()
|
||||
|
||||
if self._async_current_entries():
|
||||
return self.async_abort(reason="single_instance_allowed")
|
||||
|
||||
# Remove now unsupported config parameters
|
||||
if import_info.get(CONF_CIRCUIT):
|
||||
import_info.pop(CONF_CIRCUIT)
|
||||
import_info.pop(CONF_CIRCUIT, None)
|
||||
|
||||
# Add former optional config if missing
|
||||
# CONF_HEATING_TYPE is now required but was optional in yaml config. Add if missing.
|
||||
if import_info.get(CONF_HEATING_TYPE) is None:
|
||||
import_info[CONF_HEATING_TYPE] = DEFAULT_HEATING_TYPE.value
|
||||
|
||||
return self.async_create_entry(
|
||||
title="Configuration.yaml",
|
||||
data=import_info,
|
||||
)
|
||||
return await self.async_step_user(import_info)
|
||||
|
|
|
@ -16,6 +16,7 @@ PLATFORMS = [
|
|||
VICARE_DEVICE_CONFIG = "device_conf"
|
||||
VICARE_API = "api"
|
||||
VICARE_CIRCUITS = "circuits"
|
||||
VICARE_NAME = "ViCare"
|
||||
|
||||
CONF_CIRCUIT = "circuit"
|
||||
CONF_HEATING_TYPE = "heating_type"
|
||||
|
|
|
@ -22,7 +22,6 @@ from homeassistant.components.sensor import (
|
|||
)
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.const import (
|
||||
CONF_NAME,
|
||||
ENERGY_KILO_WATT_HOUR,
|
||||
PERCENTAGE,
|
||||
POWER_WATT,
|
||||
|
@ -39,6 +38,7 @@ from .const import (
|
|||
VICARE_API,
|
||||
VICARE_CIRCUITS,
|
||||
VICARE_DEVICE_CONFIG,
|
||||
VICARE_NAME,
|
||||
VICARE_UNIT_TO_DEVICE_CLASS,
|
||||
VICARE_UNIT_TO_UNIT_OF_MEASUREMENT,
|
||||
)
|
||||
|
@ -339,7 +339,7 @@ def _build_entity(name, vicare_api, device_config, sensor):
|
|||
|
||||
|
||||
async def _entities_from_descriptions(
|
||||
hass, name, all_devices, sensor_descriptions, iterables, config_entry
|
||||
hass, name, entities, sensor_descriptions, iterables, config_entry
|
||||
):
|
||||
"""Create entities from descriptions and list of burners/circuits."""
|
||||
for description in sensor_descriptions:
|
||||
|
@ -355,19 +355,19 @@ async def _entities_from_descriptions(
|
|||
description,
|
||||
)
|
||||
if entity is not None:
|
||||
all_devices.append(entity)
|
||||
entities.append(entity)
|
||||
|
||||
|
||||
async def async_setup_entry(
|
||||
hass: HomeAssistant,
|
||||
config_entry: ConfigEntry,
|
||||
async_add_devices: AddEntitiesCallback,
|
||||
async_add_entities: AddEntitiesCallback,
|
||||
) -> None:
|
||||
"""Create the ViCare sensor devices."""
|
||||
name = config_entry.data[CONF_NAME]
|
||||
name = VICARE_NAME
|
||||
api = hass.data[DOMAIN][config_entry.entry_id][VICARE_API]
|
||||
|
||||
all_devices = []
|
||||
entities = []
|
||||
for description in GLOBAL_SENSORS:
|
||||
entity = await hass.async_add_executor_job(
|
||||
_build_entity,
|
||||
|
@ -377,7 +377,7 @@ async def async_setup_entry(
|
|||
description,
|
||||
)
|
||||
if entity is not None:
|
||||
all_devices.append(entity)
|
||||
entities.append(entity)
|
||||
|
||||
for description in CIRCUIT_SENSORS:
|
||||
for circuit in hass.data[DOMAIN][config_entry.entry_id][VICARE_CIRCUITS]:
|
||||
|
@ -392,23 +392,23 @@ async def async_setup_entry(
|
|||
description,
|
||||
)
|
||||
if entity is not None:
|
||||
all_devices.append(entity)
|
||||
entities.append(entity)
|
||||
|
||||
try:
|
||||
await _entities_from_descriptions(
|
||||
hass, name, all_devices, BURNER_SENSORS, api.burners, config_entry
|
||||
hass, name, entities, BURNER_SENSORS, api.burners, config_entry
|
||||
)
|
||||
except PyViCareNotSupportedFeatureError:
|
||||
_LOGGER.info("No burners found")
|
||||
|
||||
try:
|
||||
await _entities_from_descriptions(
|
||||
hass, name, all_devices, COMPRESSOR_SENSORS, api.compressors, config_entry
|
||||
hass, name, entities, COMPRESSOR_SENSORS, api.compressors, config_entry
|
||||
)
|
||||
except PyViCareNotSupportedFeatureError:
|
||||
_LOGGER.info("No compressors found")
|
||||
|
||||
async_add_devices(all_devices)
|
||||
async_add_entities(entities)
|
||||
|
||||
|
||||
class ViCareSensor(SensorEntity):
|
||||
|
|
|
@ -6,8 +6,6 @@
|
|||
"title": "{name}",
|
||||
"description": "Set up ViCare integration. To generate API key go to https://developer.viessmann.com",
|
||||
"data": {
|
||||
"name": "[%key:common::config_flow::data::name%]",
|
||||
"scan_interval": "Scan Interval (seconds)",
|
||||
"username": "[%key:common::config_flow::data::email%]",
|
||||
"password": "[%key:common::config_flow::data::password%]",
|
||||
"client_id": "[%key:common::config_flow::data::api_key%]",
|
||||
|
|
|
@ -15,7 +15,6 @@
|
|||
"heating_type": "Heating type",
|
||||
"name": "Name",
|
||||
"password": "Password",
|
||||
"scan_interval": "Scan Interval (seconds)",
|
||||
"username": "Email"
|
||||
},
|
||||
"description": "Set up ViCare integration. To generate API key go to https://developer.viessmann.com",
|
||||
|
|
|
@ -14,12 +14,7 @@ from homeassistant.components.water_heater import (
|
|||
WaterHeaterEntity,
|
||||
)
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.const import (
|
||||
ATTR_TEMPERATURE,
|
||||
CONF_NAME,
|
||||
PRECISION_WHOLE,
|
||||
TEMP_CELSIUS,
|
||||
)
|
||||
from homeassistant.const import ATTR_TEMPERATURE, PRECISION_WHOLE, TEMP_CELSIUS
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||
|
||||
|
@ -29,6 +24,7 @@ from .const import (
|
|||
VICARE_API,
|
||||
VICARE_CIRCUITS,
|
||||
VICARE_DEVICE_CONFIG,
|
||||
VICARE_NAME,
|
||||
)
|
||||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
@ -76,12 +72,12 @@ def _build_entity(name, vicare_api, circuit, device_config, heating_type):
|
|||
async def async_setup_entry(
|
||||
hass: HomeAssistant,
|
||||
config_entry: ConfigEntry,
|
||||
async_add_devices: AddEntitiesCallback,
|
||||
async_add_entities: AddEntitiesCallback,
|
||||
) -> None:
|
||||
"""Set up the ViCare climate platform."""
|
||||
name = config_entry.data[CONF_NAME]
|
||||
name = VICARE_NAME
|
||||
|
||||
all_devices = []
|
||||
entities = []
|
||||
for circuit in hass.data[DOMAIN][config_entry.entry_id][VICARE_CIRCUITS]:
|
||||
suffix = ""
|
||||
if len(hass.data[DOMAIN][config_entry.entry_id][VICARE_CIRCUITS]) > 1:
|
||||
|
@ -94,9 +90,9 @@ async def async_setup_entry(
|
|||
config_entry.data[CONF_HEATING_TYPE],
|
||||
)
|
||||
if entity is not None:
|
||||
all_devices.append(entity)
|
||||
entities.append(entity)
|
||||
|
||||
async_add_devices(all_devices)
|
||||
async_add_entities(entities)
|
||||
|
||||
|
||||
class ViCareWater(WaterHeaterEntity):
|
||||
|
|
|
@ -1,20 +1,22 @@
|
|||
"""Test for ViCare."""
|
||||
from homeassistant.components.vicare.const import CONF_HEATING_TYPE
|
||||
from homeassistant.const import (
|
||||
CONF_CLIENT_ID,
|
||||
CONF_NAME,
|
||||
CONF_PASSWORD,
|
||||
CONF_SCAN_INTERVAL,
|
||||
CONF_USERNAME,
|
||||
)
|
||||
from __future__ import annotations
|
||||
|
||||
ENTRY_CONFIG = {
|
||||
from typing import Final
|
||||
|
||||
from homeassistant.components.vicare.const import CONF_HEATING_TYPE
|
||||
from homeassistant.const import CONF_CLIENT_ID, CONF_PASSWORD, CONF_USERNAME
|
||||
|
||||
ENTRY_CONFIG: Final[dict[str, str]] = {
|
||||
CONF_USERNAME: "foo@bar.com",
|
||||
CONF_PASSWORD: "1234",
|
||||
CONF_CLIENT_ID: "5678",
|
||||
CONF_HEATING_TYPE: "auto",
|
||||
CONF_SCAN_INTERVAL: 60,
|
||||
CONF_NAME: "ViCare",
|
||||
}
|
||||
|
||||
ENTRY_CONFIG_NO_HEATING_TYPE: Final[dict[str, str]] = {
|
||||
CONF_USERNAME: "foo@bar.com",
|
||||
CONF_PASSWORD: "1234",
|
||||
CONF_CLIENT_ID: "5678",
|
||||
}
|
||||
|
||||
MOCK_MAC = "B874241B7B9"
|
||||
|
|
|
@ -3,23 +3,18 @@ from unittest.mock import patch
|
|||
|
||||
from PyViCare.PyViCareUtils import PyViCareInvalidCredentialsError
|
||||
|
||||
from homeassistant import config_entries, data_entry_flow, setup
|
||||
from homeassistant import config_entries, data_entry_flow
|
||||
from homeassistant.components import dhcp
|
||||
from homeassistant.components.vicare.const import (
|
||||
CONF_CIRCUIT,
|
||||
CONF_HEATING_TYPE,
|
||||
DOMAIN,
|
||||
)
|
||||
from homeassistant.components.vicare.const import CONF_CIRCUIT, DOMAIN, VICARE_NAME
|
||||
from homeassistant.const import CONF_CLIENT_ID, CONF_PASSWORD, CONF_USERNAME
|
||||
|
||||
from . import ENTRY_CONFIG, MOCK_MAC
|
||||
from . import ENTRY_CONFIG, ENTRY_CONFIG_NO_HEATING_TYPE, MOCK_MAC
|
||||
|
||||
from tests.common import MockConfigEntry
|
||||
|
||||
|
||||
async def test_form(hass):
|
||||
"""Test we get the form."""
|
||||
await setup.async_setup_component(hass, "persistent_notification", {})
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN, context={"source": config_entries.SOURCE_USER}
|
||||
)
|
||||
|
@ -54,7 +49,6 @@ async def test_form(hass):
|
|||
|
||||
async def test_import(hass):
|
||||
"""Test that the import works."""
|
||||
await setup.async_setup_component(hass, "persistent_notification", {})
|
||||
|
||||
with patch(
|
||||
"homeassistant.components.vicare.config_flow.vicare_login",
|
||||
|
@ -71,7 +65,7 @@ async def test_import(hass):
|
|||
data=ENTRY_CONFIG,
|
||||
)
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
|
||||
assert result["title"] == "Configuration.yaml"
|
||||
assert result["title"] == VICARE_NAME
|
||||
assert result["data"] == ENTRY_CONFIG
|
||||
|
||||
await hass.async_block_till_done()
|
||||
|
@ -81,7 +75,6 @@ async def test_import(hass):
|
|||
|
||||
async def test_import_removes_circuit(hass):
|
||||
"""Test that the import works."""
|
||||
await setup.async_setup_component(hass, "persistent_notification", {})
|
||||
|
||||
with patch(
|
||||
"homeassistant.components.vicare.config_flow.vicare_login",
|
||||
|
@ -99,7 +92,7 @@ async def test_import_removes_circuit(hass):
|
|||
data=ENTRY_CONFIG,
|
||||
)
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
|
||||
assert result["title"] == "Configuration.yaml"
|
||||
assert result["title"] == VICARE_NAME
|
||||
assert result["data"] == ENTRY_CONFIG
|
||||
|
||||
await hass.async_block_till_done()
|
||||
|
@ -109,7 +102,6 @@ async def test_import_removes_circuit(hass):
|
|||
|
||||
async def test_import_adds_heating_type(hass):
|
||||
"""Test that the import works."""
|
||||
await setup.async_setup_component(hass, "persistent_notification", {})
|
||||
|
||||
with patch(
|
||||
"homeassistant.components.vicare.config_flow.vicare_login",
|
||||
|
@ -120,14 +112,13 @@ async def test_import_adds_heating_type(hass):
|
|||
"homeassistant.components.vicare.async_setup_entry",
|
||||
return_value=True,
|
||||
) as mock_setup_entry:
|
||||
del ENTRY_CONFIG[CONF_HEATING_TYPE]
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN,
|
||||
context={"source": config_entries.SOURCE_IMPORT},
|
||||
data=ENTRY_CONFIG,
|
||||
data=ENTRY_CONFIG_NO_HEATING_TYPE,
|
||||
)
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
|
||||
assert result["title"] == "Configuration.yaml"
|
||||
assert result["title"] == VICARE_NAME
|
||||
assert result["data"] == ENTRY_CONFIG
|
||||
|
||||
await hass.async_block_till_done()
|
||||
|
@ -162,7 +153,6 @@ async def test_invalid_login(hass) -> None:
|
|||
|
||||
async def test_form_dhcp(hass):
|
||||
"""Test we can setup from dhcp."""
|
||||
await setup.async_setup_component(hass, "persistent_notification", {})
|
||||
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN,
|
||||
|
@ -203,29 +193,10 @@ async def test_form_dhcp(hass):
|
|||
assert len(mock_setup_entry.mock_calls) == 1
|
||||
|
||||
|
||||
async def test_import_already_configured(hass):
|
||||
"""Test that configuring same instance is rejectes."""
|
||||
mock_entry = MockConfigEntry(
|
||||
domain=DOMAIN,
|
||||
unique_id="Configuration.yaml",
|
||||
data=ENTRY_CONFIG,
|
||||
)
|
||||
mock_entry.add_to_hass(hass)
|
||||
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN,
|
||||
context={"source": config_entries.SOURCE_IMPORT},
|
||||
data=ENTRY_CONFIG,
|
||||
)
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_ABORT
|
||||
assert result["reason"] == "already_configured"
|
||||
|
||||
|
||||
async def test_import_single_instance_allowed(hass):
|
||||
"""Test that configuring more than one instance is rejected."""
|
||||
mock_entry = MockConfigEntry(
|
||||
domain=DOMAIN,
|
||||
unique_id="Configuration.yaml",
|
||||
data=ENTRY_CONFIG,
|
||||
)
|
||||
mock_entry.add_to_hass(hass)
|
||||
|
@ -236,14 +207,13 @@ async def test_import_single_instance_allowed(hass):
|
|||
data=ENTRY_CONFIG,
|
||||
)
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_ABORT
|
||||
assert result["reason"] == "already_configured"
|
||||
assert result["reason"] == "single_instance_allowed"
|
||||
|
||||
|
||||
async def test_dhcp_single_instance_allowed(hass):
|
||||
"""Test that configuring more than one instance is rejected."""
|
||||
mock_entry = MockConfigEntry(
|
||||
domain=DOMAIN,
|
||||
unique_id="Configuration.yaml",
|
||||
data=ENTRY_CONFIG,
|
||||
)
|
||||
mock_entry.add_to_hass(hass)
|
||||
|
@ -263,7 +233,6 @@ async def test_dhcp_single_instance_allowed(hass):
|
|||
|
||||
async def test_user_input_single_instance_allowed(hass):
|
||||
"""Test that configuring more than one instance is rejected."""
|
||||
await setup.async_setup_component(hass, "persistent_notification", {})
|
||||
mock_entry = MockConfigEntry(
|
||||
domain=DOMAIN,
|
||||
unique_id="ViCare",
|
||||
|
|
Loading…
Add table
Reference in a new issue