1487 lines
50 KiB
Python
1487 lines
50 KiB
Python
"""Test config flow."""
|
|
from random import getrandbits
|
|
from ssl import SSLError
|
|
from unittest.mock import AsyncMock, patch
|
|
from uuid import uuid4
|
|
|
|
import pytest
|
|
import voluptuous as vol
|
|
import yaml
|
|
|
|
from homeassistant import config as hass_config, config_entries, data_entry_flow
|
|
from homeassistant.components import mqtt
|
|
from homeassistant.components.hassio import HassioServiceInfo
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
from tests.common import MockConfigEntry
|
|
|
|
MOCK_CLIENT_CERT = b"## mock client certificate file ##"
|
|
MOCK_CLIENT_KEY = b"## mock key file ##"
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def mock_finish_setup():
|
|
"""Mock out the finish setup method."""
|
|
with patch(
|
|
"homeassistant.components.mqtt.MQTT.async_connect", return_value=True
|
|
) as mock_finish:
|
|
yield mock_finish
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_client_cert_check_fail():
|
|
"""Mock the client certificate check."""
|
|
with patch(
|
|
"homeassistant.components.mqtt.config_flow.load_pem_x509_certificate",
|
|
side_effect=ValueError,
|
|
) as mock_cert_check:
|
|
yield mock_cert_check
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_client_key_check_fail():
|
|
"""Mock the client key file check."""
|
|
with patch(
|
|
"homeassistant.components.mqtt.config_flow.load_pem_private_key",
|
|
side_effect=ValueError,
|
|
) as mock_key_check:
|
|
yield mock_key_check
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_ssl_context():
|
|
"""Mock the SSL context used to load the cert chain and to load verify locations."""
|
|
with patch(
|
|
"homeassistant.components.mqtt.config_flow.SSLContext"
|
|
) as mock_context, patch(
|
|
"homeassistant.components.mqtt.config_flow.load_pem_private_key"
|
|
) as mock_key_check, patch(
|
|
"homeassistant.components.mqtt.config_flow.load_pem_x509_certificate"
|
|
) as mock_cert_check:
|
|
yield {
|
|
"context": mock_context,
|
|
"load_pem_x509_certificate": mock_cert_check,
|
|
"load_pem_private_key": mock_key_check,
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_reload_after_entry_update():
|
|
"""Mock out the reload after updating the entry."""
|
|
with patch(
|
|
"homeassistant.components.mqtt._async_config_entry_updated"
|
|
) as mock_reload:
|
|
yield mock_reload
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_try_connection():
|
|
"""Mock the try connection method."""
|
|
with patch("homeassistant.components.mqtt.config_flow.try_connection") as mock_try:
|
|
yield mock_try
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_try_connection_success():
|
|
"""Mock the try connection method with success."""
|
|
|
|
_mid = 1
|
|
|
|
def get_mid():
|
|
nonlocal _mid
|
|
_mid += 1
|
|
return _mid
|
|
|
|
def loop_start():
|
|
"""Simulate connect on loop start."""
|
|
mock_client().on_connect(mock_client, None, None, 0)
|
|
|
|
def _subscribe(topic, qos=0):
|
|
mid = get_mid()
|
|
mock_client().on_subscribe(mock_client, 0, mid)
|
|
return (0, mid)
|
|
|
|
def _unsubscribe(topic):
|
|
mid = get_mid()
|
|
mock_client().on_unsubscribe(mock_client, 0, mid)
|
|
return (0, mid)
|
|
|
|
with patch("paho.mqtt.client.Client") as mock_client:
|
|
mock_client().loop_start = loop_start
|
|
mock_client().subscribe = _subscribe
|
|
mock_client().unsubscribe = _unsubscribe
|
|
|
|
yield mock_client()
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_try_connection_time_out():
|
|
"""Mock the try connection method with a time out."""
|
|
|
|
# Patch prevent waiting 5 sec for a timeout
|
|
with patch("paho.mqtt.client.Client") as mock_client, patch(
|
|
"homeassistant.components.mqtt.config_flow.MQTT_TIMEOUT", 0
|
|
):
|
|
mock_client().loop_start = lambda *args: 1
|
|
yield mock_client()
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_process_uploaded_file(tmp_path):
|
|
"""Mock upload certificate files."""
|
|
file_id_ca = str(uuid4())
|
|
file_id_cert = str(uuid4())
|
|
file_id_key = str(uuid4())
|
|
|
|
def _mock_process_uploaded_file(hass, file_id):
|
|
if file_id == file_id_ca:
|
|
with open(tmp_path / "ca.crt", "wb") as cafile:
|
|
cafile.write(b"## mock CA certificate file ##")
|
|
return tmp_path / "ca.crt"
|
|
elif file_id == file_id_cert:
|
|
with open(tmp_path / "client.crt", "wb") as certfile:
|
|
certfile.write(b"## mock client certificate file ##")
|
|
return tmp_path / "client.crt"
|
|
elif file_id == file_id_key:
|
|
with open(tmp_path / "client.key", "wb") as keyfile:
|
|
keyfile.write(b"## mock key file ##")
|
|
return tmp_path / "client.key"
|
|
else:
|
|
pytest.fail(f"Unexpected file_id: {file_id}")
|
|
|
|
with patch(
|
|
"homeassistant.components.mqtt.config_flow.process_uploaded_file",
|
|
side_effect=_mock_process_uploaded_file,
|
|
) as mock_upload, patch(
|
|
# Patch temp dir name to avoid tests fail running in parallel
|
|
"homeassistant.components.mqtt.util.TEMP_DIR_NAME",
|
|
"home-assistant-mqtt" + f"-{getrandbits(10):03x}",
|
|
):
|
|
mock_upload.file_id = {
|
|
mqtt.CONF_CERTIFICATE: file_id_ca,
|
|
mqtt.CONF_CLIENT_CERT: file_id_cert,
|
|
mqtt.CONF_CLIENT_KEY: file_id_key,
|
|
}
|
|
yield mock_upload
|
|
|
|
|
|
async def test_user_connection_works(
|
|
hass, mock_try_connection, mock_finish_setup, mqtt_client_mock
|
|
):
|
|
"""Test we can finish a config flow."""
|
|
mock_try_connection.return_value = True
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
"mqtt", context={"source": config_entries.SOURCE_USER}
|
|
)
|
|
assert result["type"] == "form"
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"], {"broker": "127.0.0.1", "advanced_options": False}
|
|
)
|
|
|
|
assert result["type"] == "create_entry"
|
|
assert result["result"].data == {
|
|
"broker": "127.0.0.1",
|
|
"port": 1883,
|
|
"discovery": True,
|
|
"discovery_prefix": "homeassistant",
|
|
}
|
|
# Check we tried the connection
|
|
assert len(mock_try_connection.mock_calls) == 1
|
|
# Check config entry got setup
|
|
assert len(mock_finish_setup.mock_calls) == 1
|
|
|
|
|
|
async def test_user_v5_connection_works(
|
|
hass, mock_try_connection, mock_finish_setup, mqtt_client_mock
|
|
):
|
|
"""Test we can finish a config flow."""
|
|
mock_try_connection.return_value = True
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
"mqtt", context={"source": config_entries.SOURCE_USER}
|
|
)
|
|
assert result["type"] == "form"
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"], {"broker": "127.0.0.1", "advanced_options": True}
|
|
)
|
|
|
|
assert result["step_id"] == "broker"
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_PROTOCOL: "5",
|
|
},
|
|
)
|
|
assert result["type"] == "create_entry"
|
|
assert result["result"].data == {
|
|
"broker": "another-broker",
|
|
"discovery": True,
|
|
"discovery_prefix": "homeassistant",
|
|
"port": 2345,
|
|
"protocol": "5",
|
|
}
|
|
# Check we tried the connection
|
|
assert len(mock_try_connection.mock_calls) == 1
|
|
# Check config entry got setup
|
|
assert len(mock_finish_setup.mock_calls) == 1
|
|
|
|
|
|
async def test_user_connection_fails(
|
|
hass, mock_try_connection_time_out, mock_finish_setup
|
|
):
|
|
"""Test if connection cannot be made."""
|
|
result = await hass.config_entries.flow.async_init(
|
|
"mqtt", context={"source": config_entries.SOURCE_USER}
|
|
)
|
|
assert result["type"] == "form"
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"], {"broker": "127.0.0.1"}
|
|
)
|
|
|
|
assert result["type"] == "form"
|
|
assert result["errors"]["base"] == "cannot_connect"
|
|
|
|
# Check we tried the connection
|
|
assert len(mock_try_connection_time_out.mock_calls)
|
|
# Check config entry did not setup
|
|
assert len(mock_finish_setup.mock_calls) == 0
|
|
|
|
|
|
async def test_manual_config_starts_discovery_flow(
|
|
hass, mock_try_connection, mock_finish_setup, mqtt_client_mock
|
|
):
|
|
"""Test manual config initiates a discovery flow."""
|
|
# No flows in progress
|
|
assert hass.config_entries.flow.async_progress() == []
|
|
|
|
# MQTT config present in yaml config
|
|
assert await async_setup_component(hass, "mqtt", {"mqtt": {}})
|
|
await hass.async_block_till_done()
|
|
assert len(mock_finish_setup.mock_calls) == 0
|
|
|
|
# There should now be a discovery flow
|
|
flows = hass.config_entries.flow.async_progress()
|
|
assert len(flows) == 1
|
|
assert flows[0]["context"]["source"] == "integration_discovery"
|
|
assert flows[0]["handler"] == "mqtt"
|
|
assert flows[0]["step_id"] == "broker"
|
|
|
|
|
|
async def test_manual_config_set(
|
|
hass,
|
|
mock_try_connection,
|
|
mock_finish_setup,
|
|
mqtt_client_mock,
|
|
):
|
|
"""Test manual config does not create an entry, and entry can be setup late."""
|
|
# MQTT config present in yaml config
|
|
assert await async_setup_component(hass, "mqtt", {"mqtt": {"broker": "bla"}})
|
|
await hass.async_block_till_done()
|
|
# do not try to reload
|
|
hass.data["mqtt"].reload_needed = False
|
|
assert len(mock_finish_setup.mock_calls) == 0
|
|
|
|
mock_try_connection.return_value = True
|
|
|
|
# Start config flow
|
|
result = await hass.config_entries.flow.async_init(
|
|
"mqtt", context={"source": config_entries.SOURCE_USER}
|
|
)
|
|
assert result["type"] == "form"
|
|
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"], {"broker": "127.0.0.1"}
|
|
)
|
|
|
|
assert result["type"] == "create_entry"
|
|
assert result["result"].data == {
|
|
"broker": "127.0.0.1",
|
|
"port": 1883,
|
|
"discovery": True,
|
|
"discovery_prefix": "homeassistant",
|
|
}
|
|
# Check we tried the connection, with precedence for config entry settings
|
|
mock_try_connection.assert_called_once_with(
|
|
{
|
|
"broker": "127.0.0.1",
|
|
"port": 1883,
|
|
"discovery": True,
|
|
},
|
|
)
|
|
# Check config entry got setup
|
|
assert len(mock_finish_setup.mock_calls) == 1
|
|
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
|
|
assert config_entry.title == "127.0.0.1"
|
|
|
|
|
|
async def test_user_single_instance(hass):
|
|
"""Test we only allow a single config flow."""
|
|
MockConfigEntry(domain="mqtt").add_to_hass(hass)
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
"mqtt", context={"source": config_entries.SOURCE_USER}
|
|
)
|
|
assert result["type"] == "abort"
|
|
assert result["reason"] == "single_instance_allowed"
|
|
|
|
|
|
async def test_hassio_already_configured(hass):
|
|
"""Test we only allow a single config flow."""
|
|
MockConfigEntry(domain="mqtt").add_to_hass(hass)
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
"mqtt", context={"source": config_entries.SOURCE_HASSIO}
|
|
)
|
|
assert result["type"] == "abort"
|
|
assert result["reason"] == "already_configured"
|
|
|
|
|
|
async def test_hassio_ignored(hass: HomeAssistant) -> None:
|
|
"""Test we supervisor discovered instance can be ignored."""
|
|
MockConfigEntry(
|
|
domain=mqtt.DOMAIN, source=config_entries.SOURCE_IGNORE
|
|
).add_to_hass(hass)
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
mqtt.DOMAIN,
|
|
data=HassioServiceInfo(
|
|
config={
|
|
"addon": "Mosquitto",
|
|
"host": "mock-mosquitto",
|
|
"port": "1883",
|
|
"protocol": "3.1.1",
|
|
},
|
|
name="Mosquitto",
|
|
slug="mosquitto",
|
|
),
|
|
context={"source": config_entries.SOURCE_HASSIO},
|
|
)
|
|
assert result
|
|
assert result.get("type") == data_entry_flow.FlowResultType.ABORT
|
|
assert result.get("reason") == "already_configured"
|
|
|
|
|
|
async def test_hassio_confirm(hass, mock_try_connection_success, mock_finish_setup):
|
|
"""Test we can finish a config flow."""
|
|
mock_try_connection.return_value = True
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
"mqtt",
|
|
data=HassioServiceInfo(
|
|
config={
|
|
"addon": "Mock Addon",
|
|
"host": "mock-broker",
|
|
"port": 1883,
|
|
"username": "mock-user",
|
|
"password": "mock-pass",
|
|
"protocol": "3.1.1", # Set by the addon's discovery, ignored by HA
|
|
"ssl": False, # Set by the addon's discovery, ignored by HA
|
|
},
|
|
name="Mock Addon",
|
|
slug="mosquitto",
|
|
),
|
|
context={"source": config_entries.SOURCE_HASSIO},
|
|
)
|
|
assert result["type"] == "form"
|
|
assert result["step_id"] == "hassio_confirm"
|
|
assert result["description_placeholders"] == {"addon": "Mock Addon"}
|
|
|
|
mock_try_connection_success.reset_mock()
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"], {"discovery": True}
|
|
)
|
|
|
|
assert result["type"] == "create_entry"
|
|
assert result["result"].data == {
|
|
"broker": "mock-broker",
|
|
"port": 1883,
|
|
"username": "mock-user",
|
|
"password": "mock-pass",
|
|
"discovery": True,
|
|
"discovery_prefix": "homeassistant",
|
|
}
|
|
# Check we tried the connection
|
|
assert len(mock_try_connection_success.mock_calls)
|
|
# Check config entry got setup
|
|
assert len(mock_finish_setup.mock_calls) == 1
|
|
|
|
|
|
async def test_hassio_cannot_connect(
|
|
hass, mock_try_connection_time_out, mock_finish_setup
|
|
):
|
|
"""Test a config flow is aborted when a connection was not successful."""
|
|
mock_try_connection.return_value = True
|
|
|
|
result = await hass.config_entries.flow.async_init(
|
|
"mqtt",
|
|
data=HassioServiceInfo(
|
|
config={
|
|
"addon": "Mock Addon",
|
|
"host": "mock-broker",
|
|
"port": 1883,
|
|
"username": "mock-user",
|
|
"password": "mock-pass",
|
|
"protocol": "3.1.1", # Set by the addon's discovery, ignored by HA
|
|
"ssl": False, # Set by the addon's discovery, ignored by HA
|
|
},
|
|
name="Mock Addon",
|
|
slug="mosquitto",
|
|
),
|
|
context={"source": config_entries.SOURCE_HASSIO},
|
|
)
|
|
assert result["type"] == "form"
|
|
assert result["step_id"] == "hassio_confirm"
|
|
assert result["description_placeholders"] == {"addon": "Mock Addon"}
|
|
|
|
mock_try_connection_time_out.reset_mock()
|
|
result = await hass.config_entries.flow.async_configure(
|
|
result["flow_id"], {"discovery": True}
|
|
)
|
|
|
|
assert result["type"] == "form"
|
|
assert result["errors"]["base"] == "cannot_connect"
|
|
# Check we tried the connection
|
|
assert len(mock_try_connection_time_out.mock_calls)
|
|
# Check config entry got setup
|
|
assert len(mock_finish_setup.mock_calls) == 0
|
|
|
|
|
|
async def test_option_flow(
|
|
hass,
|
|
mqtt_mock_entry_no_yaml_config,
|
|
mock_try_connection,
|
|
):
|
|
"""Test config flow options."""
|
|
with patch(
|
|
"homeassistant.config.async_hass_config_yaml", AsyncMock(return_value={})
|
|
) as yaml_mock:
|
|
mqtt_mock = await mqtt_mock_entry_no_yaml_config()
|
|
mock_try_connection.return_value = True
|
|
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
}
|
|
|
|
mqtt_mock.async_connect.reset_mock()
|
|
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "broker"
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "pass",
|
|
},
|
|
)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "options"
|
|
|
|
await hass.async_block_till_done()
|
|
assert mqtt_mock.async_connect.call_count == 0
|
|
|
|
yaml_mock.reset_mock()
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_DISCOVERY: True,
|
|
"discovery_prefix": "homeassistant",
|
|
"birth_enable": True,
|
|
"birth_topic": "ha_state/online",
|
|
"birth_payload": "online",
|
|
"birth_qos": 1,
|
|
"birth_retain": True,
|
|
"will_enable": True,
|
|
"will_topic": "ha_state/offline",
|
|
"will_payload": "offline",
|
|
"will_qos": 2,
|
|
"will_retain": True,
|
|
},
|
|
)
|
|
assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY
|
|
assert result["data"] == {}
|
|
assert config_entry.data == {
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "pass",
|
|
mqtt.CONF_DISCOVERY: True,
|
|
mqtt.CONF_DISCOVERY_PREFIX: "homeassistant",
|
|
mqtt.CONF_BIRTH_MESSAGE: {
|
|
mqtt.ATTR_TOPIC: "ha_state/online",
|
|
mqtt.ATTR_PAYLOAD: "online",
|
|
mqtt.ATTR_QOS: 1,
|
|
mqtt.ATTR_RETAIN: True,
|
|
},
|
|
mqtt.CONF_WILL_MESSAGE: {
|
|
mqtt.ATTR_TOPIC: "ha_state/offline",
|
|
mqtt.ATTR_PAYLOAD: "offline",
|
|
mqtt.ATTR_QOS: 2,
|
|
mqtt.ATTR_RETAIN: True,
|
|
},
|
|
}
|
|
|
|
await hass.async_block_till_done()
|
|
assert config_entry.title == "another-broker"
|
|
# assert that the entry was reloaded with the new config
|
|
assert yaml_mock.await_count
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"test_error",
|
|
[
|
|
"bad_certificate",
|
|
"bad_client_cert",
|
|
"bad_client_key",
|
|
"bad_client_cert_key",
|
|
"invalid_inclusion",
|
|
None,
|
|
],
|
|
)
|
|
async def test_bad_certificate(
|
|
hass,
|
|
mqtt_mock_entry_no_yaml_config,
|
|
mock_try_connection_success,
|
|
tmp_path,
|
|
mock_ssl_context,
|
|
test_error,
|
|
mock_process_uploaded_file,
|
|
):
|
|
"""Test bad certificate tests."""
|
|
# Mock certificate files
|
|
file_id = mock_process_uploaded_file.file_id
|
|
test_input = {
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_CERTIFICATE: file_id[mqtt.CONF_CERTIFICATE],
|
|
mqtt.CONF_CLIENT_CERT: file_id[mqtt.CONF_CLIENT_CERT],
|
|
mqtt.CONF_CLIENT_KEY: file_id[mqtt.CONF_CLIENT_KEY],
|
|
"set_ca_cert": True,
|
|
"set_client_cert": True,
|
|
}
|
|
set_client_cert = True
|
|
set_ca_cert = "custom"
|
|
tls_insecure = False
|
|
if test_error == "bad_certificate":
|
|
# CA chain is not loading
|
|
mock_ssl_context["context"]().load_verify_locations.side_effect = SSLError
|
|
elif test_error == "bad_client_cert":
|
|
# Client certificate is invalid
|
|
mock_ssl_context["load_pem_x509_certificate"].side_effect = ValueError
|
|
elif test_error == "bad_client_key":
|
|
# Client key file is invalid
|
|
mock_ssl_context["load_pem_private_key"].side_effect = ValueError
|
|
elif test_error == "bad_client_cert_key":
|
|
# Client key file file and certificate do not pair
|
|
mock_ssl_context["context"]().load_cert_chain.side_effect = SSLError
|
|
elif test_error == "invalid_inclusion":
|
|
# Client key file without client cert, client cert without key file
|
|
test_input.pop(mqtt.CONF_CLIENT_KEY)
|
|
|
|
mqtt_mock = await mqtt_mock_entry_no_yaml_config()
|
|
mock_try_connection.return_value = True
|
|
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
|
|
# Add at least one advanced option to get the full form
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
mqtt.CONF_CLIENT_ID: "custom1234",
|
|
mqtt.CONF_KEEPALIVE: 60,
|
|
mqtt.CONF_TLS_INSECURE: False,
|
|
mqtt.CONF_PROTOCOL: "3.1.1",
|
|
}
|
|
|
|
mqtt_mock.async_connect.reset_mock()
|
|
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "broker"
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_KEEPALIVE: 60,
|
|
"set_client_cert": set_client_cert,
|
|
"set_ca_cert": set_ca_cert,
|
|
mqtt.CONF_TLS_INSECURE: tls_insecure,
|
|
mqtt.CONF_PROTOCOL: "3.1.1",
|
|
mqtt.CONF_CLIENT_ID: "custom1234",
|
|
},
|
|
)
|
|
test_input["set_client_cert"] = set_client_cert
|
|
test_input["set_ca_cert"] = set_ca_cert
|
|
test_input["tls_insecure"] = tls_insecure
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input=test_input,
|
|
)
|
|
if test_error is not None:
|
|
assert result["errors"]["base"] == test_error
|
|
return
|
|
assert result["errors"] == {}
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"input_value, error",
|
|
[
|
|
("", True),
|
|
("-10", True),
|
|
("10", True),
|
|
("15", False),
|
|
("26", False),
|
|
("100", False),
|
|
],
|
|
)
|
|
async def test_keepalive_validation(
|
|
hass,
|
|
mqtt_mock_entry_no_yaml_config,
|
|
mock_try_connection,
|
|
mock_reload_after_entry_update,
|
|
input_value,
|
|
error,
|
|
):
|
|
"""Test validation of the keep alive option."""
|
|
|
|
test_input = {
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_KEEPALIVE: input_value,
|
|
}
|
|
|
|
mqtt_mock = await mqtt_mock_entry_no_yaml_config()
|
|
mock_try_connection.return_value = True
|
|
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
|
|
# Add at least one advanced option to get the full form
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
mqtt.CONF_CLIENT_ID: "custom1234",
|
|
}
|
|
|
|
mqtt_mock.async_connect.reset_mock()
|
|
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "broker"
|
|
|
|
if error:
|
|
with pytest.raises(vol.MultipleInvalid):
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input=test_input,
|
|
)
|
|
return
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input=test_input,
|
|
)
|
|
assert not result["errors"]
|
|
|
|
|
|
async def test_disable_birth_will(
|
|
hass,
|
|
mqtt_mock_entry_no_yaml_config,
|
|
mock_try_connection,
|
|
mock_reload_after_entry_update,
|
|
):
|
|
"""Test disabling birth and will."""
|
|
mqtt_mock = await mqtt_mock_entry_no_yaml_config()
|
|
mock_try_connection.return_value = True
|
|
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
}
|
|
|
|
mqtt_mock.async_connect.reset_mock()
|
|
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "broker"
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "pass",
|
|
},
|
|
)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "options"
|
|
|
|
await hass.async_block_till_done()
|
|
assert mqtt_mock.async_connect.call_count == 0
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_DISCOVERY: True,
|
|
mqtt.CONF_DISCOVERY_PREFIX: "homeassistant",
|
|
"birth_enable": False,
|
|
"birth_topic": "ha_state/online",
|
|
"birth_payload": "online",
|
|
"birth_qos": 1,
|
|
"birth_retain": True,
|
|
"will_enable": False,
|
|
"will_topic": "ha_state/offline",
|
|
"will_payload": "offline",
|
|
"will_qos": 2,
|
|
"will_retain": True,
|
|
},
|
|
)
|
|
assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY
|
|
assert result["data"] == {}
|
|
assert config_entry.data == {
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "pass",
|
|
mqtt.CONF_DISCOVERY: True,
|
|
mqtt.CONF_DISCOVERY_PREFIX: "homeassistant",
|
|
mqtt.CONF_BIRTH_MESSAGE: {},
|
|
mqtt.CONF_WILL_MESSAGE: {},
|
|
}
|
|
|
|
await hass.async_block_till_done()
|
|
# assert that the entry was reloaded with the new config
|
|
assert mock_reload_after_entry_update.call_count == 1
|
|
|
|
|
|
async def test_invalid_discovery_prefix(
|
|
hass,
|
|
mqtt_mock_entry_no_yaml_config,
|
|
mock_try_connection,
|
|
mock_reload_after_entry_update,
|
|
):
|
|
"""Test setting an invalid discovery prefix."""
|
|
mqtt_mock = await mqtt_mock_entry_no_yaml_config()
|
|
mock_try_connection.return_value = True
|
|
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
mqtt.CONF_DISCOVERY: True,
|
|
mqtt.CONF_DISCOVERY_PREFIX: "homeassistant",
|
|
}
|
|
|
|
mqtt_mock.async_connect.reset_mock()
|
|
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "broker"
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
},
|
|
)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "options"
|
|
|
|
await hass.async_block_till_done()
|
|
assert mqtt_mock.async_connect.call_count == 0
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_DISCOVERY: True,
|
|
mqtt.CONF_DISCOVERY_PREFIX: "homeassistant#invalid",
|
|
},
|
|
)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "options"
|
|
assert result["errors"]["base"] == "bad_discovery_prefix"
|
|
assert config_entry.data == {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
mqtt.CONF_DISCOVERY: True,
|
|
mqtt.CONF_DISCOVERY_PREFIX: "homeassistant",
|
|
}
|
|
|
|
await hass.async_block_till_done()
|
|
# assert that the entry was not reloaded with the new config
|
|
assert mock_reload_after_entry_update.call_count == 0
|
|
|
|
|
|
def get_default(schema, key):
|
|
"""Get default value for key in voluptuous schema."""
|
|
for k in schema.keys():
|
|
if k == key:
|
|
if k.default == vol.UNDEFINED:
|
|
return None
|
|
return k.default()
|
|
|
|
|
|
def get_suggested(schema, key):
|
|
"""Get suggested value for key in voluptuous schema."""
|
|
for k in schema.keys():
|
|
if k == key:
|
|
if k.description is None or "suggested_value" not in k.description:
|
|
return None
|
|
return k.description["suggested_value"]
|
|
|
|
|
|
async def test_option_flow_default_suggested_values(
|
|
hass,
|
|
mqtt_mock_entry_no_yaml_config,
|
|
mock_try_connection_success,
|
|
mock_reload_after_entry_update,
|
|
):
|
|
"""Test config flow options has default/suggested values."""
|
|
await mqtt_mock_entry_no_yaml_config()
|
|
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "pass",
|
|
mqtt.CONF_DISCOVERY: True,
|
|
mqtt.CONF_BIRTH_MESSAGE: {
|
|
mqtt.ATTR_TOPIC: "ha_state/online",
|
|
mqtt.ATTR_PAYLOAD: "online",
|
|
mqtt.ATTR_QOS: 1,
|
|
mqtt.ATTR_RETAIN: True,
|
|
},
|
|
mqtt.CONF_WILL_MESSAGE: {
|
|
mqtt.ATTR_TOPIC: "ha_state/offline",
|
|
mqtt.ATTR_PAYLOAD: "offline",
|
|
mqtt.ATTR_QOS: 2,
|
|
mqtt.ATTR_RETAIN: False,
|
|
},
|
|
}
|
|
|
|
# Test default/suggested values from config
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "broker"
|
|
defaults = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
}
|
|
suggested = {
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "pass",
|
|
}
|
|
for k, v in defaults.items():
|
|
assert get_default(result["data_schema"].schema, k) == v
|
|
for k, v in suggested.items():
|
|
assert get_suggested(result["data_schema"].schema, k) == v
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_USERNAME: "us3r",
|
|
mqtt.CONF_PASSWORD: "p4ss",
|
|
},
|
|
)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "options"
|
|
defaults = {
|
|
mqtt.CONF_DISCOVERY: True,
|
|
"birth_qos": 1,
|
|
"birth_retain": True,
|
|
"will_qos": 2,
|
|
"will_retain": False,
|
|
}
|
|
suggested = {
|
|
"birth_topic": "ha_state/online",
|
|
"birth_payload": "online",
|
|
"will_topic": "ha_state/offline",
|
|
"will_payload": "offline",
|
|
}
|
|
for k, v in defaults.items():
|
|
assert get_default(result["data_schema"].schema, k) == v
|
|
for k, v in suggested.items():
|
|
assert get_suggested(result["data_schema"].schema, k) == v
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_DISCOVERY: False,
|
|
"birth_topic": "ha_state/onl1ne",
|
|
"birth_payload": "onl1ne",
|
|
"birth_qos": 2,
|
|
"birth_retain": False,
|
|
"will_topic": "ha_state/offl1ne",
|
|
"will_payload": "offl1ne",
|
|
"will_qos": 1,
|
|
"will_retain": True,
|
|
},
|
|
)
|
|
assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY
|
|
|
|
# Test updated default/suggested values from config
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "broker"
|
|
defaults = {
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
}
|
|
suggested = {
|
|
mqtt.CONF_USERNAME: "us3r",
|
|
mqtt.CONF_PASSWORD: "p4ss",
|
|
}
|
|
for k, v in defaults.items():
|
|
assert get_default(result["data_schema"].schema, k) == v
|
|
for k, v in suggested.items():
|
|
assert get_suggested(result["data_schema"].schema, k) == v
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={mqtt.CONF_BROKER: "another-broker", mqtt.CONF_PORT: 2345},
|
|
)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "options"
|
|
defaults = {
|
|
mqtt.CONF_DISCOVERY: False,
|
|
"birth_qos": 2,
|
|
"birth_retain": False,
|
|
"will_qos": 1,
|
|
"will_retain": True,
|
|
}
|
|
suggested = {
|
|
"birth_topic": "ha_state/onl1ne",
|
|
"birth_payload": "onl1ne",
|
|
"will_topic": "ha_state/offl1ne",
|
|
"will_payload": "offl1ne",
|
|
}
|
|
for k, v in defaults.items():
|
|
assert get_default(result["data_schema"].schema, k) == v
|
|
for k, v in suggested.items():
|
|
assert get_suggested(result["data_schema"].schema, k) == v
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_DISCOVERY: True,
|
|
"birth_topic": "ha_state/onl1ne",
|
|
"birth_payload": "onl1ne",
|
|
"birth_qos": 2,
|
|
"birth_retain": False,
|
|
"will_topic": "ha_state/offl1ne",
|
|
"will_payload": "offl1ne",
|
|
"will_qos": 1,
|
|
"will_retain": True,
|
|
},
|
|
)
|
|
assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY
|
|
|
|
# Make sure all MQTT related jobs are done before ending the test
|
|
await hass.async_block_till_done()
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"advanced_options, step_id", [(False, "options"), (True, "broker")]
|
|
)
|
|
async def test_skipping_advanced_options(
|
|
hass,
|
|
mqtt_mock_entry_no_yaml_config,
|
|
mock_try_connection,
|
|
mock_reload_after_entry_update,
|
|
advanced_options,
|
|
step_id,
|
|
):
|
|
"""Test advanced options option."""
|
|
|
|
test_input = {
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
"advanced_options": advanced_options,
|
|
}
|
|
|
|
mqtt_mock = await mqtt_mock_entry_no_yaml_config()
|
|
mock_try_connection.return_value = True
|
|
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
|
|
# Initiate with a basic setup
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
}
|
|
|
|
mqtt_mock.async_connect.reset_mock()
|
|
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "broker"
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input=test_input,
|
|
)
|
|
assert result["step_id"] == step_id
|
|
|
|
|
|
async def test_options_user_connection_fails(hass, mock_try_connection_time_out):
|
|
"""Test if connection cannot be made."""
|
|
config_entry = MockConfigEntry(domain=mqtt.DOMAIN)
|
|
config_entry.add_to_hass(hass)
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
}
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == "form"
|
|
|
|
mock_try_connection_time_out.reset_mock()
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={mqtt.CONF_BROKER: "bad-broker", mqtt.CONF_PORT: 2345},
|
|
)
|
|
|
|
assert result["type"] == "form"
|
|
assert result["errors"]["base"] == "cannot_connect"
|
|
|
|
# Check we tried the connection
|
|
assert len(mock_try_connection_time_out.mock_calls)
|
|
# Check config entry did not update
|
|
assert config_entry.data == {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
}
|
|
|
|
|
|
async def test_options_bad_birth_message_fails(hass, mock_try_connection):
|
|
"""Test bad birth message."""
|
|
config_entry = MockConfigEntry(domain=mqtt.DOMAIN)
|
|
config_entry.add_to_hass(hass)
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
}
|
|
|
|
mock_try_connection.return_value = True
|
|
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == "form"
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={mqtt.CONF_BROKER: "another-broker", mqtt.CONF_PORT: 2345},
|
|
)
|
|
|
|
assert result["type"] == "form"
|
|
assert result["step_id"] == "options"
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={"birth_topic": "ha_state/online/#"},
|
|
)
|
|
assert result["type"] == "form"
|
|
assert result["errors"]["base"] == "bad_birth"
|
|
|
|
# Check config entry did not update
|
|
assert config_entry.data == {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
}
|
|
|
|
|
|
async def test_options_bad_will_message_fails(hass, mock_try_connection):
|
|
"""Test bad will message."""
|
|
config_entry = MockConfigEntry(domain=mqtt.DOMAIN)
|
|
config_entry.add_to_hass(hass)
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
}
|
|
|
|
mock_try_connection.return_value = True
|
|
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == "form"
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={mqtt.CONF_BROKER: "another-broker", mqtt.CONF_PORT: 2345},
|
|
)
|
|
|
|
assert result["type"] == "form"
|
|
assert result["step_id"] == "options"
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={"will_topic": "ha_state/offline/#"},
|
|
)
|
|
assert result["type"] == "form"
|
|
assert result["errors"]["base"] == "bad_will"
|
|
|
|
# Check config entry did not update
|
|
assert config_entry.data == {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
}
|
|
|
|
|
|
async def test_try_connection_with_advanced_parameters(
|
|
hass,
|
|
mock_try_connection_success,
|
|
tmp_path,
|
|
mock_ssl_context,
|
|
mock_process_uploaded_file,
|
|
):
|
|
"""Test config flow with advanced parameters from config."""
|
|
|
|
with open(tmp_path / "client.crt", "wb") as certfile:
|
|
certfile.write(MOCK_CLIENT_CERT)
|
|
with open(tmp_path / "client.key", "wb") as keyfile:
|
|
keyfile.write(MOCK_CLIENT_KEY)
|
|
|
|
config = {
|
|
"certificate": "auto",
|
|
"tls_insecure": True,
|
|
"client_cert": str(tmp_path / "client.crt"),
|
|
"client_key": str(tmp_path / "client.key"),
|
|
}
|
|
new_yaml_config_file = tmp_path / "configuration.yaml"
|
|
new_yaml_config = yaml.dump({mqtt.DOMAIN: config})
|
|
new_yaml_config_file.write_text(new_yaml_config)
|
|
assert new_yaml_config_file.read_text() == new_yaml_config
|
|
|
|
config_entry = MockConfigEntry(domain=mqtt.DOMAIN)
|
|
config_entry.add_to_hass(hass)
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "pass",
|
|
mqtt.CONF_TRANSPORT: "websockets",
|
|
mqtt.CONF_WS_PATH: "/path/",
|
|
mqtt.CONF_WS_HEADERS: {"h1": "v1", "h2": "v2"},
|
|
mqtt.CONF_KEEPALIVE: 30,
|
|
mqtt.CONF_DISCOVERY: True,
|
|
mqtt.CONF_BIRTH_MESSAGE: {
|
|
mqtt.ATTR_TOPIC: "ha_state/online",
|
|
mqtt.ATTR_PAYLOAD: "online",
|
|
mqtt.ATTR_QOS: 1,
|
|
mqtt.ATTR_RETAIN: True,
|
|
},
|
|
mqtt.CONF_WILL_MESSAGE: {
|
|
mqtt.ATTR_TOPIC: "ha_state/offline",
|
|
mqtt.ATTR_PAYLOAD: "offline",
|
|
mqtt.ATTR_QOS: 2,
|
|
mqtt.ATTR_RETAIN: False,
|
|
},
|
|
}
|
|
|
|
with patch.object(hass_config, "YAML_CONFIG_FILE", new_yaml_config_file):
|
|
await async_setup_component(hass, mqtt.DOMAIN, {mqtt.DOMAIN: config})
|
|
await hass.async_block_till_done()
|
|
# Test default/suggested values from config
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["step_id"] == "broker"
|
|
defaults = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
"set_client_cert": True,
|
|
"set_ca_cert": "auto",
|
|
}
|
|
suggested = {
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "pass",
|
|
mqtt.CONF_TLS_INSECURE: True,
|
|
mqtt.CONF_PROTOCOL: "3.1.1",
|
|
mqtt.CONF_TRANSPORT: "websockets",
|
|
mqtt.CONF_WS_PATH: "/path/",
|
|
mqtt.CONF_WS_HEADERS: '{"h1":"v1","h2":"v2"}',
|
|
}
|
|
for k, v in defaults.items():
|
|
assert get_default(result["data_schema"].schema, k) == v
|
|
for k, v in suggested.items():
|
|
assert get_suggested(result["data_schema"].schema, k) == v
|
|
|
|
# test the client cert and key were migrated to the entry
|
|
assert config_entry.data[mqtt.CONF_CLIENT_CERT] == MOCK_CLIENT_CERT.decode(
|
|
"utf-8"
|
|
)
|
|
assert config_entry.data[mqtt.CONF_CLIENT_KEY] == MOCK_CLIENT_KEY.decode(
|
|
"utf-8"
|
|
)
|
|
assert config_entry.data[mqtt.CONF_CERTIFICATE] == "auto"
|
|
|
|
# test we can change username and password
|
|
# as it was configured as auto in configuration.yaml is is migrated now
|
|
mock_try_connection_success.reset_mock()
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_BROKER: "another-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_USERNAME: "us3r",
|
|
mqtt.CONF_PASSWORD: "p4ss",
|
|
"set_ca_cert": "auto",
|
|
"set_client_cert": True,
|
|
mqtt.CONF_TLS_INSECURE: True,
|
|
mqtt.CONF_TRANSPORT: "websockets",
|
|
mqtt.CONF_WS_PATH: "/new/path",
|
|
mqtt.CONF_WS_HEADERS: '{"h3": "v3"}',
|
|
},
|
|
)
|
|
assert result["type"] == data_entry_flow.FlowResultType.FORM
|
|
assert result["errors"] == {}
|
|
assert result["step_id"] == "options"
|
|
await hass.async_block_till_done()
|
|
|
|
# check if the username and password was set from config flow and not from configuration.yaml
|
|
assert mock_try_connection_success.username_pw_set.mock_calls[0][1] == (
|
|
"us3r",
|
|
"p4ss",
|
|
)
|
|
# check if tls_insecure_set is called
|
|
assert mock_try_connection_success.tls_insecure_set.mock_calls[0][1] == (True,)
|
|
|
|
# check if the ca certificate settings were not set during connection test
|
|
assert mock_try_connection_success.tls_set.mock_calls[0].kwargs[
|
|
"certfile"
|
|
] == mqtt.util.get_file_path(mqtt.CONF_CLIENT_CERT)
|
|
assert mock_try_connection_success.tls_set.mock_calls[0].kwargs[
|
|
"keyfile"
|
|
] == mqtt.util.get_file_path(mqtt.CONF_CLIENT_KEY)
|
|
|
|
# check if websockets options are set
|
|
assert mock_try_connection_success.ws_set_options.mock_calls[0][1] == (
|
|
"/new/path",
|
|
{"h3": "v3"},
|
|
)
|
|
|
|
# Accept default option
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={},
|
|
)
|
|
assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY
|
|
await hass.async_block_till_done()
|
|
|
|
|
|
async def test_setup_with_advanced_settings(
|
|
hass, mock_try_connection, tmp_path, mock_ssl_context, mock_process_uploaded_file
|
|
):
|
|
"""Test config flow setup with advanced parameters."""
|
|
file_id = mock_process_uploaded_file.file_id
|
|
|
|
config_entry = MockConfigEntry(domain=mqtt.DOMAIN)
|
|
config_entry.add_to_hass(hass)
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
}
|
|
|
|
mock_try_connection.return_value = True
|
|
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == "form"
|
|
assert result["step_id"] == "broker"
|
|
assert result["data_schema"].schema["advanced_options"]
|
|
|
|
# first iteration, basic settings
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "secret",
|
|
"advanced_options": True,
|
|
},
|
|
)
|
|
assert result["type"] == "form"
|
|
assert result["step_id"] == "broker"
|
|
assert "advanced_options" not in result["data_schema"].schema
|
|
assert result["data_schema"].schema[mqtt.CONF_CLIENT_ID]
|
|
assert result["data_schema"].schema[mqtt.CONF_KEEPALIVE]
|
|
assert result["data_schema"].schema["set_client_cert"]
|
|
assert result["data_schema"].schema["set_ca_cert"]
|
|
assert result["data_schema"].schema[mqtt.CONF_TLS_INSECURE]
|
|
assert result["data_schema"].schema[mqtt.CONF_PROTOCOL]
|
|
assert result["data_schema"].schema[mqtt.CONF_TRANSPORT]
|
|
assert mqtt.CONF_CLIENT_CERT not in result["data_schema"].schema
|
|
assert mqtt.CONF_CLIENT_KEY not in result["data_schema"].schema
|
|
|
|
# second iteration, advanced settings with request for client cert
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "secret",
|
|
mqtt.CONF_KEEPALIVE: 30,
|
|
"set_ca_cert": "auto",
|
|
"set_client_cert": True,
|
|
mqtt.CONF_TLS_INSECURE: True,
|
|
mqtt.CONF_PROTOCOL: "3.1.1",
|
|
mqtt.CONF_TRANSPORT: "websockets",
|
|
},
|
|
)
|
|
assert result["type"] == "form"
|
|
assert result["step_id"] == "broker"
|
|
assert "advanced_options" not in result["data_schema"].schema
|
|
assert result["data_schema"].schema[mqtt.CONF_CLIENT_ID]
|
|
assert result["data_schema"].schema[mqtt.CONF_KEEPALIVE]
|
|
assert result["data_schema"].schema["set_client_cert"]
|
|
assert result["data_schema"].schema["set_ca_cert"]
|
|
assert result["data_schema"].schema[mqtt.CONF_TLS_INSECURE]
|
|
assert result["data_schema"].schema[mqtt.CONF_PROTOCOL]
|
|
assert result["data_schema"].schema[mqtt.CONF_CLIENT_CERT]
|
|
assert result["data_schema"].schema[mqtt.CONF_CLIENT_KEY]
|
|
assert result["data_schema"].schema[mqtt.CONF_TRANSPORT]
|
|
assert result["data_schema"].schema[mqtt.CONF_WS_PATH]
|
|
assert result["data_schema"].schema[mqtt.CONF_WS_HEADERS]
|
|
|
|
# third iteration, advanced settings with client cert and key set and bad json payload
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "secret",
|
|
mqtt.CONF_KEEPALIVE: 30,
|
|
"set_ca_cert": "auto",
|
|
"set_client_cert": True,
|
|
mqtt.CONF_CLIENT_CERT: file_id[mqtt.CONF_CLIENT_CERT],
|
|
mqtt.CONF_CLIENT_KEY: file_id[mqtt.CONF_CLIENT_KEY],
|
|
mqtt.CONF_TLS_INSECURE: True,
|
|
mqtt.CONF_TRANSPORT: "websockets",
|
|
mqtt.CONF_WS_PATH: "/custom_path/",
|
|
mqtt.CONF_WS_HEADERS: '{"header_1": "content_header_1", "header_2": "content_header_2"',
|
|
},
|
|
)
|
|
|
|
assert result["type"] == "form"
|
|
assert result["step_id"] == "broker"
|
|
assert result["errors"]["base"] == "bad_ws_headers"
|
|
|
|
# fourth iteration, advanced settings with client cert and key set
|
|
# and correct json payload for ws_headers
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "secret",
|
|
mqtt.CONF_KEEPALIVE: 30,
|
|
"set_ca_cert": "auto",
|
|
"set_client_cert": True,
|
|
mqtt.CONF_CLIENT_CERT: file_id[mqtt.CONF_CLIENT_CERT],
|
|
mqtt.CONF_CLIENT_KEY: file_id[mqtt.CONF_CLIENT_KEY],
|
|
mqtt.CONF_TLS_INSECURE: True,
|
|
mqtt.CONF_TRANSPORT: "websockets",
|
|
mqtt.CONF_WS_PATH: "/custom_path/",
|
|
mqtt.CONF_WS_HEADERS: '{"header_1": "content_header_1", "header_2": "content_header_2"}',
|
|
},
|
|
)
|
|
|
|
assert result["type"] == "form"
|
|
assert result["step_id"] == "options"
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_DISCOVERY: True,
|
|
mqtt.CONF_DISCOVERY_PREFIX: "homeassistant_test",
|
|
},
|
|
)
|
|
assert result["type"] == "create_entry"
|
|
|
|
# Check config entry result
|
|
assert config_entry.data == {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 2345,
|
|
mqtt.CONF_USERNAME: "user",
|
|
mqtt.CONF_PASSWORD: "secret",
|
|
mqtt.CONF_KEEPALIVE: 30,
|
|
mqtt.CONF_CLIENT_CERT: "## mock client certificate file ##",
|
|
mqtt.CONF_CLIENT_KEY: "## mock key file ##",
|
|
"tls_insecure": True,
|
|
mqtt.CONF_TRANSPORT: "websockets",
|
|
mqtt.CONF_WS_PATH: "/custom_path/",
|
|
mqtt.CONF_WS_HEADERS: {
|
|
"header_1": "content_header_1",
|
|
"header_2": "content_header_2",
|
|
},
|
|
mqtt.CONF_CERTIFICATE: "auto",
|
|
mqtt.CONF_DISCOVERY: True,
|
|
mqtt.CONF_DISCOVERY_PREFIX: "homeassistant_test",
|
|
}
|
|
|
|
|
|
async def test_change_websockets_transport_to_tcp(
|
|
hass, mock_try_connection, tmp_path, mock_ssl_context, mock_process_uploaded_file
|
|
):
|
|
"""Test option flow setup with websockets transport settings."""
|
|
config_entry = MockConfigEntry(domain=mqtt.DOMAIN)
|
|
config_entry.add_to_hass(hass)
|
|
config_entry.data = {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
mqtt.CONF_TRANSPORT: "websockets",
|
|
mqtt.CONF_WS_HEADERS: {"header_1": "custom_header1"},
|
|
mqtt.CONF_WS_PATH: "/some_path",
|
|
}
|
|
|
|
mock_try_connection.return_value = True
|
|
|
|
result = await hass.config_entries.options.async_init(config_entry.entry_id)
|
|
assert result["type"] == "form"
|
|
assert result["step_id"] == "broker"
|
|
assert result["data_schema"].schema["transport"]
|
|
assert result["data_schema"].schema["ws_path"]
|
|
assert result["data_schema"].schema["ws_headers"]
|
|
|
|
# Change transport to tcp
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
mqtt.CONF_TRANSPORT: "tcp",
|
|
mqtt.CONF_WS_HEADERS: '{"header_1": "custom_header1"}',
|
|
mqtt.CONF_WS_PATH: "/some_path",
|
|
},
|
|
)
|
|
assert result["type"] == "form"
|
|
assert result["step_id"] == "options"
|
|
|
|
result = await hass.config_entries.options.async_configure(
|
|
result["flow_id"],
|
|
user_input={
|
|
mqtt.CONF_DISCOVERY: True,
|
|
mqtt.CONF_DISCOVERY_PREFIX: "homeassistant_test",
|
|
},
|
|
)
|
|
assert result["type"] == "create_entry"
|
|
|
|
# Check config entry result
|
|
assert config_entry.data == {
|
|
mqtt.CONF_BROKER: "test-broker",
|
|
mqtt.CONF_PORT: 1234,
|
|
mqtt.CONF_TRANSPORT: "tcp",
|
|
mqtt.CONF_DISCOVERY: True,
|
|
mqtt.CONF_DISCOVERY_PREFIX: "homeassistant_test",
|
|
}
|