starred/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2026-09-24 07:25:52 -05:00
Code Issues Packages Projects Releases Wiki Activity

Use explicit DOMAIN import in mqtt tests (#173093)

Browse Source
This commit is contained in:
epenet
2026-06-05 14:01:28 +02:00
committed by GitHub
parent 6c72d4337d
commit 4dbe58afc6
41 changed files with 883 additions and 865 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/components/mqtt/common.py
+73 -74
View File
@@ -17,6 +17,7 @@ from homeassistant import config as module_hass_config
from homeassistant.components import mqtt
from homeassistant.components.mqtt import debug_info
from homeassistant.components.mqtt.const import (
DOMAIN,
MQTT_CONNECTION_STATE,
SUPPORTED_COMPONENTS,
)
@@ -1004,12 +1005,10 @@ def help_custom_config(
config: ConfigType = copy.deepcopy(mqtt_base_config)
entity_instances: list[ConfigType] = []
for instance in mqtt_entity_configs:
base: ConfigType = copy.deepcopy(
mqtt_base_config[mqtt.DOMAIN][mqtt_entity_domain]
)
base: ConfigType = copy.deepcopy(mqtt_base_config[DOMAIN][mqtt_entity_domain])
base.update(instance)
entity_instances.append(base)
config[mqtt.DOMAIN][mqtt_entity_domain] = entity_instances
config[DOMAIN][mqtt_entity_domain] = entity_instances
return config
@@ -1038,7 +1037,7 @@ async def help_test_availability_without_topic(
config: ConfigType,
) -> None:
"""Test availability without defined availability topic."""
assert "availability_topic" not in config[mqtt.DOMAIN][domain]
assert "availability_topic" not in config[DOMAIN][domain]
await mqtt_mock_entry()
await hass.async_block_till_done()
@@ -1061,7 +1060,7 @@ async def help_test_default_availability_payload(
"""
# Add availability settings to config
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["availability_topic"] = "availability-topic"
config[DOMAIN][domain]["availability_topic"] = "availability-topic"
with patch("homeassistant.config.load_yaml_config_file", return_value=config):
await mqtt_mock_entry()
@@ -1108,7 +1107,7 @@ async def help_test_default_availability_list_payload(
"""
# Add availability settings to config
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["availability"] = [
config[DOMAIN][domain]["availability"] = [
{"topic": "availability-topic1"},
{"topic": "availability-topic2"},
]
@@ -1167,8 +1166,8 @@ async def help_test_default_availability_list_payload_all(
"""
# Add availability settings to config
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["availability_mode"] = "all"
config[mqtt.DOMAIN][domain]["availability"] = [
config[DOMAIN][domain]["availability_mode"] = "all"
config[DOMAIN][domain]["availability"] = [
{"topic": "availability-topic1"},
{"topic": "availability-topic2"},
]
@@ -1228,8 +1227,8 @@ async def help_test_default_availability_list_payload_any(
"""
# Add availability settings to config
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["availability_mode"] = "any"
config[mqtt.DOMAIN][domain]["availability"] = [
config[DOMAIN][domain]["availability_mode"] = "any"
config[DOMAIN][domain]["availability"] = [
{"topic": "availability-topic1"},
{"topic": "availability-topic2"},
]
@@ -1284,10 +1283,10 @@ async def help_test_default_availability_list_single(
"""
# Add availability settings to config
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["availability"] = [
config[DOMAIN][domain]["availability"] = [
{"topic": "availability-topic1"},
]
config[mqtt.DOMAIN][domain]["availability_topic"] = "availability-topic"
config[DOMAIN][domain]["availability_topic"] = "availability-topic"
with (
patch("homeassistant.config.load_yaml_config_file", return_value=config),
@@ -1316,9 +1315,9 @@ async def help_test_custom_availability_payload(
"""
# Add availability settings to config
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["availability_topic"] = "availability-topic"
config[mqtt.DOMAIN][domain]["payload_available"] = "good"
config[mqtt.DOMAIN][domain]["payload_not_available"] = "nogood"
config[DOMAIN][domain]["availability_topic"] = "availability-topic"
config[DOMAIN][domain]["payload_available"] = "good"
config[DOMAIN][domain]["payload_not_available"] = "nogood"
with patch("homeassistant.config.load_yaml_config_file", return_value=config):
await mqtt_mock_entry()
@@ -1362,17 +1361,17 @@ async def help_test_discovery_update_availability(
await mqtt_mock_entry()
# Add availability settings to config
config1 = copy.deepcopy(config)
config1[mqtt.DOMAIN][domain]["availability_topic"] = "availability-topic1"
config1[DOMAIN][domain]["availability_topic"] = "availability-topic1"
config2 = copy.deepcopy(config)
config2[mqtt.DOMAIN][domain]["availability"] = [
config2[DOMAIN][domain]["availability"] = [
{"topic": "availability-topic2"},
{"topic": "availability-topic3"},
]
config3 = copy.deepcopy(config)
config3[mqtt.DOMAIN][domain]["availability_topic"] = "availability-topic4"
data1 = json.dumps(config1[mqtt.DOMAIN][domain])
data2 = json.dumps(config2[mqtt.DOMAIN][domain])
data3 = json.dumps(config3[mqtt.DOMAIN][domain])
config3[DOMAIN][domain]["availability_topic"] = "availability-topic4"
data1 = json.dumps(config1[DOMAIN][domain])
data2 = json.dumps(config2[DOMAIN][domain])
data3 = json.dumps(config3[DOMAIN][domain])
async_fire_mqtt_message(hass, f"homeassistant/{domain}/bla/config", data1)
await hass.async_block_till_done()
@@ -1439,7 +1438,7 @@ async def help_test_setting_attribute_via_mqtt_json_message(
"""
# Add JSON attributes settings to config
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["json_attributes_topic"] = "attr-topic"
config[DOMAIN][domain]["json_attributes_topic"] = "attr-topic"
with patch("homeassistant.config.load_yaml_config_file", return_value=config):
await mqtt_mock_entry()
@@ -1465,8 +1464,8 @@ async def help_test_setting_blocked_attribute_via_mqtt_json_message(
# Add JSON attributes settings to config
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["json_attributes_topic"] = "attr-topic"
data = json.dumps(config[mqtt.DOMAIN][domain])
config[DOMAIN][domain]["json_attributes_topic"] = "attr-topic"
data = json.dumps(config[DOMAIN][domain])
async_fire_mqtt_message(hass, f"homeassistant/{domain}/bla/config", data)
await hass.async_block_till_done()
val = "abc123"
@@ -1494,8 +1493,8 @@ async def help_test_setting_attribute_with_template(
"""
# Add JSON attributes settings to config
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["json_attributes_topic"] = "attr-topic"
config[mqtt.DOMAIN][domain]["json_attributes_template"] = (
config[DOMAIN][domain]["json_attributes_topic"] = "attr-topic"
config[DOMAIN][domain]["json_attributes_template"] = (
"{{ value_json['Timer1'] | tojson }}"
)
with patch("homeassistant.config.load_yaml_config_file", return_value=config):
@@ -1524,7 +1523,7 @@ async def help_test_update_with_json_attrs_not_dict(
"""
# Add JSON attributes settings to config
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["json_attributes_topic"] = "attr-topic"
config[DOMAIN][domain]["json_attributes_topic"] = "attr-topic"
with patch("homeassistant.config.load_yaml_config_file", return_value=config):
await mqtt_mock_entry()
@@ -1548,7 +1547,7 @@ async def help_test_update_with_json_attrs_bad_json(
"""
# Add JSON attributes settings to config
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["json_attributes_topic"] = "attr-topic"
config[DOMAIN][domain]["json_attributes_topic"] = "attr-topic"
with patch("homeassistant.config.load_yaml_config_file", return_value=config):
await mqtt_mock_entry()
@@ -1572,11 +1571,11 @@ async def help_test_discovery_update_attr(
await mqtt_mock_entry()
# Add JSON attributes settings to config
config1 = copy.deepcopy(config)
config1[mqtt.DOMAIN][domain]["json_attributes_topic"] = "attr-topic1"
config1[DOMAIN][domain]["json_attributes_topic"] = "attr-topic1"
config2 = copy.deepcopy(config)
config2[mqtt.DOMAIN][domain]["json_attributes_topic"] = "attr-topic2"
data1 = json.dumps(config1[mqtt.DOMAIN][domain])
data2 = json.dumps(config2[mqtt.DOMAIN][domain])
config2[DOMAIN][domain]["json_attributes_topic"] = "attr-topic2"
data1 = json.dumps(config1[DOMAIN][domain])
data2 = json.dumps(config2[DOMAIN][domain])
async_fire_mqtt_message(hass, f"homeassistant/{domain}/bla/config", data1)
await hass.async_block_till_done()
@@ -1895,7 +1894,7 @@ async def help_test_entity_device_info_with_identifier(
"""
await mqtt_mock_entry()
# Add device settings to config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_ID)
config["unique_id"] = "veryunique"
@@ -1931,7 +1930,7 @@ async def help_test_entity_device_info_with_connection(
"""
await mqtt_mock_entry()
# Add device settings to config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_MAC)
config["unique_id"] = "veryunique"
@@ -1966,7 +1965,7 @@ async def help_test_entity_device_info_remove(
"""Test device registry remove."""
await mqtt_mock_entry()
# Add device settings to config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_ID)
config["unique_id"] = "veryunique"
@@ -1979,14 +1978,14 @@ async def help_test_entity_device_info_remove(
device = dev_registry.async_get_device(identifiers={("mqtt", "helloworld")})
assert device is not None
assert ent_registry.async_get_entity_id(domain, mqtt.DOMAIN, "veryunique")
assert ent_registry.async_get_entity_id(domain, DOMAIN, "veryunique")
async_fire_mqtt_message(hass, f"homeassistant/{domain}/bla/config", "")
await hass.async_block_till_done()
device = dev_registry.async_get_device(identifiers={("mqtt", "helloworld")})
assert device is None
assert not ent_registry.async_get_entity_id(domain, mqtt.DOMAIN, "veryunique")
assert not ent_registry.async_get_entity_id(domain, DOMAIN, "veryunique")
async def help_test_entity_device_info_update(
@@ -2001,7 +2000,7 @@ async def help_test_entity_device_info_update(
"""
await mqtt_mock_entry()
# Add device settings to config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_ID)
config["unique_id"] = "veryunique"
@@ -2039,7 +2038,7 @@ async def help_test_entity_name(
"""
await mqtt_mock_entry()
# Add device settings to config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_ID)
config["unique_id"] = "veryunique"
expected_entity_name = "test"
@@ -2074,12 +2073,12 @@ async def help_test_entity_id_update_subscriptions(
"""Test MQTT subscriptions are managed when entity_id is updated."""
# Add unique_id to config
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["unique_id"] = "TOTALLY_UNIQUE"
config[DOMAIN][domain]["unique_id"] = "TOTALLY_UNIQUE"
if topics is None:
# Add default topics to config
config[mqtt.DOMAIN][domain]["availability_topic"] = "avty-topic"
config[mqtt.DOMAIN][domain]["state_topic"] = "test-topic"
config[DOMAIN][domain]["availability_topic"] = "avty-topic"
config[DOMAIN][domain]["state_topic"] = "test-topic"
topics = ["avty-topic", "test-topic"]
assert len(topics) > 0
entity_registry = er.async_get(hass)
@@ -2130,15 +2129,15 @@ async def help_test_entity_id_update_discovery_update(
# Add unique_id to config
await mqtt_mock_entry()
config = copy.deepcopy(config)
config[mqtt.DOMAIN][domain]["unique_id"] = "TOTALLY_UNIQUE"
config[DOMAIN][domain]["unique_id"] = "TOTALLY_UNIQUE"
if topic is None:
# Add default topic to config
config[mqtt.DOMAIN][domain]["availability_topic"] = "avty-topic"
config[DOMAIN][domain]["availability_topic"] = "avty-topic"
topic = "avty-topic"
entity_registry = er.async_get(hass)
data = json.dumps(config[mqtt.DOMAIN][domain])
data = json.dumps(config[DOMAIN][domain])
async_fire_mqtt_message(hass, f"homeassistant/{domain}/bla/config", data)
await hass.async_block_till_done()
@@ -2155,8 +2154,8 @@ async def help_test_entity_id_update_discovery_update(
)
await hass.async_block_till_done()
config[mqtt.DOMAIN][domain]["availability_topic"] = f"{topic}_2"
data = json.dumps(config[mqtt.DOMAIN][domain])
config[DOMAIN][domain]["availability_topic"] = f"{topic}_2"
data = json.dumps(config[DOMAIN][domain])
async_fire_mqtt_message(hass, f"homeassistant/{domain}/bla/config", data)
await hass.async_block_till_done()
assert len(hass.states.async_entity_ids(domain)) == 1
@@ -2178,7 +2177,7 @@ async def help_test_entity_debug_info(
"""
await mqtt_mock_entry()
# Add device settings to config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_ID)
config["unique_id"] = "veryunique"
config["platform"] = "mqtt"
@@ -2219,7 +2218,7 @@ async def help_test_entity_debug_info_max_messages(
"""
await mqtt_mock_entry()
# Add device settings to config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_ID)
config["unique_id"] = "veryunique"
@@ -2282,7 +2281,7 @@ async def help_test_entity_debug_info_message(
"""
# Add device settings to config
await mqtt_mock_entry()
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_ID)
config["unique_id"] = "veryunique"
@@ -2385,7 +2384,7 @@ async def help_test_entity_debug_info_remove(
"""
await mqtt_mock_entry()
# Add device settings to config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_ID)
config["unique_id"] = "veryunique"
config["platform"] = "mqtt"
@@ -2438,7 +2437,7 @@ async def help_test_entity_debug_info_update_entity_id(
"""
await mqtt_mock_entry()
# Add device settings to config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_ID)
config["unique_id"] = "veryunique"
config["platform"] = "mqtt"
@@ -2502,7 +2501,7 @@ async def help_test_entity_disabled_by_default(
"""Test device registry remove."""
await mqtt_mock_entry()
# Add device settings to config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_ID)
config["enabled_by_default"] = False
config["unique_id"] = "veryunique1"
@@ -2514,7 +2513,7 @@ async def help_test_entity_disabled_by_default(
data = json.dumps(config)
async_fire_mqtt_message(hass, f"homeassistant/{domain}/bla1/config", data)
await hass.async_block_till_done()
entity_id = ent_registry.async_get_entity_id(domain, mqtt.DOMAIN, "veryunique1")
entity_id = ent_registry.async_get_entity_id(domain, DOMAIN, "veryunique1")
assert entity_id is not None and hass.states.get(entity_id) is None
assert dev_registry.async_get_device(identifiers={("mqtt", "helloworld")})
@@ -2524,14 +2523,14 @@ async def help_test_entity_disabled_by_default(
data = json.dumps(config)
async_fire_mqtt_message(hass, f"homeassistant/{domain}/bla2/config", data)
await hass.async_block_till_done()
entity_id = ent_registry.async_get_entity_id(domain, mqtt.DOMAIN, "veryunique2")
entity_id = ent_registry.async_get_entity_id(domain, DOMAIN, "veryunique2")
assert entity_id is not None and hass.states.get(entity_id) is not None
# Remove the enabled entity, both entities and the device should be removed
async_fire_mqtt_message(hass, f"homeassistant/{domain}/bla2/config", "")
await hass.async_block_till_done()
assert not ent_registry.async_get_entity_id(domain, mqtt.DOMAIN, "veryunique1")
assert not ent_registry.async_get_entity_id(domain, mqtt.DOMAIN, "veryunique2")
assert not ent_registry.async_get_entity_id(domain, DOMAIN, "veryunique1")
assert not ent_registry.async_get_entity_id(domain, DOMAIN, "veryunique2")
assert not dev_registry.async_get_device(identifiers={("mqtt", "helloworld")})
@@ -2544,7 +2543,7 @@ async def help_test_entity_category(
"""Test device registry remove."""
await mqtt_mock_entry()
# Add device settings to config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_ID)
ent_registry = er.async_get(hass)
@@ -2555,7 +2554,7 @@ async def help_test_entity_category(
data = json.dumps(config)
async_fire_mqtt_message(hass, f"homeassistant/{domain}/{unique_id}/config", data)
await hass.async_block_till_done()
entity_id = ent_registry.async_get_entity_id(domain, mqtt.DOMAIN, unique_id)
entity_id = ent_registry.async_get_entity_id(domain, DOMAIN, unique_id)
assert entity_id is not None and hass.states.get(entity_id)
entry = ent_registry.async_get(entity_id)
assert entry is not None and entry.entity_category is None
@@ -2567,7 +2566,7 @@ async def help_test_entity_category(
data = json.dumps(config)
async_fire_mqtt_message(hass, f"homeassistant/{domain}/{unique_id}/config", data)
await hass.async_block_till_done()
entity_id = ent_registry.async_get_entity_id(domain, mqtt.DOMAIN, unique_id)
entity_id = ent_registry.async_get_entity_id(domain, DOMAIN, unique_id)
assert entity_id is not None and hass.states.get(entity_id)
entry = ent_registry.async_get(entity_id)
assert entry is not None and entry.entity_category == EntityCategory.DIAGNOSTIC
@@ -2579,7 +2578,7 @@ async def help_test_entity_category(
data = json.dumps(config)
async_fire_mqtt_message(hass, f"homeassistant/{domain}/{unique_id}/config", data)
await hass.async_block_till_done()
assert not ent_registry.async_get_entity_id(domain, mqtt.DOMAIN, unique_id)
assert not ent_registry.async_get_entity_id(domain, DOMAIN, unique_id)
async def help_test_entity_icon_and_entity_picture(
@@ -2591,7 +2590,7 @@ async def help_test_entity_icon_and_entity_picture(
"""Test entity picture and icon."""
await mqtt_mock_entry()
# Add device settings to config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
config["device"] = copy.deepcopy(DEFAULT_CONFIG_DEVICE_INFO_ID)
ent_registry = er.async_get(hass)
@@ -2602,7 +2601,7 @@ async def help_test_entity_icon_and_entity_picture(
data = json.dumps(config)
async_fire_mqtt_message(hass, f"homeassistant/{domain}/{unique_id}/config", data)
await hass.async_block_till_done()
entity_id = ent_registry.async_get_entity_id(domain, mqtt.DOMAIN, unique_id)
entity_id = ent_registry.async_get_entity_id(domain, DOMAIN, unique_id)
state = hass.states.get(entity_id)
assert entity_id is not None and state
assert state.attributes.get("icon") is None
@@ -2615,7 +2614,7 @@ async def help_test_entity_icon_and_entity_picture(
data = json.dumps(config)
async_fire_mqtt_message(hass, f"homeassistant/{domain}/{unique_id}/config", data)
await hass.async_block_till_done()
entity_id = ent_registry.async_get_entity_id(domain, mqtt.DOMAIN, unique_id)
entity_id = ent_registry.async_get_entity_id(domain, DOMAIN, unique_id)
state = hass.states.get(entity_id)
assert entity_id is not None and state
assert state.attributes.get("icon") is None
@@ -2629,7 +2628,7 @@ async def help_test_entity_icon_and_entity_picture(
data = json.dumps(config)
async_fire_mqtt_message(hass, f"homeassistant/{domain}/{unique_id}/config", data)
await hass.async_block_till_done()
entity_id = ent_registry.async_get_entity_id(domain, mqtt.DOMAIN, unique_id)
entity_id = ent_registry.async_get_entity_id(domain, DOMAIN, unique_id)
state = hass.states.get(entity_id)
assert entity_id is not None and state
assert state.attributes.get("icon") == "mdi:emoji-happy-outline"
@@ -2662,7 +2661,7 @@ async def help_test_publishing_with_custom_encoding(
setup_config = []
service_data = {}
for test_id, test_data in test_config.items():
test_config_setup: dict[str, Any] = copy.copy(config[mqtt.DOMAIN][domain])
test_config_setup: dict[str, Any] = copy.copy(config[DOMAIN][domain])
test_config_setup.update(
{
topic: f"cmd/{test_id}",
@@ -2802,7 +2801,7 @@ async def help_test_reloadable(
) -> None:
"""Test reloading an MQTT platform."""
# Set up with empty config
config = copy.deepcopy(config[mqtt.DOMAIN][domain])
config = copy.deepcopy(config[DOMAIN][domain])
# Create and test an old config of 2 entities based on the config supplied
old_config_1 = copy.deepcopy(config)
old_config_1["name"] = "test_old_1"
@@ -2810,11 +2809,11 @@ async def help_test_reloadable(
old_config_2["name"] = "test_old_2"
old_config = {
mqtt.DOMAIN: {domain: [old_config_1, old_config_2]},
DOMAIN: {domain: [old_config_1, old_config_2]},
}
# Start the MQTT entry with the old config
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={mqtt.CONF_BROKER: "test-broker"},
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
@@ -2838,7 +2837,7 @@ async def help_test_reloadable(
new_config_extra["name"] = "test_new_3"
new_config = {
mqtt.DOMAIN: {domain: [new_config_1, new_config_2, new_config_extra]},
DOMAIN: {domain: [new_config_1, new_config_2, new_config_extra]},
}
with patch("homeassistant.config.load_yaml_config_file", return_value=new_config):
# Reload the mqtt entry with the new config
@@ -2859,7 +2858,7 @@ async def help_test_reloadable(
async def help_test_unload_config_entry(hass: HomeAssistant) -> None:
"""Test unloading the MQTT config entry."""
mqtt_config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
mqtt_config_entry = hass.config_entries.async_entries(DOMAIN)[0]
assert mqtt_config_entry.state is ConfigEntryState.LOADED
assert await hass.config_entries.async_unload(mqtt_config_entry.entry_id)
@@ -2878,14 +2877,14 @@ async def help_test_unload_config_entry_with_platform(
"""Test unloading the MQTT config entry with a specific platform domain."""
# prepare setup through configuration.yaml
config_setup: dict[str, dict[str, Any]] = copy.deepcopy(config)
config_setup[mqtt.DOMAIN][domain]["name"] = "config_setup"
config_setup[DOMAIN][domain]["name"] = "config_setup"
config_name = config_setup
with patch("homeassistant.config.load_yaml_config_file", return_value=config_name):
await mqtt_mock_entry()
# prepare setup through discovery
discovery_setup = copy.deepcopy(config[mqtt.DOMAIN][domain])
discovery_setup = copy.deepcopy(config[DOMAIN][domain])
discovery_setup["name"] = "discovery_setup"
async_fire_mqtt_message(
hass, f"homeassistant/{domain}/bla/config", json.dumps(discovery_setup)
tests/components/mqtt/conftest.py
+3 -2
View File
@@ -10,6 +10,7 @@ from unittest.mock import AsyncMock, MagicMock, patch
import pytest
from homeassistant.components import mqtt
from homeassistant.components.mqtt import DOMAIN
from homeassistant.components.mqtt.models import MessageCallbackType, ReceiveMessage
from homeassistant.components.mqtt.util import EnsureJobAfterCooldown
from homeassistant.const import EVENT_HOMEASSISTANT_STARTED
@@ -115,14 +116,14 @@ async def setup_with_birth_msg_client_mock(
patch("homeassistant.components.mqtt.client.SUBSCRIBE_COOLDOWN", 0.0),
):
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data=mqtt_config_entry_data or {mqtt.CONF_BROKER: "test-broker"},
options=mqtt_config_entry_options or {},
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
entry.add_to_hass(hass)
hass.config.components.add(mqtt.DOMAIN)
hass.config.components.add(DOMAIN)
assert await hass.config_entries.async_setup(entry.entry_id)
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
tests/components/mqtt/test_alarm_control_panel.py
+19 -18
View File
@@ -8,7 +8,7 @@ from unittest.mock import patch
import pytest
from homeassistant.components import alarm_control_panel, mqtt
from homeassistant.components import alarm_control_panel
from homeassistant.components.alarm_control_panel import (
AlarmControlPanelEntityFeature,
AlarmControlPanelState,
@@ -16,6 +16,7 @@ from homeassistant.components.alarm_control_panel import (
from homeassistant.components.mqtt.alarm_control_panel import (
MQTT_ALARM_ATTRIBUTES_BLOCKED,
)
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.models import PublishPayloadType
from homeassistant.const import (
ATTR_CODE,
@@ -83,7 +84,7 @@ DEFAULT_FEATURES = (
)
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
alarm_control_panel.DOMAIN: {
"name": "test",
"state_topic": "alarm/state",
@@ -93,7 +94,7 @@ DEFAULT_CONFIG = {
}
DEFAULT_CONFIG_CODE_NOT_REQUIRED = {
mqtt.DOMAIN: {
DOMAIN: {
alarm_control_panel.DOMAIN: {
"name": "test",
"state_topic": "alarm/state",
@@ -104,7 +105,7 @@ DEFAULT_CONFIG_CODE_NOT_REQUIRED = {
}
DEFAULT_CONFIG_CODE = {
mqtt.DOMAIN: {
DOMAIN: {
alarm_control_panel.DOMAIN: {
"name": "test",
"state_topic": "alarm/state",
@@ -116,7 +117,7 @@ DEFAULT_CONFIG_CODE = {
}
DEFAULT_CONFIG_REMOTE_CODE = {
mqtt.DOMAIN: {
DOMAIN: {
alarm_control_panel.DOMAIN: {
"name": "test",
"state_topic": "alarm/state",
@@ -128,7 +129,7 @@ DEFAULT_CONFIG_REMOTE_CODE = {
}
DEFAULT_CONFIG_REMOTE_CODE_TEXT = {
mqtt.DOMAIN: {
DOMAIN: {
alarm_control_panel.DOMAIN: {
"name": "test",
"state_topic": "alarm/state",
@@ -151,7 +152,7 @@ def does_not_raise():
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
alarm_control_panel.DOMAIN: {
"name": "test",
"command_topic": "alarm/command",
@@ -162,7 +163,7 @@ def does_not_raise():
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
alarm_control_panel.DOMAIN: {
"name": "test",
"state_topic": "alarm/state",
@@ -173,7 +174,7 @@ def does_not_raise():
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
alarm_control_panel.DOMAIN: {
"name": "test",
"command_topic": "alarm/command",
@@ -753,7 +754,7 @@ async def test_disarm_publishes_mqtt_with_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
alarm_control_panel.DOMAIN: {
"name": "test",
"command_topic": "test-topic",
@@ -953,7 +954,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
alarm_control_panel.DOMAIN: [
{
"name": "Test 1",
@@ -983,7 +984,7 @@ async def test_discovery_removal_alarm(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test removal of discovered alarm_control_panel."""
data = json.dumps(DEFAULT_CONFIG[mqtt.DOMAIN][alarm_control_panel.DOMAIN])
data = json.dumps(DEFAULT_CONFIG[DOMAIN][alarm_control_panel.DOMAIN])
await help_test_discovery_removal(
hass, mqtt_mock_entry, alarm_control_panel.DOMAIN, data
)
@@ -993,8 +994,8 @@ async def test_discovery_update_alarm_topic_and_template(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered alarm_control_panel."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][alarm_control_panel.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][alarm_control_panel.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][alarm_control_panel.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][alarm_control_panel.DOMAIN])
config1["name"] = "Beer"
config2["name"] = "Milk"
config1["state_topic"] = "alarm/state1"
@@ -1027,8 +1028,8 @@ async def test_discovery_update_alarm_template(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered alarm_control_panel."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][alarm_control_panel.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][alarm_control_panel.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][alarm_control_panel.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][alarm_control_panel.DOMAIN])
config1["name"] = "Beer"
config2["name"] = "Milk"
config1["state_topic"] = "alarm/state1"
@@ -1059,7 +1060,7 @@ async def test_discovery_update_unchanged_alarm(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered alarm_control_panel."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][alarm_control_panel.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][alarm_control_panel.DOMAIN])
config1["name"] = "Beer"
data1 = json.dumps(config1)
@@ -1109,7 +1110,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
alarm_control_panel.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][alarm_control_panel.DOMAIN],
DEFAULT_CONFIG[DOMAIN][alarm_control_panel.DOMAIN],
topic,
value,
)
tests/components/mqtt/test_binary_sensor.py
+27 -28
View File
@@ -11,7 +11,8 @@ from freezegun import freeze_time
from freezegun.api import FrozenDateTimeFactory
import pytest
from homeassistant.components import binary_sensor, mqtt
from homeassistant.components import binary_sensor
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.const import (
EVENT_STATE_CHANGED,
STATE_OFF,
@@ -63,7 +64,7 @@ from tests.common import (
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -76,7 +77,7 @@ DEFAULT_CONFIG = {
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -111,7 +112,7 @@ async def test_setting_sensor_value_expires_availability_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -288,7 +289,7 @@ async def test_expiration_on_discovery_and_discovery_update_of_binary_sensor(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -326,7 +327,7 @@ async def test_setting_sensor_value_via_mqtt_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -370,7 +371,7 @@ async def test_invalid_sensor_value_via_mqtt_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -405,7 +406,7 @@ async def test_setting_sensor_value_via_mqtt_message_and_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -446,7 +447,7 @@ async def test_setting_sensor_value_via_mqtt_message_and_template2(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"encoding": "",
@@ -485,7 +486,7 @@ async def test_setting_sensor_value_via_mqtt_msg_and_template_and_raw_state_enco
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -520,7 +521,7 @@ async def test_setting_sensor_value_via_mqtt_message_empty_template(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"device_class": "motion",
@@ -532,7 +533,7 @@ async def test_setting_sensor_value_via_mqtt_message_empty_template(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"device_class": None,
@@ -560,7 +561,7 @@ async def test_valid_device_class(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"device_class": "abc123",
@@ -622,7 +623,7 @@ async def test_custom_availability_payload(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -661,7 +662,7 @@ async def test_force_update_disabled(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -701,7 +702,7 @@ async def test_force_update_enabled(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -801,7 +802,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
binary_sensor.DOMAIN: [
{
"name": "Test 1",
@@ -829,7 +830,7 @@ async def test_discovery_removal_binary_sensor(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test removal of discovered binary_sensor."""
data = json.dumps(DEFAULT_CONFIG[mqtt.DOMAIN][binary_sensor.DOMAIN])
data = json.dumps(DEFAULT_CONFIG[DOMAIN][binary_sensor.DOMAIN])
await help_test_discovery_removal(hass, mqtt_mock_entry, binary_sensor.DOMAIN, data)
@@ -837,8 +838,8 @@ async def test_discovery_update_binary_sensor_topic_template(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered binary_sensor."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][binary_sensor.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][binary_sensor.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][binary_sensor.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][binary_sensor.DOMAIN])
config1["name"] = "Beer"
config2["name"] = "Milk"
config1["state_topic"] = "sensor/state1"
@@ -873,8 +874,8 @@ async def test_discovery_update_binary_sensor_template(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered binary_sensor."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][binary_sensor.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][binary_sensor.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][binary_sensor.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][binary_sensor.DOMAIN])
config1["name"] = "Beer"
config2["name"] = "Milk"
config1["state_topic"] = "sensor/state1"
@@ -929,7 +930,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
binary_sensor.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][binary_sensor.DOMAIN],
DEFAULT_CONFIG[DOMAIN][binary_sensor.DOMAIN],
topic,
value,
attribute,
@@ -941,7 +942,7 @@ async def test_discovery_update_unchanged_binary_sensor(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered binary_sensor."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][binary_sensor.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][binary_sensor.DOMAIN])
config1["name"] = "Beer"
data1 = json.dumps(config1)
@@ -1097,9 +1098,7 @@ async def test_cleanup_triggers_and_restoring_state(
freezer.move_to("2022-02-02 12:01:10+01:00")
await help_test_reload_with_config(
hass, caplog, tmp_path, {mqtt.DOMAIN: hass_config}
)
await help_test_reload_with_config(hass, caplog, tmp_path, {DOMAIN: hass_config})
state = hass.states.get("binary_sensor.test1")
assert state.state == state1
@@ -1137,7 +1136,7 @@ async def test_skip_restoring_state_with_over_due_expire_trigger(
freezer.move_to("2022-02-02 12:02:00+01:00")
domain = binary_sensor.DOMAIN
config3: ConfigType = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][domain])
config3: ConfigType = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][domain])
config3["name"] = "test3"
config3["expire_after"] = 10
config3["state_topic"] = "test-topic3"
tests/components/mqtt/test_button.py
+14 -13
View File
@@ -6,7 +6,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import button, mqtt
from homeassistant.components import button
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.const import ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
@@ -42,7 +43,7 @@ from .common import (
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {button.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
DOMAIN: {button.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
}
@@ -51,7 +52,7 @@ DEFAULT_CONFIG = {
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
button.DOMAIN: {
"command_topic": "command-topic",
"name": "test",
@@ -93,7 +94,7 @@ async def test_sending_mqtt_commands(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
button.DOMAIN: {
"command_topic": "command-topic",
"name": "test",
@@ -110,7 +111,7 @@ async def test_sending_mqtt_commands(
}
},
{
mqtt.DOMAIN: {
DOMAIN: {
button.DOMAIN: {
"command_topic": "command-topic",
"name": "test",
@@ -152,7 +153,7 @@ async def test_sending_mqtt_commands_with_message_expiry_interval(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
button.DOMAIN: {
"command_topic": "command-topic",
"command_template": '{ "{{ value }}": "{{ entity_id }}" }',
@@ -215,7 +216,7 @@ async def test_default_availability_payload(
) -> None:
"""Test availability by default payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
button.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -233,7 +234,7 @@ async def test_custom_availability_payload(
) -> None:
"""Test availability by custom payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
button.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -309,7 +310,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
button.DOMAIN: [
{
"name": "Test 1",
@@ -345,8 +346,8 @@ async def test_discovery_update_button(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered button."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][button.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][button.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][button.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][button.DOMAIN])
config1["name"] = "Beer"
config2["name"] = "Milk"
@@ -446,7 +447,7 @@ async def test_entity_debug_info_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
button.DOMAIN: {
"name": "test",
"command_topic": "test-topic",
@@ -470,7 +471,7 @@ async def test_invalid_device_class(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
button.DOMAIN: [
{
"name": "Test 1",
tests/components/mqtt/test_camera.py
+8 -7
View File
@@ -7,8 +7,9 @@ from unittest.mock import patch
import pytest
from homeassistant.components import camera, mqtt
from homeassistant.components import camera
from homeassistant.components.mqtt.camera import MQTT_CAMERA_ATTRIBUTES_BLOCKED
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.core import HomeAssistant
from .common import (
@@ -45,12 +46,12 @@ from tests.typing import (
MqttMockPahoClient,
)
DEFAULT_CONFIG = {mqtt.DOMAIN: {camera.DOMAIN: {"name": "test", "topic": "test_topic"}}}
DEFAULT_CONFIG = {DOMAIN: {camera.DOMAIN: {"name": "test", "topic": "test_topic"}}}
@pytest.mark.parametrize(
"hass_config",
[{mqtt.DOMAIN: {camera.DOMAIN: {"topic": "test/camera", "name": "Test Camera"}}}],
[{DOMAIN: {camera.DOMAIN: {"topic": "test/camera", "name": "Test Camera"}}}],
)
async def test_run_camera_setup(
hass: HomeAssistant,
@@ -76,7 +77,7 @@ async def test_run_camera_setup(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
camera.DOMAIN: {
"topic": "test/camera",
"name": "Test Camera",
@@ -110,7 +111,7 @@ async def test_run_camera_b64_encoded(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
"camera": {
"topic": "test/camera",
"name": "Test Camera",
@@ -250,7 +251,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
camera.DOMAIN: [
{
"name": "Test 1",
@@ -278,7 +279,7 @@ async def test_discovery_removal_camera(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test removal of discovered camera."""
data = json.dumps(DEFAULT_CONFIG[mqtt.DOMAIN][camera.DOMAIN])
data = json.dumps(DEFAULT_CONFIG[DOMAIN][camera.DOMAIN])
await help_test_discovery_removal(hass, mqtt_mock_entry, camera.DOMAIN, data)
tests/components/mqtt/test_client.py
+21 -21
View File
@@ -15,7 +15,7 @@ import pytest
from homeassistant.components import mqtt
from homeassistant.components.mqtt.client import RECONNECT_INTERVAL_SECONDS
from homeassistant.components.mqtt.const import SUPPORTED_COMPONENTS
from homeassistant.components.mqtt.const import DOMAIN, SUPPORTED_COMPONENTS
from homeassistant.components.mqtt.models import MessageCallbackType, ReceiveMessage
from homeassistant.config_entries import ConfigEntryDisabler, ConfigEntryState
from homeassistant.const import (
@@ -99,7 +99,7 @@ async def test_mqtt_await_ack_at_disconnect(hass: HomeAssistant) -> None:
)
mqtt_client.publish = MagicMock(return_value=FakeInfo())
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={
"certificate": "auto",
mqtt.CONF_BROKER: "test-broker",
@@ -276,7 +276,7 @@ async def test_message_expiry_interval_fails_for_legacy_protocols(
await mqtt.async_publish(
hass, "test-topic", "test-payload", 2, True, message_expiry_interval=60
)
assert exc.value.translation_domain == mqtt.DOMAIN
assert exc.value.translation_domain == DOMAIN
assert exc.value.translation_key == "mqtt_message_expiry_interval_not_supported"
assert exc.value.translation_placeholders == {
"topic": "test-topic",
@@ -399,7 +399,7 @@ async def test_subscribe_mqtt_config_entry_disabled(
"""Test the subscription of a topic when MQTT config entry is disabled."""
mqtt_mock.connected = True
mqtt_config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
mqtt_config_entry = hass.config_entries.async_entries(DOMAIN)[0]
mqtt_config_entry_state = mqtt_config_entry.state
assert mqtt_config_entry_state is ConfigEntryState.LOADED
@@ -1340,7 +1340,7 @@ async def test_initial_setup_logs_error(
) -> None:
"""Test for setup failure if initial client connection fails."""
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={mqtt.CONF_BROKER: "test-broker"},
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
@@ -1487,7 +1487,7 @@ async def test_publish_error(
) -> None:
"""Test publish error."""
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={mqtt.CONF_BROKER: "test-broker"},
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
@@ -1681,7 +1681,7 @@ async def test_handle_mqtt_timeout_on_callback(
)
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={mqtt.CONF_BROKER: "test-broker"},
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
@@ -1718,7 +1718,7 @@ async def test_setup_raises_config_entry_not_ready_if_no_connect_broker(
) -> None:
"""Test for setup failure if connection to broker is missing."""
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={mqtt.CONF_BROKER: "test-broker"},
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
@@ -1863,14 +1863,14 @@ async def test_custom_birth_message(
"""Test sending birth message."""
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data=mqtt_config_entry_data,
options=mqtt_config_entry_options,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
entry.add_to_hass(hass)
hass.config.components.add(mqtt.DOMAIN)
hass.config.components.add(DOMAIN)
assert await hass.config_entries.async_setup(entry.entry_id)
mock_debouncer.clear()
hass.bus.async_fire(EVENT_HOMEASSISTANT_STARTED)
@@ -1913,14 +1913,14 @@ async def test_no_birth_message(
) -> None:
"""Test disabling birth message."""
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data=mqtt_config_entry_data,
options=mqtt_config_entry_options,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
entry.add_to_hass(hass)
hass.config.components.add(mqtt.DOMAIN)
hass.config.components.add(DOMAIN)
mock_debouncer.clear()
assert await hass.config_entries.async_setup(entry.entry_id)
# Wait for discovery cooldown
@@ -1953,14 +1953,14 @@ async def test_delayed_birth_message(
await hass.async_block_till_done()
birth = asyncio.Event()
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data=mqtt_config_entry_data,
options=mqtt_config_entry_options,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
entry.add_to_hass(hass)
hass.config.components.add(mqtt.DOMAIN)
hass.config.components.add(DOMAIN)
assert await hass.config_entries.async_setup(entry.entry_id)
@callback
@@ -2033,14 +2033,14 @@ async def test_custom_will_message(
) -> None:
"""Test will message."""
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data=mqtt_config_entry_data,
options=mqtt_config_entry_options,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
entry.add_to_hass(hass)
hass.config.components.add(mqtt.DOMAIN)
hass.config.components.add(DOMAIN)
assert await hass.config_entries.async_setup(entry.entry_id)
await hass.async_block_till_done()
@@ -2071,14 +2071,14 @@ async def test_no_will_message(
) -> None:
"""Test will message."""
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data=mqtt_config_entry_data,
options=mqtt_config_entry_options,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
entry.add_to_hass(hass)
hass.config.components.add(mqtt.DOMAIN)
hass.config.components.add(DOMAIN)
assert await hass.config_entries.async_setup(entry.entry_id)
await hass.async_block_till_done()
@@ -2619,7 +2619,7 @@ async def test_subscriptions_id_generation(hass: HomeAssistant) -> None:
assert new_id_3 == mqtt.models.MAX_28BIT
with pytest.raises(HomeAssistantError) as exc:
generator.get_or_generate("test4/#")
assert exc.value.translation_domain == mqtt.DOMAIN
assert exc.value.translation_domain == DOMAIN
assert exc.value.translation_key == "mqtt_max_subscription_id_reached"
generator.release("test2/#")
@@ -2629,7 +2629,7 @@ async def test_subscriptions_id_generation(hass: HomeAssistant) -> None:
with pytest.raises(HomeAssistantError) as exc:
generator.get_or_generate("test5/#")
assert exc.value.translation_domain == mqtt.DOMAIN
assert exc.value.translation_domain == DOMAIN
assert exc.value.translation_key == "mqtt_max_subscription_id_reached"
generator.release("test1/#")
@@ -2642,5 +2642,5 @@ async def test_subscriptions_id_generation(hass: HomeAssistant) -> None:
with pytest.raises(HomeAssistantError) as exc:
generator.get_or_generate("test7/#")
assert exc.value.translation_domain == mqtt.DOMAIN
assert exc.value.translation_domain == DOMAIN
assert exc.value.translation_key == "mqtt_max_subscription_id_reached"
tests/components/mqtt/test_climate.py
+17 -16
View File
@@ -8,7 +8,7 @@ from unittest.mock import call, patch
import pytest
import voluptuous as vol
from homeassistant.components import climate, mqtt
from homeassistant.components import climate
from homeassistant.components.climate import (
ATTR_CURRENT_HUMIDITY,
ATTR_CURRENT_TEMPERATURE,
@@ -34,6 +34,7 @@ from homeassistant.components.mqtt.climate import (
)
from homeassistant.components.mqtt.const import (
DEFAULT_CLIMATE_INITIAL_TEMPERATURE as DEFAULT_INITIAL_TEMPERATURE,
DOMAIN,
)
from homeassistant.const import ATTR_TEMPERATURE, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
@@ -83,7 +84,7 @@ from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
ENTITY_CLIMATE = "climate.test"
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
climate.DOMAIN: {
"name": "test",
"mode_command_topic": "mode-topic",
@@ -1421,7 +1422,7 @@ async def test_get_target_temperature_low_high_with_templates(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
climate.DOMAIN: {
"name": "test",
"mode_command_topic": "mode-topic",
@@ -1588,7 +1589,7 @@ async def test_get_with_templates(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
climate.DOMAIN: {
"name": "test",
"mode_command_topic": "mode-topic",
@@ -2021,7 +2022,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
climate.DOMAIN: [
{
"name": "Test 1",
@@ -2074,7 +2075,7 @@ async def test_encoding_subscribable_topics(
attribute_value: Any,
) -> None:
"""Test handling of incoming encoded payload."""
config = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][climate.DOMAIN])
config = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][climate.DOMAIN])
await help_test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
@@ -2091,7 +2092,7 @@ async def test_discovery_removal_climate(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test removal of discovered climate."""
data = json.dumps(DEFAULT_CONFIG[mqtt.DOMAIN][climate.DOMAIN])
data = json.dumps(DEFAULT_CONFIG[DOMAIN][climate.DOMAIN])
await help_test_discovery_removal(hass, mqtt_mock_entry, climate.DOMAIN, data)
@@ -2172,7 +2173,7 @@ async def test_entity_id_update_subscriptions(
) -> None:
"""Test MQTT subscriptions are managed when entity_id is updated."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
climate.DOMAIN: {
"name": "test",
"mode_state_topic": "test-topic",
@@ -2199,7 +2200,7 @@ async def test_entity_debug_info_message(
) -> None:
"""Test MQTT debug info."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
climate.DOMAIN: {
"name": "test",
"mode_command_topic": "command-topic",
@@ -2362,8 +2363,8 @@ async def test_publishing_with_custom_encoding(
domain = climate.DOMAIN
config = copy.deepcopy(DEFAULT_CONFIG)
if topic != "preset_mode_command_topic":
del config[mqtt.DOMAIN][domain]["preset_mode_command_topic"]
del config[mqtt.DOMAIN][domain]["preset_modes"]
del config[DOMAIN][domain]["preset_mode_command_topic"]
del config[DOMAIN][domain]["preset_modes"]
await help_test_publishing_with_custom_encoding(
hass,
@@ -2384,7 +2385,7 @@ async def test_publishing_with_custom_encoding(
[
( # test_valid_humidity_min_max
{
mqtt.DOMAIN: {
DOMAIN: {
climate.DOMAIN: {
"name": "test",
"min_humidity": 20,
@@ -2396,7 +2397,7 @@ async def test_publishing_with_custom_encoding(
),
( # test_invalid_humidity_min_max_1
{
mqtt.DOMAIN: {
DOMAIN: {
climate.DOMAIN: {
"name": "test",
"min_humidity": 0,
@@ -2408,7 +2409,7 @@ async def test_publishing_with_custom_encoding(
),
( # test_invalid_humidity_min_max_2
{
mqtt.DOMAIN: {
DOMAIN: {
climate.DOMAIN: {
"name": "test",
"max_humidity": 20,
@@ -2420,7 +2421,7 @@ async def test_publishing_with_custom_encoding(
),
( # test_valid_humidity_state
{
mqtt.DOMAIN: {
DOMAIN: {
climate.DOMAIN: {
"name": "test",
"target_humidity_state_topic": "humidity-state",
@@ -2432,7 +2433,7 @@ async def test_publishing_with_custom_encoding(
),
( # test_invalid_humidity_state
{
mqtt.DOMAIN: {
DOMAIN: {
climate.DOMAIN: {
"name": "test",
"target_humidity_state_topic": "humidity-state",
tests/components/mqtt/test_config_flow.py
+48 -49
View File
@@ -17,12 +17,11 @@ import voluptuous as vol
from homeassistant import config_entries
from homeassistant.components import mqtt
from homeassistant.components.hassio import AddonError
from homeassistant.components.mqtt import DOMAIN
from homeassistant.components.mqtt.config_flow import (
PWD_NOT_CHANGED,
TRANSLATION_DESCRIPTION_PLACEHOLDERS,
)
from homeassistant.components.mqtt.const import CONF_DISCOVERY_QOS
from homeassistant.components.mqtt.const import CONF_DISCOVERY_QOS, DOMAIN
from homeassistant.components.mqtt.util import learn_more_url
from homeassistant.config_entries import ConfigSubentry, ConfigSubentryData
from homeassistant.const import (
@@ -542,7 +541,7 @@ async def test_manual_config_set(
)
# Check config entry got setup
assert len(mock_finish_setup.mock_calls) == 1
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry = hass.config_entries.async_entries(DOMAIN)[0]
assert config_entry.title == "127.0.0.1"
@@ -564,7 +563,7 @@ async def test_user_single_instance(hass: HomeAssistant) -> None:
async def test_hassio_already_configured(hass: HomeAssistant) -> None:
"""Test we only allow a single config flow."""
MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
).add_to_hass(hass)
@@ -579,14 +578,14 @@ async def test_hassio_already_configured(hass: HomeAssistant) -> None:
async def test_hassio_ignored(hass: HomeAssistant) -> None:
"""Test we supervisor discovered instance can be ignored."""
MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
source=config_entries.SOURCE_IGNORE,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
).add_to_hass(hass)
result = await hass.config_entries.flow.async_init(
mqtt.DOMAIN,
DOMAIN,
data=HassioServiceInfo(
config={
"addon": "Mosquitto",
@@ -1102,7 +1101,7 @@ async def test_option_flow(
"homeassistant.config.async_hass_config_yaml", AsyncMock(return_value={})
) as yaml_mock:
await mqtt_mock_entry()
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry = hass.config_entries.async_entries(DOMAIN)[0]
result = await hass.config_entries.options.async_init(config_entry.entry_id)
assert result["type"] is FlowResultType.FORM
@@ -1259,7 +1258,7 @@ async def test_bad_certificate(
test_input.pop(mqtt.CONF_CLIENT_KEY)
mqtt_mock = await mqtt_mock_entry()
config_entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
# Add at least one advanced option to get the full form
hass.config_entries.async_update_entry(
config_entry,
@@ -1336,7 +1335,7 @@ async def test_keepalive_validation(
mqtt_mock = await mqtt_mock_entry()
mock_try_connection.return_value = True
config_entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
# Add at least one advanced option to get the full form
hass.config_entries.async_update_entry(
config_entry,
@@ -1377,7 +1376,7 @@ async def test_disable_birth_will(
"""Test disabling birth and will."""
await mqtt_mock_entry()
mock_try_connection.return_value = True
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry = hass.config_entries.async_entries(DOMAIN)[0]
hass.config_entries.async_update_entry(
config_entry,
data={
@@ -1442,7 +1441,7 @@ async def test_invalid_discovery_prefix(
"""Test setting an invalid discovery prefix."""
mqtt_mock = await mqtt_mock_entry()
mock_try_connection.return_value = True
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry = hass.config_entries.async_entries(DOMAIN)[0]
hass.config_entries.async_update_entry(
config_entry,
data={
@@ -1507,7 +1506,7 @@ async def test_option_flow_default_suggested_values(
) -> None:
"""Test config flow options has default/suggested values."""
await mqtt_mock_entry()
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry = hass.config_entries.async_entries(DOMAIN)[0]
hass.config_entries.async_update_entry(
config_entry,
data={
@@ -1654,7 +1653,7 @@ async def test_step_reauth(
# Prepare the config entry
config_entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data=test_input,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
@@ -1735,7 +1734,7 @@ async def test_step_hassio_reauth(
# Prepare the config entry
config_entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data=entry_data,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
@@ -1823,7 +1822,7 @@ async def test_step_hassio_reauth_no_discovery_info(
# Prepare the config entry
config_entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data=entry_data,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
@@ -1854,7 +1853,7 @@ async def test_reconfigure_user_connection_fails(
) -> None:
"""Test if connection cannot be made."""
config_entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
@@ -1892,7 +1891,7 @@ async def test_options_bad_birth_message_fails(
) -> None:
"""Test bad birth message."""
config_entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
@@ -1930,7 +1929,7 @@ async def test_options_bad_will_message_fails(
) -> None:
"""Test bad will message."""
config_entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
@@ -1975,7 +1974,7 @@ async def test_try_connection_with_advanced_parameters(
) -> None:
"""Test config flow with advanced parameters from config."""
config_entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
@@ -2113,7 +2112,7 @@ async def test_setup_with_advanced_settings(
file_id = mock_process_uploaded_file.file_id
config_entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
@@ -2299,7 +2298,7 @@ async def test_setup_with_certificates(
file_id = mock_process_uploaded_file.file_id
config_entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
@@ -2420,7 +2419,7 @@ async def test_change_websockets_transport_to_tcp(
) -> None:
"""Test reconfiguration flow changing websockets transport settings."""
config_entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
@@ -2489,7 +2488,7 @@ async def test_reconfigure_flow_form(
) -> None:
"""Test reconfigure flow."""
await mqtt_mock_entry()
entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
result = await entry.start_reconfigure_flow(hass)
assert result["type"] is FlowResultType.FORM
assert result["step_id"] == "broker"
@@ -2543,7 +2542,7 @@ async def test_reconfigure_no_changed_password(
) -> None:
"""Test reconfigure flow."""
await mqtt_mock_entry()
entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
result = await entry.start_reconfigure_flow(hass)
assert result["type"] is FlowResultType.FORM
assert result["step_id"] == "broker"
@@ -2604,7 +2603,7 @@ async def test_migrate_config_entry(
expected_minor_version: int,
) -> None:
"""Test migrating a config entry."""
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry = hass.config_entries.async_entries(DOMAIN)[0]
# Mock to a migratable or compatbible config entry version
hass.config_entries.async_update_entry(
config_entry,
@@ -2646,7 +2645,7 @@ async def test_migrate_of_incompatible_config_entry(
options: dict[str, Any],
) -> None:
"""Test migrating a config entry."""
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry = hass.config_entries.async_entries(DOMAIN)[0]
# Mock an incompatible config entry version
hass.config_entries.async_update_entry(
config_entry,
@@ -4110,7 +4109,7 @@ async def test_subentry_configflow(
component = next(iter(config_subentries_data["components"].values()))
await mqtt_mock_entry()
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry = hass.config_entries.async_entries(DOMAIN)[0]
result = await hass.config_entries.subentries.async_init(
(config_entry.entry_id, "device"),
@@ -4267,7 +4266,7 @@ async def test_subentry_reconfigure_remove_entity(
) -> None:
"""Test the subentry ConfigFlow reconfigure removing an entity."""
await mqtt_mock_entry()
config_entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
subentry_id: str
subentry: ConfigSubentry
subentry_id, subentry = next(iter(config_entry.subentries.items()))
@@ -4276,7 +4275,7 @@ async def test_subentry_reconfigure_remove_entity(
assert result["step_id"] == "summary_menu"
# assert we have a device for the subentry
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
device = device_registry.async_get_device(identifiers={(DOMAIN, subentry_id)})
assert device is not None
# assert we have an entity for all subentry components
@@ -4295,7 +4294,7 @@ async def test_subentry_reconfigure_remove_entity(
unique_entity_id = f"{subentry_id}_{key}"
entity_id = entity_registry.async_get_entity_id(
domain=component["platform"],
platform=mqtt.DOMAIN,
platform=DOMAIN,
unique_id=unique_entity_id,
)
assert entity_id is not None
@@ -4356,7 +4355,7 @@ async def test_subentry_reconfigure_remove_entity(
unique_entity_id = f"{subentry_id}_{object_list[1]}"
entity_id = entity_registry.async_get_entity_id(
domain=components[object_list[1]]["platform"],
platform=mqtt.DOMAIN,
platform=DOMAIN,
unique_id=unique_entity_id,
)
assert entity_id is None
@@ -4391,7 +4390,7 @@ async def test_subentry_reconfigure_edit_entity_multi_entitites(
) -> None:
"""Test the subentry ConfigFlow reconfigure with multi entities."""
await mqtt_mock_entry()
config_entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
subentry_id: str
subentry: ConfigSubentry
subentry_id, subentry = next(iter(config_entry.subentries.items()))
@@ -4401,7 +4400,7 @@ async def test_subentry_reconfigure_edit_entity_multi_entitites(
assert result["step_id"] == "summary_menu"
# assert we have a device for the subentry
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
device = device_registry.async_get_device(identifiers={(DOMAIN, subentry_id)})
assert device is not None
# assert we have an entity for all subentry components
@@ -4419,7 +4418,7 @@ async def test_subentry_reconfigure_edit_entity_multi_entitites(
for key in components:
unique_entity_id = f"{subentry_id}_{key}"
entity_id = entity_registry.async_get_entity_id(
domain="notify", platform=mqtt.DOMAIN, unique_id=unique_entity_id
domain="notify", platform=DOMAIN, unique_id=unique_entity_id
)
assert entity_id is not None
entity_entry = entity_registry.async_get(entity_id)
@@ -4833,7 +4832,7 @@ async def test_subentry_reconfigure_edit_entity_single_entity(
) -> None:
"""Test the subentry ConfigFlow reconfigure with single entity."""
await mqtt_mock_entry()
config_entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
subentry_id: str
subentry: ConfigSubentry
subentry_id, subentry = next(iter(config_entry.subentries.items()))
@@ -4842,7 +4841,7 @@ async def test_subentry_reconfigure_edit_entity_single_entity(
assert result["step_id"] == "summary_menu"
# assert we have a device for the subentry
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
device = device_registry.async_get_device(identifiers={(DOMAIN, subentry_id)})
assert device is not None
# assert we have an entity for the subentry component
@@ -4854,7 +4853,7 @@ async def test_subentry_reconfigure_edit_entity_single_entity(
unique_entity_id = f"{subentry_id}_{component_id}"
entity_id = entity_registry.async_get_entity_id(
domain=component["platform"], platform=mqtt.DOMAIN, unique_id=unique_entity_id
domain=component["platform"], platform=DOMAIN, unique_id=unique_entity_id
)
assert entity_id is not None
entity_entry = entity_registry.async_get(entity_id)
@@ -4972,7 +4971,7 @@ async def test_subentry_reconfigure_edit_entity_reset_fields(
) -> None:
"""Test the subentry ConfigFlow reconfigure resets filtered out fields."""
await mqtt_mock_entry()
config_entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
subentry_id: str
subentry: ConfigSubentry
subentry_id, subentry = next(iter(config_entry.subentries.items()))
@@ -4981,7 +4980,7 @@ async def test_subentry_reconfigure_edit_entity_reset_fields(
assert result["step_id"] == "summary_menu"
# assert we have a device for the subentry
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
device = device_registry.async_get_device(identifiers={(DOMAIN, subentry_id)})
assert device is not None
# assert we have an entity for the subentry component
@@ -4994,7 +4993,7 @@ async def test_subentry_reconfigure_edit_entity_reset_fields(
unique_entity_id = f"{subentry_id}_{component_id}"
entity_id = entity_registry.async_get_entity_id(
domain=component["platform"], platform=mqtt.DOMAIN, unique_id=unique_entity_id
domain=component["platform"], platform=DOMAIN, unique_id=unique_entity_id
)
assert entity_id is not None
entity_entry = entity_registry.async_get(entity_id)
@@ -5107,7 +5106,7 @@ async def test_subentry_reconfigure_add_entity(
) -> None:
"""Test the subentry ConfigFlow reconfigure and add an entity."""
await mqtt_mock_entry()
config_entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
subentry_id: str
subentry: ConfigSubentry
subentry_id, subentry = next(iter(config_entry.subentries.items()))
@@ -5116,7 +5115,7 @@ async def test_subentry_reconfigure_add_entity(
assert result["step_id"] == "summary_menu"
# assert we have a device for the subentry
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
device = device_registry.async_get_device(identifiers={(DOMAIN, subentry_id)})
assert device is not None
# assert we have an entity for the subentry component
@@ -5125,7 +5124,7 @@ async def test_subentry_reconfigure_add_entity(
component_id_1, component1 = next(iter(components.items()))
unique_entity_id = f"{subentry_id}_{component_id_1}"
entity_id = entity_registry.async_get_entity_id(
domain=component1["platform"], platform=mqtt.DOMAIN, unique_id=unique_entity_id
domain=component1["platform"], platform=DOMAIN, unique_id=unique_entity_id
)
assert entity_id is not None
entity_entry = entity_registry.async_get(entity_id)
@@ -5213,7 +5212,7 @@ async def test_subentry_reconfigure_update_device_properties(
) -> None:
"""Test the subentry ConfigFlow reconfigure and update device properties."""
await mqtt_mock_entry()
config_entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
subentry_id: str
subentry: ConfigSubentry
subentry_id, subentry = next(iter(config_entry.subentries.items()))
@@ -5222,7 +5221,7 @@ async def test_subentry_reconfigure_update_device_properties(
assert result["step_id"] == "summary_menu"
# assert we have a device for the subentry
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
device = device_registry.async_get_device(identifiers={(DOMAIN, subentry_id)})
assert device is not None
# assert we have an entity for all subentry components
@@ -5368,7 +5367,7 @@ async def test_subentry_reconfigure_availablity(
) -> None:
"""Test the subentry ConfigFlow reconfigure and update device properties."""
await mqtt_mock_entry()
config_entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
subentry_id: str
subentry: ConfigSubentry
subentry_id, subentry = next(iter(config_entry.subentries.items()))
@@ -5496,7 +5495,7 @@ async def test_subentry_reconfigure_export_settings(
) -> None:
"""Test the subentry ConfigFlow reconfigure export feature."""
await mqtt_mock_entry()
config_entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
subentry_id: str
subentry: ConfigSubentry
subentry_id, subentry = next(iter(config_entry.subentries.items()))
@@ -5505,7 +5504,7 @@ async def test_subentry_reconfigure_export_settings(
assert result["step_id"] == "summary_menu"
# assert we have a device for the subentry
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
device = device_registry.async_get_device(identifiers={(DOMAIN, subentry_id)})
assert device is not None
# assert we entity for all subentry components
@@ -5562,7 +5561,7 @@ async def test_subentry_configflow_section_feature(
) -> None:
"""Test subentry sections are hidden with no configurable options."""
await mqtt_mock_entry()
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry = hass.config_entries.async_entries(DOMAIN)[0]
result = await hass.config_entries.subentries.async_init(
(config_entry.entry_id, "device"),
tests/components/mqtt/test_cover.py
+80 -82
View File
@@ -6,7 +6,7 @@ from unittest.mock import patch
import pytest
from homeassistant.components import cover, mqtt
from homeassistant.components import cover
from homeassistant.components.cover import (
ATTR_CURRENT_POSITION,
ATTR_CURRENT_TILT_POSITION,
@@ -14,7 +14,7 @@ from homeassistant.components.cover import (
ATTR_TILT_POSITION,
CoverState,
)
from homeassistant.components.mqtt.const import CONF_STATE_TOPIC
from homeassistant.components.mqtt.const import CONF_STATE_TOPIC, DOMAIN
from homeassistant.components.mqtt.cover import (
CONF_GET_POSITION_TEMPLATE,
CONF_GET_POSITION_TOPIC,
@@ -81,16 +81,14 @@ from .common import (
from tests.common import async_fire_mqtt_message
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {cover.DOMAIN: {"name": "test", "state_topic": "test-topic"}}
}
DEFAULT_CONFIG = {DOMAIN: {cover.DOMAIN: {"name": "test", "state_topic": "test-topic"}}}
@pytest.mark.parametrize(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -134,7 +132,7 @@ async def test_state_via_state_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -180,7 +178,7 @@ async def test_opening_and_closing_state_via_custom_state_payload(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "position-topic",
@@ -234,7 +232,7 @@ async def test_open_closed_state_from_position_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "position-topic",
@@ -295,7 +293,7 @@ async def test_open_closed_state_from_position_optimistic_alt_positions(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"tilt_command_topic": "set-position-topic",
@@ -356,7 +354,7 @@ async def test_tilt_open_closed_toggle_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"tilt_command_topic": "set-position-topic",
@@ -421,7 +419,7 @@ async def test_tilt_open_closed_toggle_optimistic_alt_positions(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "get-position-topic",
@@ -462,7 +460,7 @@ async def test_position_via_position_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -503,7 +501,7 @@ async def test_state_via_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -546,7 +544,7 @@ async def test_state_via_template_and_entity_id(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -592,7 +590,7 @@ async def test_state_via_template_with_json_value(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "get-position-topic",
@@ -638,7 +636,7 @@ async def test_position_via_template_and_entity_id(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {"name": "test", "qos": 0, "command_topic": "abc"}
}
},
@@ -646,7 +644,7 @@ async def test_position_via_template_and_entity_id(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"qos": 0,
@@ -660,7 +658,7 @@ async def test_position_via_template_and_entity_id(
# ({"set_position_topic": "abc"}, True), - not a valid configuration
(
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"qos": 0,
@@ -673,7 +671,7 @@ async def test_position_via_template_and_entity_id(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"qos": 0,
@@ -685,7 +683,7 @@ async def test_position_via_template_and_entity_id(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"qos": 0,
@@ -716,7 +714,7 @@ async def test_optimistic_flag(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -784,7 +782,7 @@ async def test_optimistic_state_change(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"optimistic": True,
@@ -859,7 +857,7 @@ async def test_optimistic_state_change_with_position(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -894,7 +892,7 @@ async def test_send_open_cover_command(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -929,7 +927,7 @@ async def test_send_close_cover_command(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -965,7 +963,7 @@ async def test_send_stop_cover_command(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -979,7 +977,7 @@ async def test_send_stop_cover_command(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -1021,7 +1019,7 @@ async def test_send_stop_tilt_command(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "get-position-topic",
@@ -1076,7 +1074,7 @@ async def test_current_cover_position(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "get-position-topic",
@@ -1142,7 +1140,7 @@ async def test_current_cover_position_inverted(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -1167,7 +1165,7 @@ async def test_optimistic_position(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "get-position-topic",
@@ -1209,7 +1207,7 @@ async def test_position_update(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "get-position-topic",
@@ -1229,7 +1227,7 @@ async def test_position_update(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "get-position-topic",
@@ -1274,7 +1272,7 @@ async def test_set_position_templated(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "get-position-topic",
@@ -1322,7 +1320,7 @@ async def test_set_position_templated_and_attributes(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "get-position-topic",
@@ -1363,7 +1361,7 @@ async def test_set_tilt_templated(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "get-position-topic",
@@ -1457,7 +1455,7 @@ async def test_set_tilt_templated_and_attributes(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "state-topic",
@@ -1493,7 +1491,7 @@ async def test_set_position_untemplated(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"position_topic": "state-topic",
@@ -1531,7 +1529,7 @@ async def test_set_position_untemplated_custom_percentage_range(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"qos": 0,
@@ -1558,7 +1556,7 @@ async def test_no_command_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -1584,7 +1582,7 @@ async def test_no_payload_close(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -1610,7 +1608,7 @@ async def test_no_payload_open(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -1636,7 +1634,7 @@ async def test_no_payload_stop(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"command_topic": "test",
"name": "test",
@@ -1664,7 +1662,7 @@ async def test_with_command_topic_and_tilt(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -1695,7 +1693,7 @@ async def test_tilt_defaults(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -1785,7 +1783,7 @@ async def test_tilt_via_invocation_defaults(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -1877,7 +1875,7 @@ async def test_tilt_given_value(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -1957,7 +1955,7 @@ async def test_tilt_given_value_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2039,7 +2037,7 @@ async def test_tilt_given_value_altered_range(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2080,7 +2078,7 @@ async def test_tilt_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2124,7 +2122,7 @@ async def test_tilt_via_topic_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2177,7 +2175,7 @@ async def test_tilt_via_topic_template_json_value(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2227,7 +2225,7 @@ async def test_tilt_via_topic_altered_range(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2264,7 +2262,7 @@ async def test_tilt_status_out_of_range_warning(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2299,7 +2297,7 @@ async def test_tilt_status_not_numeric_warning(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2349,7 +2347,7 @@ async def test_tilt_via_topic_altered_range_inverted(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2402,7 +2400,7 @@ async def test_tilt_via_topic_template_altered_range(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2440,7 +2438,7 @@ async def test_tilt_position(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2479,7 +2477,7 @@ async def test_tilt_position_templated(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2559,7 +2557,7 @@ async def test_custom_availability_payload(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"device_class": "garage",
@@ -2583,7 +2581,7 @@ async def test_valid_device_class(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"device_class": "abc123",
@@ -2669,7 +2667,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: [
{
"name": "Test 1",
@@ -2807,7 +2805,7 @@ async def test_entity_debug_info_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2868,7 +2866,7 @@ async def test_state_and_position_topics_state_not_set_via_position_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2925,7 +2923,7 @@ async def test_set_state_via_position_using_stopped_state(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -2963,7 +2961,7 @@ async def test_position_via_position_topic_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -3010,7 +3008,7 @@ async def test_position_via_position_topic_template_json_value(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -3053,7 +3051,7 @@ async def test_position_template_with_entity_id(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -3084,7 +3082,7 @@ async def test_position_via_position_topic_template_return_json(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -3117,7 +3115,7 @@ async def test_position_via_position_topic_template_return_json_warning(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -3165,7 +3163,7 @@ async def test_position_and_tilt_via_position_topic_template_return_json(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -3213,7 +3211,7 @@ async def test_position_via_position_topic_template_all_variables(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -3271,7 +3269,7 @@ async def test_set_state_via_stopped_state_no_position_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -3301,7 +3299,7 @@ async def test_position_via_position_topic_template_return_invalid_json(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -3328,7 +3326,7 @@ async def test_set_position_topic_without_get_position_topic_error(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -3354,7 +3352,7 @@ async def test_value_template_without_state_topic_error(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -3381,7 +3379,7 @@ async def test_position_template_without_position_topic_error(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -3407,7 +3405,7 @@ async def test_set_position_template_without_set_position_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -3433,7 +3431,7 @@ async def test_tilt_command_template_without_tilt_command_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
cover.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -3494,7 +3492,7 @@ async def test_publishing_with_custom_encoding(
"""Test publishing MQTT payload with different encoding."""
domain = cover.DOMAIN
config = deepcopy(DEFAULT_CONFIG)
config[mqtt.DOMAIN][domain]["position_topic"] = "some-position-topic"
config[DOMAIN][domain]["position_topic"] = "some-position-topic"
await help_test_publishing_with_custom_encoding(
hass,
@@ -3541,7 +3539,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
cover.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][cover.DOMAIN],
DEFAULT_CONFIG[DOMAIN][cover.DOMAIN],
topic,
value,
attribute,
tests/components/mqtt/test_date.py
+10 -9
View File
@@ -6,7 +6,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import date, mqtt
from homeassistant.components import date
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.const import ATTR_ASSUMED_STATE, ATTR_ENTITY_ID, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
@@ -44,7 +45,7 @@ from tests.common import async_fire_mqtt_message
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {date.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
DOMAIN: {date.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
}
@@ -64,7 +65,7 @@ async def async_set_value(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
date.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -116,7 +117,7 @@ async def test_controlling_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
date.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -155,7 +156,7 @@ async def test_controlling_validation_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
date.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -219,7 +220,7 @@ async def test_default_availability_payload(
) -> None:
"""Test availability by default payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
date.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -237,7 +238,7 @@ async def test_custom_availability_payload(
) -> None:
"""Test availability by custom payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
date.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -313,7 +314,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
date.DOMAIN: [
{
"name": "Test 1",
@@ -492,7 +493,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
date.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][date.DOMAIN],
DEFAULT_CONFIG[DOMAIN][date.DOMAIN],
topic,
value,
attribute,
tests/components/mqtt/test_datetime.py
+13 -12
View File
@@ -8,7 +8,8 @@ from dateutil.tz import UTC
from freezegun import freeze_time
import pytest
from homeassistant.components import datetime, mqtt
from homeassistant.components import datetime
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.const import ATTR_ASSUMED_STATE, ATTR_ENTITY_ID, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
@@ -46,7 +47,7 @@ from tests.common import async_fire_mqtt_message
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {datetime.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
DOMAIN: {datetime.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
}
@@ -68,7 +69,7 @@ async def async_set_value(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
datetime.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -84,7 +85,7 @@ async def async_set_value(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
datetime.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -143,7 +144,7 @@ async def test_controlling_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
datetime.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -186,7 +187,7 @@ async def test_controlling_validation_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
datetime.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -224,7 +225,7 @@ async def test_ambiguous_date_time_state_update(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
datetime.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -257,7 +258,7 @@ async def test_date_time_with_invalid_timezone_identifier(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
datetime.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -324,7 +325,7 @@ async def test_default_availability_payload(
) -> None:
"""Test availability by default payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
datetime.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -348,7 +349,7 @@ async def test_custom_availability_payload(
) -> None:
"""Test availability by custom payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
datetime.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -430,7 +431,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
datetime.DOMAIN: [
{
"name": "Test 1",
@@ -611,7 +612,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
datetime.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][datetime.DOMAIN],
DEFAULT_CONFIG[DOMAIN][datetime.DOMAIN],
topic,
value,
attribute,
tests/components/mqtt/test_device_tracker.py
+3 -3
View File
@@ -5,7 +5,7 @@ from datetime import UTC, datetime
from freezegun.api import FrozenDateTimeFactory
import pytest
from homeassistant.components import device_tracker, mqtt
from homeassistant.components import device_tracker
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.const import STATE_HOME, STATE_NOT_HOME, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
@@ -27,7 +27,7 @@ from tests.typing import (
)
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
device_tracker.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -693,7 +693,7 @@ async def test_setting_blocked_attribute_via_mqtt_json_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
device_tracker.DOMAIN: {"name": "jan", "state_topic": "/location/jan"}
}
}
tests/components/mqtt/test_diagnostics.py
+3 -2
View File
@@ -6,6 +6,7 @@ from unittest.mock import ANY
import pytest
from homeassistant.components import mqtt
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers import device_registry as dr, entity_registry as er
@@ -29,7 +30,7 @@ async def test_entry_diagnostics(
) -> None:
"""Test config entry diagnostics."""
mqtt_mock = await mqtt_mock_entry()
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry = hass.config_entries.async_entries(DOMAIN)[0]
mqtt_mock.connected = True
await get_diagnostics_for_config_entry(hass, hass_client, config_entry)
@@ -167,7 +168,7 @@ async def test_redact_diagnostics(
expected_config["data"]["password"] = "**REDACTED**"
expected_config["data"]["username"] = "**REDACTED**"
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry = hass.config_entries.async_entries(DOMAIN)[0]
mqtt_mock.connected = True
# Discover a device with a device tracker
tests/components/mqtt/test_discovery.py
+9 -9
View File
@@ -18,7 +18,7 @@ from homeassistant.components.mqtt.abbreviations import (
ABBREVIATIONS,
DEVICE_ABBREVIATIONS,
)
from homeassistant.components.mqtt.const import SUPPORTED_COMPONENTS
from homeassistant.components.mqtt.const import DOMAIN, SUPPORTED_COMPONENTS
from homeassistant.components.mqtt.discovery import (
MQTT_DISCOVERY_DONE,
MQTT_DISCOVERY_NEW,
@@ -207,7 +207,7 @@ async def test_subscribing_config_topic(
) -> None:
"""Test setting up discovery."""
mqtt_mock = await mqtt_mock_entry()
entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
entry = hass.config_entries.async_entries(DOMAIN)[0]
discovery_topic = "homeassistant"
await async_start(hass, discovery_topic, entry)
@@ -1754,7 +1754,7 @@ async def test_cleanup_device_manual(
assert state is not None
# Remove MQTT from the device
mqtt_config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
mqtt_config_entry = hass.config_entries.async_entries(DOMAIN)[0]
mock_debouncer.clear()
response = await ws_client.remove_device(
device_entry.id, mqtt_config_entry.entry_id
@@ -2018,7 +2018,7 @@ async def test_cleanup_device_multiple_config_entries(
)
assert device_entry is not None
mqtt_config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
mqtt_config_entry = hass.config_entries.async_entries(DOMAIN)[0]
sensor_config = {
"device": {"connections": [["mac", "12:34:56:AB:CD:EF"]]},
@@ -2063,7 +2063,7 @@ async def test_cleanup_device_multiple_config_entries(
assert state is not None
# Remove MQTT from the device
mqtt_config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
mqtt_config_entry = hass.config_entries.async_entries(DOMAIN)[0]
response = await ws_client.remove_device(
device_entry.id, mqtt_config_entry.entry_id
)
@@ -2136,7 +2136,7 @@ async def test_cleanup_device_multiple_config_entries_mqtt(
connections={("mac", "12:34:56:AB:CD:EF")},
)
mqtt_config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
mqtt_config_entry = hass.config_entries.async_entries(DOMAIN)[0]
sensor_config = {
"device": {"connections": [["mac", "12:34:56:AB:CD:EF"]]},
@@ -2599,7 +2599,7 @@ async def test_mqtt_integration_discovery_flow_fitering_on_redundant_payload(
birth.set()
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={mqtt.CONF_BROKER: "mock-broker"},
options=ENTRY_DEFAULT_BIRTH_MESSAGE,
version=mqtt.CONFIG_ENTRY_VERSION,
@@ -2670,7 +2670,7 @@ async def test_mqtt_discovery_flow_starts_once(
birth.set()
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={mqtt.CONF_BROKER: "mock-broker"},
options=ENTRY_DEFAULT_BIRTH_MESSAGE,
version=mqtt.CONFIG_ENTRY_VERSION,
@@ -2808,7 +2808,7 @@ async def test_mqtt_discovery_flow_subscribes_atr_configured_qos(
birth.set()
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={mqtt.CONF_BROKER: "mock-broker"},
options=mqtt_config_entry_options,
version=mqtt.CONFIG_ENTRY_VERSION,
tests/components/mqtt/test_event.py
+7 -6
View File
@@ -7,7 +7,8 @@ from unittest.mock import patch
from freezegun.api import FrozenDateTimeFactory
import pytest
from homeassistant.components import event, mqtt
from homeassistant.components import event
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.event import MQTT_EVENT_ATTRIBUTES_BLOCKED
from homeassistant.const import STATE_UNKNOWN
from homeassistant.core import HomeAssistant
@@ -56,7 +57,7 @@ from tests.common import MockConfigEntry, async_fire_mqtt_message
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
event.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -186,7 +187,7 @@ async def test_setting_event_value_with_invalid_payload(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
event.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -319,7 +320,7 @@ async def test_discovery_update_availability(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
event.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -405,7 +406,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
event.DOMAIN: [
{
"name": "Test 1",
@@ -630,7 +631,7 @@ async def test_entity_category(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
event.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
tests/components/mqtt/test_fan.py
+36 -35
View File
@@ -7,7 +7,7 @@ from unittest.mock import patch
import pytest
from voluptuous.error import MultipleInvalid
from homeassistant.components import fan, mqtt
from homeassistant.components import fan
from homeassistant.components.fan import (
ATTR_DIRECTION,
ATTR_OSCILLATING,
@@ -16,6 +16,7 @@ from homeassistant.components.fan import (
ATTR_PRESET_MODES,
NotValidPresetModeError,
)
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.fan import (
CONF_DIRECTION_COMMAND_TOPIC,
CONF_DIRECTION_STATE_TOPIC,
@@ -72,7 +73,7 @@ from tests.components.fan import common
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -82,7 +83,7 @@ DEFAULT_CONFIG = {
}
@pytest.mark.parametrize("hass_config", [{mqtt.DOMAIN: {fan.DOMAIN: {"name": "test"}}}])
@pytest.mark.parametrize("hass_config", [{DOMAIN: {fan.DOMAIN: {"name": "test"}}}])
@pytest.mark.usefixtures("hass")
async def test_fail_setup_if_no_command_topic(
caplog: pytest.LogCaptureFixture, mqtt_mock_entry: MqttMockHAClientGenerator
@@ -96,7 +97,7 @@ async def test_fail_setup_if_no_command_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -232,7 +233,7 @@ async def test_controlling_state_via_topic(
help_custom_config(
fan.DOMAIN,
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"command_topic": "command-topic",
"percentage_command_topic": "percentage-command-topic",
@@ -295,7 +296,7 @@ async def test_controlling_state_via_topic_with_different_speed_range(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -354,7 +355,7 @@ async def test_controlling_state_via_topic_no_percentage_topics(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -472,7 +473,7 @@ async def test_controlling_state_via_topic_and_json_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test",
"state_topic": "shared-state-topic",
@@ -587,7 +588,7 @@ async def test_controlling_state_via_topic_and_json_message_shared_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -734,7 +735,7 @@ async def test_sending_mqtt_commands_and_optimistic(
help_custom_config(
fan.DOMAIN,
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test1",
"command_topic": "command-topic",
@@ -842,7 +843,7 @@ async def test_sending_mqtt_commands_with_alternate_speed_range(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -989,7 +990,7 @@ async def test_sending_mqtt_commands_and_optimistic_no_legacy(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -1197,7 +1198,7 @@ async def test_sending_mqtt_command_templates_(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -1260,7 +1261,7 @@ async def test_sending_mqtt_commands_and_optimistic_no_percentage_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -1576,7 +1577,7 @@ async def test_encoding_subscribable_topics(
attribute_value: Any,
) -> None:
"""Test handling of incoming encoded payload."""
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][fan.DOMAIN])
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][fan.DOMAIN])
config[ATTR_PRESET_MODES] = ["eco", "auto"]
config[CONF_PRESET_MODE_COMMAND_TOPIC] = "fan/some_preset_mode_command_topic"
config[CONF_PERCENTAGE_COMMAND_TOPIC] = "fan/some_percentage_command_topic"
@@ -1598,7 +1599,7 @@ async def test_encoding_subscribable_topics(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -1672,7 +1673,7 @@ async def test_attributes(
(
"test1",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test1",
"command_topic": "command-topic",
@@ -1686,7 +1687,7 @@ async def test_attributes(
(
"test2",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test2",
"command_topic": "command-topic",
@@ -1703,7 +1704,7 @@ async def test_attributes(
(
"test3",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test3",
"command_topic": "command-topic",
@@ -1720,7 +1721,7 @@ async def test_attributes(
(
"test4",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test4",
"command_topic": "command-topic",
@@ -1735,7 +1736,7 @@ async def test_attributes(
(
"test5",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test5",
"command_topic": "command-topic",
@@ -1753,7 +1754,7 @@ async def test_attributes(
(
"test6",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test6",
"command_topic": "command-topic",
@@ -1771,7 +1772,7 @@ async def test_attributes(
(
"test7",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test7",
"command_topic": "command-topic",
@@ -1788,7 +1789,7 @@ async def test_attributes(
(
"test8",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test8",
"command_topic": "command-topic",
@@ -1807,7 +1808,7 @@ async def test_attributes(
(
"test9",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test9",
"command_topic": "command-topic",
@@ -1825,7 +1826,7 @@ async def test_attributes(
(
"test10",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test10",
"command_topic": "command-topic",
@@ -1843,7 +1844,7 @@ async def test_attributes(
(
"test11",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test11",
"command_topic": "command-topic",
@@ -1863,7 +1864,7 @@ async def test_attributes(
(
"test12",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test12",
"command_topic": "command-topic",
@@ -1882,7 +1883,7 @@ async def test_attributes(
(
"test13",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test13",
"command_topic": "command-topic",
@@ -1899,7 +1900,7 @@ async def test_attributes(
(
"test14",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test14",
"command_topic": "command-topic",
@@ -1916,7 +1917,7 @@ async def test_attributes(
(
"test15",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test15",
"command_topic": "command-topic",
@@ -1932,7 +1933,7 @@ async def test_attributes(
(
"test16",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test16",
"command_topic": "command-topic",
@@ -1951,7 +1952,7 @@ async def test_attributes(
(
"test17",
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: {
"name": "test17",
"command_topic": "command-topic",
@@ -2096,7 +2097,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
fan.DOMAIN: [
{
"name": "Test 1",
@@ -2293,7 +2294,7 @@ async def test_publishing_with_custom_encoding(
domain = fan.DOMAIN
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG)
if topic == "preset_mode_command_topic":
config[mqtt.DOMAIN][domain]["preset_modes"] = ["auto", "eco"]
config[DOMAIN][domain]["preset_modes"] = ["auto", "eco"]
await help_test_publishing_with_custom_encoding(
hass,
tests/components/mqtt/test_humidifier.py
+29 -31
View File
@@ -7,7 +7,7 @@ from unittest.mock import patch
import pytest
from voluptuous.error import MultipleInvalid
from homeassistant.components import humidifier, mqtt
from homeassistant.components import humidifier
from homeassistant.components.humidifier import (
ATTR_CURRENT_HUMIDITY,
ATTR_HUMIDITY,
@@ -16,7 +16,7 @@ from homeassistant.components.humidifier import (
SERVICE_SET_MODE,
HumidifierAction,
)
from homeassistant.components.mqtt.const import CONF_CURRENT_HUMIDITY_TOPIC
from homeassistant.components.mqtt.const import CONF_CURRENT_HUMIDITY_TOPIC, DOMAIN
from homeassistant.components.mqtt.humidifier import (
CONF_MODE_COMMAND_TOPIC,
CONF_MODE_STATE_TOPIC,
@@ -71,7 +71,7 @@ from tests.common import async_fire_mqtt_message
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -133,7 +133,7 @@ async def async_set_humidity(
@pytest.mark.parametrize(
"hass_config", [{mqtt.DOMAIN: {humidifier.DOMAIN: {"name": "test"}}}]
"hass_config", [{DOMAIN: {humidifier.DOMAIN: {"name": "test"}}}]
)
@pytest.mark.usefixtures("hass")
async def test_fail_setup_if_no_command_topic(
@@ -148,7 +148,7 @@ async def test_fail_setup_if_no_command_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"action_topic": "action-topic",
@@ -308,7 +308,7 @@ async def test_controlling_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"action_topic": "action-topic",
@@ -449,7 +449,7 @@ async def test_controlling_state_via_topic_and_json_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"state_topic": "shared-state-topic",
@@ -528,7 +528,7 @@ async def test_controlling_state_via_topic_and_json_message_shared_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -627,7 +627,7 @@ async def test_sending_mqtt_commands_and_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -731,7 +731,7 @@ async def test_sending_mqtt_command_templates_(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -870,9 +870,7 @@ async def test_encoding_subscribable_topics(
attribute_value: Any,
) -> None:
"""Test handling of incoming encoded payload."""
config: dict[str, Any] = copy.deepcopy(
DEFAULT_CONFIG[mqtt.DOMAIN][humidifier.DOMAIN]
)
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][humidifier.DOMAIN])
config["modes"] = ["eco", "auto"]
config[CONF_MODE_COMMAND_TOPIC] = "humidifier/some_mode_command_topic"
await help_test_encoding_subscribable_topics(
@@ -891,7 +889,7 @@ async def test_encoding_subscribable_topics(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -941,7 +939,7 @@ async def test_attributes(
[
( # test valid case 1
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -953,7 +951,7 @@ async def test_attributes(
),
( # test valid case 2
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -966,7 +964,7 @@ async def test_attributes(
),
( # test valid case 3
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -979,7 +977,7 @@ async def test_attributes(
),
( # test valid case 4
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -992,7 +990,7 @@ async def test_attributes(
),
( # test invalid device_class
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -1005,7 +1003,7 @@ async def test_attributes(
),
( # test mode_command_topic without modes
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -1018,7 +1016,7 @@ async def test_attributes(
),
( # test invalid humidity min max case 1
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -1032,7 +1030,7 @@ async def test_attributes(
),
( # test invalid humidity min max case 2
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -1046,7 +1044,7 @@ async def test_attributes(
),
( # test invalid mode, is reset payload
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -1075,7 +1073,7 @@ async def test_validity_configurations(
(
"test1",
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test1",
"command_topic": "command-topic",
@@ -1089,7 +1087,7 @@ async def test_validity_configurations(
(
"test2",
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test2",
"command_topic": "command-topic",
@@ -1105,7 +1103,7 @@ async def test_validity_configurations(
(
"test3",
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test3",
"command_topic": "command-topic",
@@ -1119,7 +1117,7 @@ async def test_validity_configurations(
(
"test4",
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test4",
"command_topic": "command-topic",
@@ -1135,7 +1133,7 @@ async def test_validity_configurations(
(
"test5",
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test5",
"command_topic": "command-topic",
@@ -1148,7 +1146,7 @@ async def test_validity_configurations(
(
"test6",
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: {
"name": "test6",
"target_humidity_command_topic": "humidity-command-topic",
@@ -1291,7 +1289,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
humidifier.DOMAIN: [
{
"name": "Test 1",
@@ -1494,7 +1492,7 @@ async def test_publishing_with_custom_encoding(
domain = humidifier.DOMAIN
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG)
if topic == "mode_command_topic":
config[mqtt.DOMAIN][domain]["modes"] = ["auto", "eco"]
config[DOMAIN][domain]["modes"] = ["auto", "eco"]
await help_test_publishing_with_custom_encoding(
hass,
tests/components/mqtt/test_image.py
+16 -17
View File
@@ -10,7 +10,8 @@ import httpx
import pytest
import respx
from homeassistant.components import image, mqtt
from homeassistant.components import image
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.const import STATE_UNAVAILABLE, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
@@ -49,15 +50,13 @@ from tests.typing import (
MqttMockPahoClient,
)
DEFAULT_CONFIG = {
mqtt.DOMAIN: {image.DOMAIN: {"name": "test", "image_topic": "test_topic"}}
}
DEFAULT_CONFIG = {DOMAIN: {image.DOMAIN: {"name": "test", "image_topic": "test_topic"}}}
@pytest.mark.freeze_time("2023-04-01 00:00:00+00:00")
@pytest.mark.parametrize(
"hass_config",
[{mqtt.DOMAIN: {image.DOMAIN: {"image_topic": "test/image", "name": "Test"}}}],
[{DOMAIN: {image.DOMAIN: {"image_topic": "test/image", "name": "Test"}}}],
)
async def test_run_image_setup(
hass: HomeAssistant,
@@ -93,7 +92,7 @@ async def test_run_image_setup(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
image.DOMAIN: {
"image_topic": "test/image",
"name": "Test",
@@ -147,7 +146,7 @@ async def test_run_image_b64_encoded(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
"image": {
"image_topic": "test/image",
"name": "Test",
@@ -200,7 +199,7 @@ async def test_image_b64_encoded_with_availability(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
"image": {
"url_topic": "test/image",
"name": "Test",
@@ -281,7 +280,7 @@ async def test_image_from_url(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
"image": {
"url_topic": "test/image",
"name": "Test",
@@ -334,7 +333,7 @@ async def test_image_from_url_with_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
"image": {
"url_topic": "test/image",
"name": "Test",
@@ -402,7 +401,7 @@ async def test_image_from_url_content_type(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
"image": {
"url_topic": "test/image",
"name": "Test",
@@ -462,7 +461,7 @@ async def test_image_from_url_fails(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
"image": {
"url_topic": "test/image",
"content_type": "image/jpg",
@@ -475,7 +474,7 @@ async def test_image_from_url_fails(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
"image": {
"url_topic": "test/image",
"image_topic": "test/image-data-topic",
@@ -488,7 +487,7 @@ async def test_image_from_url_fails(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
"image": {
"name": "Test",
"encoding": "utf-8",
@@ -601,7 +600,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
image.DOMAIN: [
{
"name": "Test 1",
@@ -629,7 +628,7 @@ async def test_discovery_removal_image(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test removal of discovered image."""
data = json.dumps(DEFAULT_CONFIG[mqtt.DOMAIN][image.DOMAIN])
data = json.dumps(DEFAULT_CONFIG[DOMAIN][image.DOMAIN])
await help_test_discovery_removal(hass, mqtt_mock_entry, image.DOMAIN, data)
@@ -806,7 +805,7 @@ async def test_skipped_async_ha_write_state(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
image.DOMAIN: {
"name": "test",
"url_topic": "test-topic",
tests/components/mqtt/test_init.py
+45 -48
View File
@@ -18,6 +18,7 @@ import voluptuous as vol
from homeassistant import core as ha
from homeassistant.components import mqtt
from homeassistant.components.mqtt import debug_info
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.models import (
MessageCallbackType,
MqttCommandTemplateException,
@@ -272,7 +273,7 @@ async def test_service_call_without_topic_does_not_publish(
mqtt_mock = await mqtt_mock_entry()
with pytest.raises(vol.Invalid):
await hass.services.async_call(
mqtt.DOMAIN,
DOMAIN,
mqtt.SERVICE_PUBLISH,
{},
blocking=True,
@@ -284,7 +285,7 @@ async def test_service_call_mqtt_entry_does_not_publish(
hass: HomeAssistant, mqtt_client_mock: MqttMockPahoClient
) -> None:
"""Test the service call if topic is missing."""
assert await async_setup_component(hass, mqtt.DOMAIN, {})
assert await async_setup_component(hass, DOMAIN, {})
with pytest.raises(
ServiceValidationError,
match=(
@@ -292,7 +293,7 @@ async def test_service_call_mqtt_entry_does_not_publish(
),
):
await hass.services.async_call(
mqtt.DOMAIN,
DOMAIN,
mqtt.SERVICE_PUBLISH,
{
mqtt.ATTR_TOPIC: "test_topic",
@@ -312,7 +313,7 @@ async def test_service_call_with_template_topic_renders_invalid_topic(
mqtt_mock = await mqtt_mock_entry()
with pytest.raises(vol.Invalid) as exc:
await hass.services.async_call(
mqtt.DOMAIN,
DOMAIN,
mqtt.SERVICE_PUBLISH,
{
mqtt.ATTR_TOPIC: "test/{{ '+' if True else 'topic' }}/topic",
@@ -356,7 +357,7 @@ async def test_mqtt_publish_action_call_with_raw_data(
"""
mqtt_mock = await mqtt_mock_entry()
await hass.services.async_call(
mqtt.DOMAIN,
DOMAIN,
mqtt.SERVICE_PUBLISH,
{
mqtt.ATTR_TOPIC: "test/topic",
@@ -372,7 +373,7 @@ async def test_mqtt_publish_action_call_with_raw_data(
"homeassistant.components.mqtt.models.literal_eval"
) as literal_eval_mock:
await hass.services.async_call(
mqtt.DOMAIN,
DOMAIN,
mqtt.SERVICE_PUBLISH,
{
mqtt.ATTR_TOPIC: "test/topic",
@@ -383,7 +384,7 @@ async def test_mqtt_publish_action_call_with_raw_data(
literal_eval_mock.assert_not_called()
await hass.services.async_call(
mqtt.DOMAIN,
DOMAIN,
mqtt.SERVICE_PUBLISH,
{
mqtt.ATTR_TOPIC: "test/topic",
@@ -404,7 +405,7 @@ async def test_service_call_with_ascii_qos_retain_flags(
"""
mqtt_mock = await mqtt_mock_entry()
await hass.services.async_call(
mqtt.DOMAIN,
DOMAIN,
mqtt.SERVICE_PUBLISH,
{
mqtt.ATTR_TOPIC: "test/topic",
@@ -423,7 +424,7 @@ async def test_service_call_with_ascii_qos_retain_flags(
# Test service call without payload
await hass.services.async_call(
mqtt.DOMAIN,
DOMAIN,
mqtt.SERVICE_PUBLISH,
{
mqtt.ATTR_TOPIC: "test/topic",
@@ -485,7 +486,7 @@ async def test_publish_action_with_message_expiry_interval(
"""Test the publish action message expiry interval option."""
mqtt_mock = await mqtt_mock_entry()
await hass.services.async_call(
mqtt.DOMAIN,
DOMAIN,
mqtt.SERVICE_PUBLISH,
{
mqtt.ATTR_TOPIC: "test/topic",
@@ -708,7 +709,7 @@ def test_entity_device_info_schema() -> None:
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
"sensor": [
{
"name": "test-sensor",
@@ -807,13 +808,13 @@ async def test_reload_entry_with_restored_subscriptions(
"""Test reloading the config entry with with subscriptions restored."""
# Setup the MQTT entry
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={mqtt.CONF_BROKER: "test-broker"},
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
entry.add_to_hass(hass)
hass.config.components.add(mqtt.DOMAIN)
hass.config.components.add(DOMAIN)
with patch("homeassistant.config.load_yaml_config_file", return_value={}):
await hass.config_entries.async_setup(entry.entry_id)
@@ -873,7 +874,7 @@ async def test_reload_entry_with_restored_subscriptions(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
"light": {
"platform": "mqtt",
"name": "test",
@@ -894,7 +895,7 @@ async def test_setup_manual_mqtt_with_platform_key(
)
@pytest.mark.parametrize("hass_config", [{mqtt.DOMAIN: {"light": {"name": "test"}}}])
@pytest.mark.parametrize("hass_config", [{DOMAIN: {"light": {"name": "test"}}}])
async def test_setup_manual_mqtt_with_invalid_config(
mqtt_mock_entry: MqttMockHAClientGenerator, caplog: pytest.LogCaptureFixture
) -> None:
@@ -916,13 +917,13 @@ async def test_default_entry_setting_are_applied(
# Config entry data is incomplete but valid according the schema
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={"broker": "test-broker", "port": 1234},
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
entry.add_to_hass(hass)
hass.config.components.add(mqtt.DOMAIN)
hass.config.components.add(DOMAIN)
assert await hass.config_entries.async_setup(entry.entry_id)
await hass.async_block_till_done()
@@ -1122,7 +1123,7 @@ async def test_mqtt_ws_remove_discovered_device(
assert device_entry is not None
client = await hass_ws_client(hass)
mqtt_config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
mqtt_config_entry = hass.config_entries.async_entries(DOMAIN)[0]
response = await client.remove_device(device_entry.id, mqtt_config_entry.entry_id)
assert response["success"]
@@ -1460,7 +1461,7 @@ async def test_debug_info_non_mqtt(
)
assert await async_setup_component(
hass, mqtt.DOMAIN, {mqtt.DOMAIN: {domain: {"platform": "test"}}}
hass, DOMAIN, {DOMAIN: {domain: {"platform": "test"}}}
)
debug_info_data = debug_info.info_for_device(hass, device_entry.id)
@@ -1759,20 +1760,20 @@ async def test_unload_config_entry(
) -> None:
"""Test unloading the MQTT entry."""
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={mqtt.CONF_BROKER: "test-broker", CONF_PROTOCOL: "5"},
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
)
entry.add_to_hass(hass)
assert await async_setup_component(hass, mqtt.DOMAIN, {})
assert hass.services.has_service(mqtt.DOMAIN, "dump")
assert hass.services.has_service(mqtt.DOMAIN, "publish")
assert await async_setup_component(hass, DOMAIN, {})
assert hass.services.has_service(DOMAIN, "dump")
assert hass.services.has_service(DOMAIN, "publish")
# Make sure the debouncer finishes
await mock_debouncer.wait()
mqtt_config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
mqtt_config_entry = hass.config_entries.async_entries(DOMAIN)[0]
assert mqtt_config_entry.state is ConfigEntryState.LOADED
# Publish just before unloading to test await cleanup
@@ -1802,8 +1803,8 @@ async def test_unload_config_entry(
assert new_mqtt_config_entry.state is ConfigEntryState.NOT_LOADED
await hass.async_block_till_done(wait_background_tasks=True)
assert hass.services.has_service(mqtt.DOMAIN, "dump")
assert hass.services.has_service(mqtt.DOMAIN, "publish")
assert hass.services.has_service(DOMAIN, "dump")
assert hass.services.has_service(DOMAIN, "publish")
assert "No ACK from MQTT server" not in caplog.text
@@ -1811,9 +1812,9 @@ async def test_publish_or_subscribe_without_valid_config_entry(
hass: HomeAssistant, record_calls: MessageCallbackType
) -> None:
"""Test internal publish function with bad use cases."""
assert await async_setup_component(hass, mqtt.DOMAIN, {})
assert hass.services.has_service(mqtt.DOMAIN, "dump")
assert hass.services.has_service(mqtt.DOMAIN, "publish")
assert await async_setup_component(hass, DOMAIN, {})
assert hass.services.has_service(DOMAIN, "dump")
assert hass.services.has_service(DOMAIN, "publish")
with pytest.raises(HomeAssistantError):
await mqtt.async_publish(
hass, "some-topic", "test-payload", qos=0, retain=False, encoding=None
@@ -1846,7 +1847,7 @@ async def test_disabling_and_enabling_entry(
) -> None:
"""Test disabling and enabling the config entry."""
await mqtt_mock_entry()
entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
entry = hass.config_entries.async_entries(DOMAIN)[0]
assert entry.state is ConfigEntryState.LOADED
# Late discovery of a mqtt entity
config_tag = (
@@ -1913,7 +1914,7 @@ async def test_disabling_and_enabling_entry(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
"light": [
{
"name": "test1",
@@ -1932,7 +1933,7 @@ async def test_disabling_and_enabling_entry(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
"light": [
{
"name": "test1",
@@ -2019,8 +2020,8 @@ async def test_link_config_entry(
entity_names = ["test_manual", "test_discovery"]
# Check if both entities were linked to the MQTT config entry
mqtt_config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
mqtt_platforms = async_get_platforms(hass, mqtt.DOMAIN)
mqtt_config_entry = hass.config_entries.async_entries(DOMAIN)[0]
mqtt_platforms = async_get_platforms(hass, DOMAIN)
@callback
def _check_entities() -> int:
@@ -2100,12 +2101,12 @@ async def test_reload_config_entry(
await hass.async_block_till_done()
assert hass.states.get("sensor.test_discovery") is not None
entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
entry = hass.config_entries.async_entries(DOMAIN)[0]
@callback
def _check_entities() -> int:
entities: list[Entity] = []
mqtt_platforms = async_get_platforms(hass, mqtt.DOMAIN)
mqtt_platforms = async_get_platforms(hass, DOMAIN)
for mqtt_platform in mqtt_platforms:
assert mqtt_platform.config_entry is entry
entities += (entity for entity in mqtt_platform.entities.values())
@@ -2439,16 +2440,14 @@ async def test_mqtt_integration_level_imports(attr: str) -> None:
@pytest.mark.usefixtures("mqtt_client_mock")
@pytest.mark.parametrize(
"hass_config", [{mqtt.DOMAIN: {"sensor": {"state_topic": "test-topic"}}}]
"hass_config", [{DOMAIN: {"sensor": {"state_topic": "test-topic"}}}]
)
async def test_yaml_config_without_entry(
hass: HomeAssistant, hass_config: ConfigType, issue_registry: ir.IssueRegistry
) -> None:
"""Test a repair issue is created for YAML setup without an active config entry."""
await async_setup_component(hass, mqtt.DOMAIN, hass_config)
issue = issue_registry.async_get_issue(
mqtt.DOMAIN, "yaml_setup_without_active_setup"
)
await async_setup_component(hass, DOMAIN, hass_config)
issue = issue_registry.async_get_issue(DOMAIN, "yaml_setup_without_active_setup")
assert issue is not None
assert (
issue.learn_more_url == "https://www.home-assistant.io/integrations/mqtt/"
@@ -2457,7 +2456,7 @@ async def test_yaml_config_without_entry(
@pytest.mark.parametrize(
"hass_config", [{mqtt.DOMAIN: {"sensor": {"state_topic": "test-topic"}}}]
"hass_config", [{DOMAIN: {"sensor": {"state_topic": "test-topic"}}}]
)
async def test_yaml_config_with_active_mqtt_config_entry(
hass: HomeAssistant,
@@ -2467,9 +2466,7 @@ async def test_yaml_config_with_active_mqtt_config_entry(
) -> None:
"""Test no repair issue is created for YAML setup with an active config entry."""
await mqtt_mock_entry()
issue = issue_registry.async_get_issue(
mqtt.DOMAIN, "yaml_setup_without_active_setup"
)
issue = issue_registry.async_get_issue(DOMAIN, "yaml_setup_without_active_setup")
state = hass.states.get("sensor.mqtt_sensor")
assert state is not None
assert issue is None
@@ -2520,7 +2517,7 @@ async def test_mqtt_protocol_successful_migration_to_v5(
"The MQTT protocol version was successfully updated to version 5"
in caplog.text
)
entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
entry = hass.config_entries.async_entries(DOMAIN)[0]
assert entry.data[mqtt.CONF_PROTOCOL] == mqtt.PROTOCOL_5
@@ -2582,7 +2579,7 @@ async def test_mqtt_protocol_failed_migration_to_v5(
issue_registry = ir.async_get(hass)
assert len(issue_registry.issues) == 1
issue = issue_registry.async_get_issue(mqtt.DOMAIN, "protocol_5_migration")
issue = issue_registry.async_get_issue(DOMAIN, "protocol_5_migration")
assert issue is not None
assert issue.translation_key == "protocol_5_migration"
assert issue.translation_placeholders == {
@@ -2593,5 +2590,5 @@ async def test_mqtt_protocol_failed_migration_to_v5(
"The MQTT protocol version was successfully updated to version 5"
not in caplog.text
)
entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
entry = hass.config_entries.async_entries(DOMAIN)[0]
assert entry.data.get(mqtt.CONF_PROTOCOL, mqtt.PROTOCOL_311) == current_protocol
tests/components/mqtt/test_lawn_mower.py
+16 -17
View File
@@ -7,7 +7,7 @@ from unittest.mock import patch
import pytest
from homeassistant.components import lawn_mower, mqtt
from homeassistant.components import lawn_mower
from homeassistant.components.lawn_mower import (
DOMAIN as LAWN_MOWER_DOMAIN,
SERVICE_DOCK,
@@ -15,6 +15,7 @@ from homeassistant.components.lawn_mower import (
SERVICE_START_MOWING,
LawnMowerEntityFeature,
)
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.lawn_mower import MQTT_LAWN_MOWER_ATTRIBUTES_BLOCKED
from homeassistant.const import ATTR_ASSUMED_STATE, ATTR_ENTITY_ID, STATE_UNKNOWN
from homeassistant.core import HomeAssistant, State
@@ -63,7 +64,7 @@ DEFAULT_FEATURES = (
)
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
lawn_mower.DOMAIN: {
"name": "test",
"dock_command_topic": "dock-test-topic",
@@ -78,7 +79,7 @@ DEFAULT_CONFIG = {
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lawn_mower.DOMAIN: {
"activity_state_topic": "test/lawn_mower_stat",
"dock_command_topic": "dock-test-topic",
@@ -135,7 +136,7 @@ async def test_run_lawn_mower_setup_and_state_updates(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
lawn_mower.DOMAIN: {
"pause_command_topic": "pause-test-topic",
"name": "test",
@@ -146,7 +147,7 @@ async def test_run_lawn_mower_setup_and_state_updates(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
lawn_mower.DOMAIN: {
"dock_command_topic": "dock-test-topic",
"start_mowing_command_topic": "start_mowing-test-topic",
@@ -175,7 +176,7 @@ async def test_supported_features(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lawn_mower.DOMAIN: {
"activity_state_topic": "test/lawn_mower_stat",
"name": "Test Lawn Mower",
@@ -289,7 +290,7 @@ async def test_run_lawn_mower_service_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lawn_mower.DOMAIN: {
"pause_command_topic": "test/lawn_mower_pause_cmd",
"name": "Test Lawn Mower",
@@ -315,7 +316,7 @@ async def test_restore_lawn_mower_from_invalid_state(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lawn_mower.DOMAIN: {
"name": "Test Lawn Mower",
"dock_command_topic": "test/lawn_mower_dock_cmd",
@@ -501,7 +502,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lawn_mower.DOMAIN: [
{
"name": "Test 1",
@@ -529,7 +530,7 @@ async def test_discovery_removal_lawn_mower(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test removal of discovered lawn_mower."""
data = json.dumps(DEFAULT_CONFIG[mqtt.DOMAIN][lawn_mower.DOMAIN])
data = json.dumps(DEFAULT_CONFIG[DOMAIN][lawn_mower.DOMAIN])
await help_test_discovery_removal(hass, mqtt_mock_entry, lawn_mower.DOMAIN, data)
@@ -628,7 +629,7 @@ async def test_entity_id_update_subscriptions(
) -> None:
"""Test MQTT subscriptions are managed when entity_id is updated."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
lawn_mower.DOMAIN: {
"name": "test",
"activity_state_topic": "test-topic",
@@ -687,7 +688,7 @@ async def test_entity_debug_info_message(
) -> None:
"""Test MQTT debug info."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
lawn_mower.DOMAIN: {
"activity_state_topic": "test/lawn_mower_stat",
"dock_command_topic": "dock-test-topic",
@@ -714,7 +715,7 @@ async def test_entity_debug_info_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lawn_mower.DOMAIN: {
"dock_command_topic": "dock-test-topic",
"pause_command_topic": "pause-test-topic",
@@ -826,9 +827,7 @@ async def test_encoding_subscribable_topics(
attribute_value: Any,
) -> None:
"""Test handling of incoming encoded payload."""
config: dict[str, Any] = copy.deepcopy(
DEFAULT_CONFIG[mqtt.DOMAIN][lawn_mower.DOMAIN]
)
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][lawn_mower.DOMAIN])
config["actions"] = ["milk", "beer"]
await help_test_encoding_subscribable_topics(
hass,
@@ -935,7 +934,7 @@ async def test_skipped_async_ha_write_state(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lawn_mower.DOMAIN: {
"name": "test",
"activity_state_topic": "test-topic",
tests/components/mqtt/test_light.py
+43 -44
View File
@@ -192,7 +192,8 @@ from unittest.mock import call, patch
import pytest
from homeassistant.components import light, mqtt
from homeassistant.components import light
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.light.schema_basic import (
CONF_BRIGHTNESS_COMMAND_TOPIC,
CONF_COLOR_TEMP_COMMAND_TOPIC,
@@ -246,13 +247,11 @@ from tests.components.light import common
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {light.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
DOMAIN: {light.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
}
@pytest.mark.parametrize(
"hass_config", [{mqtt.DOMAIN: {light.DOMAIN: {"name": "test"}}}]
)
@pytest.mark.parametrize("hass_config", [{DOMAIN: {light.DOMAIN: {"name": "test"}}}])
async def test_fail_setup_if_no_command_topic(
hass: HomeAssistant,
mqtt_mock_entry: MqttMockHAClientGenerator,
@@ -267,7 +266,7 @@ async def test_fail_setup_if_no_command_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"state_topic": "test_light_rgb/status",
@@ -421,7 +420,7 @@ async def test_no_min_max_kelvin(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"state_topic": "test_light_rgb/status",
@@ -548,7 +547,7 @@ async def test_controlling_state_via_topic(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"state_topic": "test_light_color_temp/status",
@@ -574,7 +573,7 @@ async def test_controlling_state_via_topic(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"state_topic": "test_light_color_temp/status",
@@ -803,7 +802,7 @@ async def test_received_rgbx_values_set_state_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"state_topic": "test_light_rgb/status",
@@ -935,7 +934,7 @@ async def test_invalid_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"state_topic": "test_scale/status",
@@ -985,7 +984,7 @@ async def test_brightness_controlling_scale(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"state_topic": "test_scale_rgb/status",
@@ -1059,7 +1058,7 @@ async def test_brightness_from_rgb_controlling_scale(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"state_topic": "test_light_rgb/status",
@@ -1179,7 +1178,7 @@ async def test_controlling_state_via_topic_with_templates(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light_rgb/set",
@@ -1445,7 +1444,7 @@ async def test_sending_mqtt_commands_and_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light_rgb/set",
@@ -1494,7 +1493,7 @@ async def test_sending_mqtt_rgb_command_with_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light_rgb/set",
@@ -1543,7 +1542,7 @@ async def test_sending_mqtt_rgbw_command_with_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light_rgb/set",
@@ -1593,7 +1592,7 @@ async def test_sending_mqtt_rgbww_command_with_template(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light_color_temp/set",
@@ -1614,7 +1613,7 @@ async def test_sending_mqtt_rgbww_command_with_template(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light_color_temp/set",
@@ -1674,7 +1673,7 @@ async def test_sending_mqtt_color_temp_command_with_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light/set",
@@ -1718,7 +1717,7 @@ async def test_on_command_first(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light/set",
@@ -1761,7 +1760,7 @@ async def test_on_command_last(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light/set",
@@ -1826,7 +1825,7 @@ async def test_on_command_brightness(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light/set",
@@ -1908,7 +1907,7 @@ async def test_on_command_brightness_scaled(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light/set",
@@ -2010,7 +2009,7 @@ async def test_on_command_rgb(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light/set",
@@ -2126,7 +2125,7 @@ async def test_on_command_rgbw(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light/set",
@@ -2246,7 +2245,7 @@ async def test_on_command_rgbww(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light/set",
@@ -2293,7 +2292,7 @@ async def test_on_command_rgb_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light/set",
@@ -2345,7 +2344,7 @@ async def test_on_command_rgbw_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light/set",
@@ -2397,7 +2396,7 @@ async def test_on_command_rgbww_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "tasmota_B94927/cmnd/POWER",
@@ -2509,7 +2508,7 @@ async def test_on_command_white(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"state_topic": "test_light_rgb/status",
@@ -2664,7 +2663,7 @@ async def test_explicit_color_mode(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"state_topic": "test_light_rgb/status",
@@ -2752,7 +2751,7 @@ async def test_explicit_color_mode_templated(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"state_topic": "tasmota_B94927/tele/STATE",
@@ -2825,7 +2824,7 @@ async def test_white_state_update(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light/set",
@@ -2976,7 +2975,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: [
{
"name": "Test 1",
@@ -3636,7 +3635,7 @@ async def test_entity_debug_info_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_max_mireds/set",
@@ -3760,9 +3759,9 @@ async def test_publishing_with_custom_encoding(
domain = light.DOMAIN
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG)
if topic == "effect_command_topic":
config[mqtt.DOMAIN][domain]["effect_list"] = ["random", "color_loop"]
config[DOMAIN][domain]["effect_list"] = ["random", "color_loop"]
elif topic == "white_command_topic":
config[mqtt.DOMAIN][domain]["rgb_command_topic"] = "some-cmd-topic"
config[DOMAIN][domain]["rgb_command_topic"] = "some-cmd-topic"
await help_test_publishing_with_custom_encoding(
hass,
@@ -3847,7 +3846,7 @@ async def test_encoding_subscribable_topics(
init_payload: tuple[str, str] | None,
) -> None:
"""Test handling of incoming encoded payload."""
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][light.DOMAIN])
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][light.DOMAIN])
config[CONF_EFFECT_COMMAND_TOPIC] = "light/CONF_EFFECT_COMMAND_TOPIC"
config[CONF_RGB_COMMAND_TOPIC] = "light/CONF_RGB_COMMAND_TOPIC"
config[CONF_BRIGHTNESS_COMMAND_TOPIC] = "light/CONF_BRIGHTNESS_COMMAND_TOPIC"
@@ -3888,7 +3887,7 @@ async def test_encoding_subscribable_topics_brightness(
init_payload: tuple[str, str] | None,
) -> None:
"""Test handling of incoming encoded payload for a brightness only light."""
config = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][light.DOMAIN])
config = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][light.DOMAIN])
config[CONF_BRIGHTNESS_COMMAND_TOPIC] = "light/CONF_BRIGHTNESS_COMMAND_TOPIC"
await help_test_encoding_subscribable_topics(
@@ -3908,7 +3907,7 @@ async def test_encoding_subscribable_topics_brightness(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light_brightness/set",
@@ -3962,7 +3961,7 @@ async def test_sending_mqtt_brightness_command_with_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light_brightness/set",
@@ -4017,7 +4016,7 @@ async def test_sending_mqtt_effect_command_with_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light_hs/set",
@@ -4065,7 +4064,7 @@ async def test_sending_mqtt_hs_command_with_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"name": "test",
"command_topic": "test_light_xy/set",
tests/components/mqtt/test_light_json.py
+40 -39
View File
@@ -88,7 +88,8 @@ from unittest.mock import call, patch
import pytest
from homeassistant.components import light, mqtt
from homeassistant.components import light
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.light.schema_basic import (
MQTT_LIGHT_ATTRIBUTES_BLOCKED,
)
@@ -139,7 +140,7 @@ from tests.components.light import common
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -150,7 +151,7 @@ DEFAULT_CONFIG = {
COLOR_MODES_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"brightness": True,
"effect": True,
@@ -262,7 +263,7 @@ async def test_simple_on_off_light(
@pytest.mark.parametrize(
"hass_config", [{mqtt.DOMAIN: {light.DOMAIN: {"schema": "json", "name": "test"}}}]
"hass_config", [{DOMAIN: {light.DOMAIN: {"schema": "json", "name": "test"}}}]
)
async def test_fail_setup_if_no_command_topic(
mqtt_mock_entry: MqttMockHAClientGenerator,
@@ -391,7 +392,7 @@ async def test_controlling_state_with_unknown_color_mode(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -447,7 +448,7 @@ async def test_no_color_brightness_color_temp_if_no_topics(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -458,7 +459,7 @@ async def test_no_color_brightness_color_temp_if_no_topics(
}
},
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -518,7 +519,7 @@ async def test_brightness_only(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -529,7 +530,7 @@ async def test_brightness_only(
}
},
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -570,7 +571,7 @@ async def test_single_color_mode_turn_on(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -580,7 +581,7 @@ async def test_single_color_mode_turn_on(
}
},
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -627,7 +628,7 @@ async def test_restore_state_with_none_color_mode(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -687,7 +688,7 @@ async def test_color_temp_only(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -778,7 +779,7 @@ async def test_controlling_state_color_temp_kelvin(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -791,7 +792,7 @@ async def test_controlling_state_color_temp_kelvin(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -806,7 +807,7 @@ async def test_controlling_state_color_temp_kelvin(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -821,7 +822,7 @@ async def test_controlling_state_color_temp_kelvin(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -836,7 +837,7 @@ async def test_controlling_state_color_temp_kelvin(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -1048,7 +1049,7 @@ async def test_controlling_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"command_topic": "test_light_rgb/set",
"effect": True,
@@ -1296,7 +1297,7 @@ async def test_sending_mqtt_commands_and_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -1361,7 +1362,7 @@ async def test_sending_hs_color(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -1422,7 +1423,7 @@ async def test_sending_rgb_color_no_brightness(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"command_topic": "test_light_rgb/set",
"name": "test",
@@ -1507,7 +1508,7 @@ async def test_sending_rgb_color_no_brightness2(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -1579,7 +1580,7 @@ async def test_sending_rgb_color_with_brightness(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -1653,7 +1654,7 @@ async def test_sending_rgb_color_with_scaled_brightness(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -1710,7 +1711,7 @@ async def test_sending_scaled_white(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -1774,7 +1775,7 @@ async def test_sending_xy_color(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -1848,7 +1849,7 @@ async def test_effect(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -1931,7 +1932,7 @@ async def test_flash_short_and_long(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -1985,7 +1986,7 @@ async def test_transition(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -2048,7 +2049,7 @@ async def test_brightness_scale(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -2246,7 +2247,7 @@ async def test_update_discovery_with_members_without_init(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -2537,7 +2538,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: [
{
"name": "Test 1",
@@ -2713,7 +2714,7 @@ async def test_entity_debug_info_message(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -2728,7 +2729,7 @@ async def test_entity_debug_info_message(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -2743,7 +2744,7 @@ async def test_entity_debug_info_message(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -2758,7 +2759,7 @@ async def test_entity_debug_info_message(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "json",
"name": "test",
@@ -2826,7 +2827,7 @@ async def test_publishing_with_custom_encoding(
domain = light.DOMAIN
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG)
if topic == "effect_command_topic":
config[mqtt.DOMAIN][domain]["effect_list"] = ["random", "color_loop"]
config[DOMAIN][domain]["effect_list"] = ["random", "color_loop"]
await help_test_publishing_with_custom_encoding(
hass,
@@ -2878,7 +2879,7 @@ async def test_encoding_subscribable_topics(
init_payload: tuple[str, str] | None,
) -> None:
"""Test handling of incoming encoded payload."""
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][light.DOMAIN])
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][light.DOMAIN])
config["color_mode"] = True
config["supported_color_modes"] = [
"color_temp",
tests/components/mqtt/test_light_template.py
+24 -23
View File
@@ -31,7 +31,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import light, mqtt
from homeassistant.components import light
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.light.schema_basic import (
MQTT_LIGHT_ATTRIBUTES_BLOCKED,
)
@@ -81,7 +82,7 @@ from tests.components.light import common
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -96,10 +97,10 @@ DEFAULT_CONFIG = {
@pytest.mark.parametrize(
"hass_config",
[
({mqtt.DOMAIN: {light.DOMAIN: {"schema": "template", "name": "test"}}},),
({DOMAIN: {light.DOMAIN: {"schema": "template", "name": "test"}}},),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -110,7 +111,7 @@ DEFAULT_CONFIG = {
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -122,7 +123,7 @@ DEFAULT_CONFIG = {
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -149,7 +150,7 @@ async def test_setup_fails(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -185,7 +186,7 @@ async def test_rgb_light(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -204,7 +205,7 @@ async def test_rgb_light(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -254,7 +255,7 @@ async def test_single_color_mode(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -309,7 +310,7 @@ async def test_state_change_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -427,7 +428,7 @@ async def test_state_brightness_color_effect_temp_change_via_topic(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -459,7 +460,7 @@ async def test_state_brightness_color_effect_temp_change_via_topic(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -626,7 +627,7 @@ async def test_sending_mqtt_commands_and_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -765,7 +766,7 @@ async def test_sending_mqtt_commands_non_optimistic_brightness_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"effect_list": ["rainbow", "colorloop"],
@@ -820,7 +821,7 @@ async def test_effect(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -871,7 +872,7 @@ async def test_flash(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -915,7 +916,7 @@ async def test_transition(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -1113,7 +1114,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: [
{
"name": "Test 1",
@@ -1280,7 +1281,7 @@ async def test_entity_debug_info_message(
) -> None:
"""Test MQTT debug info."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -1304,7 +1305,7 @@ async def test_entity_debug_info_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
light.DOMAIN: {
"schema": "template",
"name": "test",
@@ -1368,7 +1369,7 @@ async def test_publishing_with_custom_encoding(
domain = light.DOMAIN
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG)
if topic == "effect_command_topic":
config[mqtt.DOMAIN][domain]["effect_list"] = ["random", "color_loop"]
config[DOMAIN][domain]["effect_list"] = ["random", "color_loop"]
await help_test_publishing_with_custom_encoding(
hass,
@@ -1411,7 +1412,7 @@ async def test_encoding_subscribable_topics(
init_payload,
) -> None:
"""Test handling of incoming encoded payload."""
config = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][light.DOMAIN])
config = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][light.DOMAIN])
config["state_template"] = "{{ value }}"
await help_test_encoding_subscribable_topics(
hass,
tests/components/mqtt/test_lock.py
+12 -11
View File
@@ -5,7 +5,7 @@ from unittest.mock import patch
import pytest
from homeassistant.components import lock, mqtt
from homeassistant.components import lock
from homeassistant.components.lock import (
SERVICE_LOCK,
SERVICE_OPEN,
@@ -13,6 +13,7 @@ from homeassistant.components.lock import (
LockEntityFeature,
LockState,
)
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.lock import MQTT_LOCK_ATTRIBUTES_BLOCKED
from homeassistant.const import (
ATTR_ASSUMED_STATE,
@@ -58,11 +59,11 @@ from tests.common import async_fire_mqtt_message
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {lock.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
DOMAIN: {lock.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
}
CONFIG_WITH_STATES = {
mqtt.DOMAIN: {
DOMAIN: {
lock.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -324,7 +325,7 @@ async def test_controlling_non_default_state_via_topic_and_json_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lock.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -376,7 +377,7 @@ async def test_sending_mqtt_commands_and_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lock.DOMAIN: {
"name": "test",
"code_format": "^\\d{4}$",
@@ -437,7 +438,7 @@ async def test_sending_mqtt_commands_with_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lock.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -491,7 +492,7 @@ async def test_sending_mqtt_commands_and_explicit_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lock.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -557,7 +558,7 @@ async def test_sending_mqtt_commands_support_open_and_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lock.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -625,7 +626,7 @@ async def test_sending_mqtt_commands_support_open_and_explicit_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lock.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -835,7 +836,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
lock.DOMAIN: [
{
"name": "Test 1",
@@ -1053,7 +1054,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
lock.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][lock.DOMAIN],
DEFAULT_CONFIG[DOMAIN][lock.DOMAIN],
topic,
value,
attribute,
tests/components/mqtt/test_mixins.py
+23 -22
View File
@@ -6,6 +6,7 @@ from unittest.mock import call, patch
import pytest
from homeassistant.components import mqtt, sensor
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.sensor import DEFAULT_NAME as DEFAULT_SENSOR_NAME
from homeassistant.config_entries import ConfigSubentryData
from homeassistant.const import (
@@ -36,7 +37,7 @@ from tests.typing import MqttMockHAClientGenerator
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -104,7 +105,7 @@ async def test_availability_with_shared_state_topic(
[
( # default_entity_name_without_device_name
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"state_topic": "test-topic",
"unique_id": "veryunique",
@@ -119,7 +120,7 @@ async def test_availability_with_shared_state_topic(
),
( # default_entity_name_with_device_name
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"state_topic": "test-topic",
"unique_id": "veryunique",
@@ -134,7 +135,7 @@ async def test_availability_with_shared_state_topic(
),
( # name_follows_device_class
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"state_topic": "test-topic",
"unique_id": "veryunique",
@@ -150,7 +151,7 @@ async def test_availability_with_shared_state_topic(
),
( # name_follows_device_class_without_device_name
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"state_topic": "test-topic",
"unique_id": "veryunique",
@@ -166,7 +167,7 @@ async def test_availability_with_shared_state_topic(
),
( # name_overrides_device_class
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "MySensor",
"state_topic": "test-topic",
@@ -183,7 +184,7 @@ async def test_availability_with_shared_state_topic(
),
( # name_set_no_device_name_set
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "MySensor",
"state_topic": "test-topic",
@@ -200,7 +201,7 @@ async def test_availability_with_shared_state_topic(
),
( # none_entity_name_with_device_name
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": None,
"state_topic": "test-topic",
@@ -217,7 +218,7 @@ async def test_availability_with_shared_state_topic(
),
( # none_entity_name_without_device_name
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": None,
"state_topic": "test-topic",
@@ -234,7 +235,7 @@ async def test_availability_with_shared_state_topic(
),
( # entity_name_and_device_name_the_same
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "Hello world",
"state_topic": "test-topic",
@@ -254,7 +255,7 @@ async def test_availability_with_shared_state_topic(
),
( # entity_name_startswith_device_name1
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "World automation",
"state_topic": "test-topic",
@@ -274,7 +275,7 @@ async def test_availability_with_shared_state_topic(
),
( # entity_name_startswith_device_name2
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "world automation",
"state_topic": "test-topic",
@@ -328,7 +329,7 @@ async def test_default_entity_and_device_name(
await hass.async_block_till_done()
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={mqtt.CONF_BROKER: "mock-broker", mqtt.CONF_PROTOCOL: "5"},
version=mqtt.CONFIG_ENTRY_VERSION,
minor_version=mqtt.CONFIG_ENTRY_MINOR_VERSION,
@@ -417,7 +418,7 @@ async def test_name_attribute_is_set_or_not(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -427,7 +428,7 @@ async def test_name_attribute_is_set_or_not(
}
},
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -439,7 +440,7 @@ async def test_name_attribute_is_set_or_not(
}
},
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -473,7 +474,7 @@ async def test_value_template_fails(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -501,7 +502,7 @@ async def test_registry_enabled_by_default(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -647,7 +648,7 @@ async def test_loading_subentries(
) -> None:
"""Test loading subentries."""
await mqtt_mock_entry()
entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
entry = hass.config_entries.async_entries(DOMAIN)[0]
subentry_id = next(iter(entry.subentries))
# Each subentry has one device
device = device_registry.async_get_device({("mqtt", subentry_id)})
@@ -658,7 +659,7 @@ async def test_loading_subentries(
platform = component["platform"]
entity_id = f"{platform}.{slugify(device.name)}_{slugify(component['name'])}"
entity_entry_entity_id = entity_registry.async_get_entity_id(
platform, mqtt.DOMAIN, f"{subentry_id}_{object_id}"
platform, DOMAIN, f"{subentry_id}_{object_id}"
)
assert entity_entry_entity_id == entity_id
state = hass.states.get(entity_id)
@@ -699,7 +700,7 @@ async def test_loading_subentry_with_bad_component_schema(
) -> None:
"""Test loading subentries."""
await mqtt_mock_entry()
entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
entry = hass.config_entries.async_entries(DOMAIN)[0]
subentry_id = next(iter(entry.subentries))
# Each subentry has one device
device = device_registry.async_get_device({("mqtt", subentry_id)})
@@ -729,7 +730,7 @@ async def test_qos_on_mqtt_device_from_subentry(
) -> None:
"""Test QoS is set correctly on entities from MQTT device."""
mqtt_mock = await mqtt_mock_entry()
entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
entry = hass.config_entries.async_entries(DOMAIN)[0]
subentry_id = next(iter(entry.subentries))
# Each subentry has one device
device = device_registry.async_get_device({("mqtt", subentry_id)})
tests/components/mqtt/test_notify.py
+8 -7
View File
@@ -6,7 +6,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import mqtt, notify
from homeassistant.components import notify
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.notify import ATTR_MESSAGE
from homeassistant.const import ATTR_ENTITY_ID, ATTR_FRIENDLY_NAME, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
@@ -41,7 +42,7 @@ from .common import (
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {notify.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
DOMAIN: {notify.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
}
@@ -50,7 +51,7 @@ DEFAULT_CONFIG = {
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
notify.DOMAIN: {
"command_topic": "command-topic",
"name": "test",
@@ -90,7 +91,7 @@ async def test_sending_mqtt_commands(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
notify.DOMAIN: {
"command_topic": "command-topic",
"command_template": '{ "{{ entity_id }}": "{{ value }}" }',
@@ -231,7 +232,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
notify.DOMAIN: [
{
"name": "Test 1",
@@ -267,8 +268,8 @@ async def test_discovery_update_notify(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered notify."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][notify.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][notify.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][notify.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][notify.DOMAIN])
config1["name"] = "Beer"
config2["name"] = "Milk"
tests/components/mqtt/test_number.py
+28 -27
View File
@@ -6,7 +6,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import mqtt, number
from homeassistant.components import number
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.number import (
CONF_MAX,
CONF_MIN,
@@ -69,7 +70,7 @@ from tests.common import async_fire_mqtt_message, mock_restore_cache_with_extra_
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {number.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
DOMAIN: {number.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
}
@@ -78,7 +79,7 @@ DEFAULT_CONFIG = {
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -95,7 +96,7 @@ DEFAULT_CONFIG = {
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -112,7 +113,7 @@ DEFAULT_CONFIG = {
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -163,7 +164,7 @@ async def test_run_number_setup(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -260,7 +261,7 @@ async def test_native_value_validation(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"name": "test",
"command_topic": "test-topic-cmd",
@@ -316,7 +317,7 @@ async def test_equivalent_unit_of_measurement(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -368,7 +369,7 @@ async def test_value_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"command_topic": "test/number",
"device_class": "temperature",
@@ -406,7 +407,7 @@ async def test_restore_native_value(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"command_topic": "test/number",
"name": "Test Number",
@@ -489,7 +490,7 @@ async def test_run_number_service_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"command_topic": "test/number",
"name": "Test Number",
@@ -572,7 +573,7 @@ async def test_run_number_service_optimistic_with_command_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"command_topic": "test/number/set",
"state_topic": "test/number",
@@ -612,7 +613,7 @@ async def test_run_number_service(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"command_topic": "test/number/set",
"state_topic": "test/number",
@@ -753,7 +754,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: [
{
"name": "Test 1",
@@ -783,7 +784,7 @@ async def test_discovery_removal_number(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test removal of discovered number."""
data = json.dumps(DEFAULT_CONFIG[mqtt.DOMAIN][number.DOMAIN])
data = json.dumps(DEFAULT_CONFIG[DOMAIN][number.DOMAIN])
await help_test_discovery_removal(hass, mqtt_mock_entry, number.DOMAIN, data)
@@ -910,7 +911,7 @@ async def test_entity_debug_info_message(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -927,7 +928,7 @@ async def test_entity_debug_info_message(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -963,7 +964,7 @@ async def test_min_max_step_attributes(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -988,7 +989,7 @@ async def test_invalid_min_max_attributes(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -1013,7 +1014,7 @@ async def test_default_mode(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -1026,7 +1027,7 @@ async def test_default_mode(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -1039,7 +1040,7 @@ async def test_default_mode(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -1069,7 +1070,7 @@ async def test_mode(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -1082,7 +1083,7 @@ async def test_mode(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -1110,7 +1111,7 @@ async def test_invalid_mode(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -1141,7 +1142,7 @@ async def test_mqtt_payload_not_a_number_warning(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
number.DOMAIN: {
"state_topic": "test/state_number",
"command_topic": "test/cmd_number",
@@ -1242,7 +1243,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
number.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][number.DOMAIN],
DEFAULT_CONFIG[DOMAIN][number.DOMAIN],
topic,
value,
attribute,
tests/components/mqtt/test_repairs.py
+9 -9
View File
@@ -7,7 +7,7 @@ from unittest.mock import patch
import pytest
from homeassistant.components import mqtt
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.config_entries import ConfigSubentry, ConfigSubentryData
from homeassistant.const import SERVICE_RELOAD
from homeassistant.core import HomeAssistant
@@ -34,7 +34,7 @@ async def help_setup_yaml(hass: HomeAssistant, config: dict[str, str]) -> None:
return_value=parse_yaml(config["yaml"]),
):
await hass.services.async_call(
mqtt.DOMAIN,
DOMAIN,
SERVICE_RELOAD,
{},
blocking=True,
@@ -80,7 +80,7 @@ async def test_subentry_reconfigure_export_settings(
) -> None:
"""Test the subentry ConfigFlow YAML export with migration to YAML."""
await mqtt_mock_entry()
config_entry: MockConfigEntry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry: MockConfigEntry = hass.config_entries.async_entries(DOMAIN)[0]
subentry_id: str
subentry: ConfigSubentry
subentry_id, subentry = next(iter(config_entry.subentries.items()))
@@ -89,7 +89,7 @@ async def test_subentry_reconfigure_export_settings(
assert result["step_id"] == "summary_menu"
# assert we have a device for the subentry
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
device = device_registry.async_get_device(identifiers={(DOMAIN, subentry_id)})
assert device.config_entries_subentries[config_entry.entry_id] == {subentry_id}
assert device is not None
@@ -135,7 +135,7 @@ async def test_subentry_reconfigure_export_settings(
await setup_helper(hass, suggested_values_from_schema)
# Assert the subentry device was not effected by the exported configs
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
device = device_registry.async_get_device(identifiers={(DOMAIN, subentry_id)})
assert device.config_entries_subentries[config_entry.entry_id] == {subentry_id}
assert device is not None
@@ -145,13 +145,13 @@ async def test_subentry_reconfigure_export_settings(
assert len(events) == 1
issue_id = events[0].data["issue_id"]
issue_registry = ir.async_get(hass)
repair_issue = issue_registry.async_get_issue(mqtt.DOMAIN, issue_id)
repair_issue = issue_registry.async_get_issue(DOMAIN, issue_id)
assert repair_issue.translation_key == translation_key
await async_process_repairs_platforms(hass)
client = await hass_client()
data = await start_repair_fix_flow(client, mqtt.DOMAIN, issue_id)
data = await start_repair_fix_flow(client, DOMAIN, issue_id)
flow_id = data["flow_id"]
assert data["description_placeholders"] == {"name": "Milk notifier"}
@@ -161,7 +161,7 @@ async def test_subentry_reconfigure_export_settings(
assert data["type"] == "create_entry"
# Assert the subentry is removed and no other entity has linked the device
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
device = device_registry.async_get_device(identifiers={(DOMAIN, subentry_id)})
assert device is None
await hass.async_block_till_done(wait_background_tasks=True)
@@ -174,6 +174,6 @@ async def test_subentry_reconfigure_export_settings(
# The MQTT device was now set up from the new source
await hass.async_block_till_done(wait_background_tasks=True)
device = device_registry.async_get_device(identifiers={(mqtt.DOMAIN, subentry_id)})
device = device_registry.async_get_device(identifiers={(DOMAIN, subentry_id)})
assert device.config_entries_subentries[config_entry.entry_id] == {None}
assert device is not None
tests/components/mqtt/test_scene.py
+9 -8
View File
@@ -6,7 +6,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import mqtt, scene
from homeassistant.components import scene
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.const import ATTR_ENTITY_ID, SERVICE_TURN_ON, STATE_UNKNOWN
from homeassistant.core import HomeAssistant, State
@@ -41,7 +42,7 @@ from tests.common import mock_restore_cache
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
scene.DOMAIN: {
"name": "test",
"command_topic": "test-topic",
@@ -55,7 +56,7 @@ DEFAULT_CONFIG = {
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
scene.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -110,7 +111,7 @@ async def test_default_availability_payload(
) -> None:
"""Test availability by default payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
scene.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -134,7 +135,7 @@ async def test_custom_availability_payload(
) -> None:
"""Test availability by custom payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
scene.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -158,7 +159,7 @@ async def test_custom_availability_payload(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
scene.DOMAIN: [
{
"name": "Test 1",
@@ -194,8 +195,8 @@ async def test_discovery_update_payload(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered scene."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][scene.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][scene.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][scene.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][scene.DOMAIN])
config1["name"] = "Beer"
config2["name"] = "Milk"
config1["payload_on"] = "ON"
tests/components/mqtt/test_select.py
+15 -14
View File
@@ -9,7 +9,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import mqtt, select
from homeassistant.components import select
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.select import MQTT_SELECT_ATTRIBUTES_BLOCKED
from homeassistant.components.select import (
ATTR_OPTION,
@@ -57,7 +58,7 @@ from tests.common import async_fire_mqtt_message, mock_restore_cache
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
select.DOMAIN: {
"name": "test",
"command_topic": "test-topic",
@@ -70,7 +71,7 @@ DEFAULT_CONFIG = {
def _test_run_select_setup_params(topic: str) -> Generator[tuple[ConfigType, str]]:
yield (
{
mqtt.DOMAIN: {
DOMAIN: {
select.DOMAIN: {
"state_topic": topic,
"command_topic": "test/select_cmd",
@@ -127,7 +128,7 @@ async def test_run_select_setup(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
select.DOMAIN: {
"state_topic": "test/select_stat",
"command_topic": "test/select_cmd",
@@ -171,7 +172,7 @@ async def test_value_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
select.DOMAIN: {
"command_topic": "test/select_cmd",
"name": "Test Select",
@@ -213,7 +214,7 @@ async def test_run_select_service_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
select.DOMAIN: {
"command_topic": "test/select_cmd",
"name": "Test Select",
@@ -256,7 +257,7 @@ async def test_run_select_service_optimistic_with_command_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
select.DOMAIN: {
"command_topic": "test/select/set",
"state_topic": "test/select",
@@ -297,7 +298,7 @@ async def test_run_select_service(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
select.DOMAIN: {
"command_topic": "test/select/set",
"state_topic": "test/select",
@@ -441,7 +442,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
select.DOMAIN: [
{
"name": "Test 1",
@@ -473,7 +474,7 @@ async def test_discovery_removal_select(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test removal of discovered select."""
data = json.dumps(DEFAULT_CONFIG[mqtt.DOMAIN][select.DOMAIN])
data = json.dumps(DEFAULT_CONFIG[DOMAIN][select.DOMAIN])
await help_test_discovery_removal(hass, mqtt_mock_entry, select.DOMAIN, data)
@@ -605,7 +606,7 @@ def _test_options_attributes_options_config(
for option in request:
yield (
{
mqtt.DOMAIN: {
DOMAIN: {
select.DOMAIN: {
"command_topic": "test/select/set",
"state_topic": "test/select",
@@ -636,7 +637,7 @@ async def test_options_attributes(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
select.DOMAIN: {
"state_topic": "test/select_stat",
"command_topic": "test/select_cmd",
@@ -690,7 +691,7 @@ async def test_publishing_with_custom_encoding(
"""Test publishing MQTT payload with different encoding."""
domain = select.DOMAIN
config = copy.deepcopy(DEFAULT_CONFIG)
config[mqtt.DOMAIN][domain]["options"] = ["milk", "beer"]
config[DOMAIN][domain]["options"] = ["milk", "beer"]
await help_test_publishing_with_custom_encoding(
hass,
@@ -731,7 +732,7 @@ async def test_encoding_subscribable_topics(
attribute_value: Any,
) -> None:
"""Test handling of incoming encoded payload."""
config = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][select.DOMAIN])
config = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][select.DOMAIN])
config["options"] = ["milk", "beer"]
await help_test_encoding_subscribable_topics(
hass,
tests/components/mqtt/test_sensor.py
+37 -36
View File
@@ -13,7 +13,8 @@ from freezegun import freeze_time
from freezegun.api import FrozenDateTimeFactory
import pytest
from homeassistant.components import mqtt, sensor
from homeassistant.components import sensor
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.sensor import MQTT_SENSOR_ATTRIBUTES_BLOCKED
from homeassistant.const import (
ATTR_UNIT_OF_MEASUREMENT,
@@ -84,7 +85,7 @@ from tests.common import (
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {sensor.DOMAIN: {"name": "test", "state_topic": "test-topic"}}
DOMAIN: {sensor.DOMAIN: {"name": "test", "state_topic": "test-topic"}}
}
@@ -92,7 +93,7 @@ DEFAULT_CONFIG = {
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -122,7 +123,7 @@ async def test_setting_sensor_value_via_mqtt_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -164,7 +165,7 @@ async def test_setting_enum_sensor_value_via_mqtt_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -200,7 +201,7 @@ async def test_handling_undecoded_sensor_value(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -363,7 +364,7 @@ async def test_setting_sensor_native_value_handling_via_mqtt_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -420,7 +421,7 @@ async def test_setting_numeric_sensor_native_value_handling_via_mqtt_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -454,7 +455,7 @@ async def test_setting_sensor_value_expires_availability_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -550,7 +551,7 @@ async def expires_helper(hass: HomeAssistant) -> None:
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -582,7 +583,7 @@ async def test_setting_sensor_value_via_mqtt_json_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -617,7 +618,7 @@ async def test_setting_sensor_value_via_mqtt_json_message_and_default_current_st
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_class": "total",
@@ -667,7 +668,7 @@ async def test_setting_sensor_last_reset_via_mqtt_json_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_class": "total",
@@ -705,7 +706,7 @@ async def test_setting_sensor_last_reset_via_mqtt_json_message_2(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -742,7 +743,7 @@ async def test_force_update_disabled(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -865,7 +866,7 @@ async def test_discovery_update_availability(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -887,7 +888,7 @@ async def test_invalid_device_class(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -912,7 +913,7 @@ async def test_invalid_unit_of_measurement(
)
# A repair issue was logged for the failing YAML config
assert len(events) == 1
assert events[0].data["domain"] == mqtt.DOMAIN
assert events[0].data["domain"] == DOMAIN
# Assert the sensor is not created
state = hass.states.get("sensor.test")
assert state is None
@@ -923,7 +924,7 @@ async def test_invalid_unit_of_measurement(
[
pytest.param(
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -939,7 +940,7 @@ async def test_invalid_unit_of_measurement(
),
pytest.param(
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -955,7 +956,7 @@ async def test_invalid_unit_of_measurement(
),
pytest.param(
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -971,7 +972,7 @@ async def test_invalid_unit_of_measurement(
),
pytest.param(
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -987,7 +988,7 @@ async def test_invalid_unit_of_measurement(
),
pytest.param(
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -1003,7 +1004,7 @@ async def test_invalid_unit_of_measurement(
),
pytest.param(
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -1061,7 +1062,7 @@ async def test_device_class_with_equivalent_unit_of_measurement_received(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -1115,7 +1116,7 @@ async def test_equivalent_unit_of_measurement_received_without_device_class(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: [
{
"name": "Test 1",
@@ -1191,7 +1192,7 @@ async def test_valid_device_class_and_uom(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -1213,7 +1214,7 @@ async def test_invalid_state_class(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -1240,7 +1241,7 @@ async def test_invalid_state_class_with_unit_of_measurement(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -1254,7 +1255,7 @@ async def test_invalid_state_class_with_unit_of_measurement(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -1268,7 +1269,7 @@ async def test_invalid_state_class_with_unit_of_measurement(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -1294,7 +1295,7 @@ async def test_invalid_options_config(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: [
{
"name": "Test 1",
@@ -1402,7 +1403,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: [
{
"name": "Test 1",
@@ -1678,7 +1679,7 @@ async def test_entity_category(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
sensor.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -1838,7 +1839,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
sensor.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][sensor.DOMAIN],
DEFAULT_CONFIG[DOMAIN][sensor.DOMAIN],
topic,
value,
attribute,
@@ -2023,7 +2024,7 @@ async def test_value_incorrect_state_class_config(
)
caplog.clear()
config_payload = hass_config[mqtt.DOMAIN][sensor.DOMAIN][0]
config_payload = hass_config[DOMAIN][sensor.DOMAIN][0]
async_fire_mqtt_message(
hass, "homeassistant/sensor/bla/config", json.dumps(config_payload)
)
tests/components/mqtt/test_siren.py
+19 -18
View File
@@ -6,7 +6,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import mqtt, siren
from homeassistant.components import siren
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.siren import ATTR_VOLUME_LEVEL
from homeassistant.const import (
ATTR_ASSUMED_STATE,
@@ -55,7 +56,7 @@ from tests.common import async_fire_mqtt_message
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {siren.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
DOMAIN: {siren.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
}
@@ -82,7 +83,7 @@ async def async_turn_off(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
siren.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -119,7 +120,7 @@ async def test_controlling_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
siren.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -164,7 +165,7 @@ async def test_sending_mqtt_commands_and_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
siren.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -205,7 +206,7 @@ async def test_controlling_state_via_topic_and_json_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
siren.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -292,7 +293,7 @@ async def test_controlling_state_and_attributes_with_json_message_without_templa
help_custom_config(
siren.DOMAIN,
{
mqtt.DOMAIN: {
DOMAIN: {
siren.DOMAIN: {
"command_topic": "command-topic",
}
@@ -387,7 +388,7 @@ async def test_filtering_not_supported_attributes_optimistic(
help_custom_config(
siren.DOMAIN,
{
mqtt.DOMAIN: {
DOMAIN: {
siren.DOMAIN: {
"command_topic": "command-topic",
}
@@ -495,7 +496,7 @@ async def test_default_availability_payload(
) -> None:
"""Test availability by default payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
siren.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -521,7 +522,7 @@ async def test_custom_availability_payload(
) -> None:
"""Test availability by custom payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
siren.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -547,7 +548,7 @@ async def test_custom_availability_payload(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
siren.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -644,7 +645,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
siren.DOMAIN: [
{
"name": "Test 1",
@@ -686,8 +687,8 @@ async def test_discovery_update_siren_topic_template(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered siren."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][siren.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][siren.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][siren.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][siren.DOMAIN])
config1["name"] = "Beer"
config2["name"] = "Milk"
config1["state_topic"] = "siren/state1"
@@ -722,8 +723,8 @@ async def test_discovery_update_siren_template(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered siren."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][siren.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][siren.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][siren.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][siren.DOMAIN])
config1["name"] = "Beer"
config2["name"] = "Milk"
config1["state_topic"] = "siren/state1"
@@ -987,7 +988,7 @@ async def test_publishing_with_custom_encoding(
"""Test publishing MQTT payload with command templates and different encoding."""
domain = siren.DOMAIN
config: dict[str, Any] = copy.deepcopy(DEFAULT_CONFIG)
config[mqtt.DOMAIN][domain][siren.ATTR_AVAILABLE_TONES] = ["siren", "xylophone"]
config[DOMAIN][domain][siren.ATTR_AVAILABLE_TONES] = ["siren", "xylophone"]
await help_test_publishing_with_custom_encoding(
hass,
@@ -1029,7 +1030,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
siren.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][siren.DOMAIN],
DEFAULT_CONFIG[DOMAIN][siren.DOMAIN],
topic,
value,
attribute,
tests/components/mqtt/test_switch.py
+18 -17
View File
@@ -6,7 +6,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import mqtt, switch
from homeassistant.components import switch
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.const import (
ATTR_ASSUMED_STATE,
ATTR_DEVICE_CLASS,
@@ -52,7 +53,7 @@ from tests.components.switch import common
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {switch.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
DOMAIN: {switch.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
}
@@ -61,7 +62,7 @@ DEFAULT_CONFIG = {
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
switch.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -76,7 +77,7 @@ DEFAULT_CONFIG = {
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
switch.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -124,7 +125,7 @@ async def test_controlling_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
switch.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -171,7 +172,7 @@ async def test_sending_mqtt_commands_and_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
switch.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -195,7 +196,7 @@ async def test_sending_inital_state_and_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
switch.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -239,7 +240,7 @@ async def test_sending_mqtt_commands_with_command_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
switch.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -302,7 +303,7 @@ async def test_default_availability_payload(
) -> None:
"""Test availability by default payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
switch.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -328,7 +329,7 @@ async def test_custom_availability_payload(
) -> None:
"""Test availability by custom payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
switch.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -354,7 +355,7 @@ async def test_custom_availability_payload(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
switch.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -451,7 +452,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
switch.DOMAIN: [
{
"name": "Test 1",
@@ -493,8 +494,8 @@ async def test_discovery_update_switch_topic_template(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered switch."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][switch.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][switch.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][switch.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][switch.DOMAIN])
config1["name"] = "Beer"
config2["name"] = "Milk"
config1["state_topic"] = "switch/state1"
@@ -529,8 +530,8 @@ async def test_discovery_update_switch_template(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test update of discovered switch."""
config1 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][switch.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][switch.DOMAIN])
config1 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][switch.DOMAIN])
config2 = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][switch.DOMAIN])
config1["name"] = "Beer"
config2["name"] = "Milk"
config1["state_topic"] = "switch/state1"
@@ -737,7 +738,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
switch.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][switch.DOMAIN],
DEFAULT_CONFIG[DOMAIN][switch.DOMAIN],
topic,
value,
attribute,
tests/components/mqtt/test_text.py
+15 -14
View File
@@ -5,7 +5,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import mqtt, text
from homeassistant.components import text
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.const import ATTR_ASSUMED_STATE, ATTR_ENTITY_ID, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
@@ -44,7 +45,7 @@ from tests.common import async_fire_mqtt_message
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {text.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
DOMAIN: {text.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
}
@@ -64,7 +65,7 @@ async def async_set_value(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
text.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -106,7 +107,7 @@ async def test_controlling_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
text.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -163,7 +164,7 @@ async def test_forced_text_length(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
text.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -237,7 +238,7 @@ async def test_controlling_validation_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
text.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -260,7 +261,7 @@ async def test_attribute_validation_max_greater_then_min(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
text.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -283,7 +284,7 @@ async def test_attribute_validation_max_not_greater_then_max_state_length(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
text.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -313,7 +314,7 @@ async def test_validation_payload_greater_then_max_state_length(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
text.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -356,7 +357,7 @@ async def test_sending_mqtt_commands_and_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
text.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -421,7 +422,7 @@ async def test_default_availability_payload(
) -> None:
"""Test availability by default payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
text.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -439,7 +440,7 @@ async def test_custom_availability_payload(
) -> None:
"""Test availability by custom payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
text.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -515,7 +516,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
text.DOMAIN: [
{
"name": "Test 1",
@@ -758,7 +759,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
text.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][text.DOMAIN],
DEFAULT_CONFIG[DOMAIN][text.DOMAIN],
topic,
value,
attribute,
tests/components/mqtt/test_time.py
+10 -9
View File
@@ -6,7 +6,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import mqtt, time
from homeassistant.components import time
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.const import ATTR_ASSUMED_STATE, ATTR_ENTITY_ID, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
@@ -44,7 +45,7 @@ from tests.common import async_fire_mqtt_message
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {time.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
DOMAIN: {time.DOMAIN: {"name": "test", "command_topic": "test-topic"}}
}
@@ -64,7 +65,7 @@ async def async_set_value(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
time.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -116,7 +117,7 @@ async def test_controlling_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
time.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -155,7 +156,7 @@ async def test_controlling_validation_state_via_topic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
time.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -221,7 +222,7 @@ async def test_default_availability_payload(
) -> None:
"""Test availability by default payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
time.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -239,7 +240,7 @@ async def test_custom_availability_payload(
) -> None:
"""Test availability by custom payload with defined topic."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
time.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -315,7 +316,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
time.DOMAIN: [
{
"name": "Test 1",
@@ -494,7 +495,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
time.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][time.DOMAIN],
DEFAULT_CONFIG[DOMAIN][time.DOMAIN],
topic,
value,
attribute,
tests/components/mqtt/test_update.py
+16 -15
View File
@@ -6,7 +6,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import mqtt, update
from homeassistant.components import update
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.update import DOMAIN as UPDATE_DOMAIN, SERVICE_INSTALL
from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON, STATE_UNKNOWN
from homeassistant.core import HomeAssistant
@@ -42,7 +43,7 @@ from tests.common import async_fire_mqtt_message
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: {
"name": "test",
"state_topic": "test-topic",
@@ -59,7 +60,7 @@ DEFAULT_CONFIG = {
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: {
"state_topic": "test/installed-version",
"latest_version_topic": "test/latest-version",
@@ -76,7 +77,7 @@ DEFAULT_CONFIG = {
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: {
"state_topic": "test/installed-version",
"latest_version_topic": "test/latest-version",
@@ -132,7 +133,7 @@ async def test_run_update_setup(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: {
"state_topic": "test/installed-version",
"latest_version_topic": "test/latest-version",
@@ -182,7 +183,7 @@ async def test_run_update_setup_float(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: {
"state_topic": "test/installed-version",
"value_template": "{{ value_json.installed }}",
@@ -227,7 +228,7 @@ async def test_value_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: {
"state_topic": "test/update",
"value_template": (
@@ -298,7 +299,7 @@ async def test_errornous_value_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: {
"state_topic": "test/installed-version",
"value_template": "{{ value_json.installed }}",
@@ -343,7 +344,7 @@ async def test_value_template_float(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: {
"state_topic": "test/state-topic",
"name": "Test Update",
@@ -371,7 +372,7 @@ async def test_empty_json_state_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: {
"state_topic": "test/state-topic",
"name": "Test Update",
@@ -424,7 +425,7 @@ async def test_invalid_json_state_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: {
"state_topic": "test/state-topic",
"name": "Test Update",
@@ -520,7 +521,7 @@ async def test_json_state_message(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: {
"state_topic": "test/state-topic",
"value_template": '{{ {"installed_version": value_json.installed, '
@@ -561,7 +562,7 @@ async def test_json_state_message_with_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: {
"state_topic": "test/installed-version",
"latest_version_topic": "test/latest-version",
@@ -695,7 +696,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
update.DOMAIN: [
{
"name": "Bear",
@@ -725,7 +726,7 @@ async def test_discovery_removal_update(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test removal of discovered update."""
data = json.dumps(DEFAULT_CONFIG[mqtt.DOMAIN][update.DOMAIN])
data = json.dumps(DEFAULT_CONFIG[DOMAIN][update.DOMAIN])
await help_test_discovery_removal(hass, mqtt_mock_entry, update.DOMAIN, data)
tests/components/mqtt/test_util.py
+7 -6
View File
@@ -11,6 +11,7 @@ from unittest.mock import MagicMock, patch
import pytest
from homeassistant.components import mqtt
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.models import MessageCallbackType
from homeassistant.components.mqtt.util import EnsureJobAfterCooldown
from homeassistant.config_entries import ConfigEntryDisabler, ConfigEntryState
@@ -222,7 +223,7 @@ async def test_waiting_for_client_not_loaded(
await hass.async_block_till_done()
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={"broker": "test-broker"},
state=ConfigEntryState.NOT_LOADED,
version=mqtt.CONFIG_ENTRY_VERSION,
@@ -263,7 +264,7 @@ async def test_waiting_for_client_loaded(
assert await mqtt.async_wait_for_mqtt_client(hass)
unsub = await mqtt.async_subscribe(hass, "test_topic", lambda msg: None)
entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
entry = hass.config_entries.async_entries(DOMAIN)[0]
assert entry.state is ConfigEntryState.LOADED
await _async_just_in_time_subscribe()
@@ -281,7 +282,7 @@ async def test_waiting_for_client_entry_fails(
await hass.async_block_till_done()
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={"broker": "test-broker"},
state=ConfigEntryState.NOT_LOADED,
version=mqtt.CONFIG_ENTRY_VERSION,
@@ -311,7 +312,7 @@ async def test_waiting_for_client_setup_fails(
await hass.async_block_till_done()
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={"broker": "test-broker"},
state=ConfigEntryState.NOT_LOADED,
version=mqtt.CONFIG_ENTRY_VERSION,
@@ -340,7 +341,7 @@ async def test_waiting_for_client_timeout(
await hass.async_block_till_done()
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={"broker": "test-broker"},
state=ConfigEntryState.NOT_LOADED,
version=mqtt.CONFIG_ENTRY_VERSION,
@@ -361,7 +362,7 @@ async def test_waiting_for_client_with_disabled_entry(
await hass.async_block_till_done()
entry = MockConfigEntry(
domain=mqtt.DOMAIN,
domain=DOMAIN,
data={"broker": "test-broker"},
state=ConfigEntryState.NOT_LOADED,
version=mqtt.CONFIG_ENTRY_VERSION,
tests/components/mqtt/test_vacuum.py
+16 -12
View File
@@ -7,9 +7,13 @@ from unittest.mock import call, patch
import pytest
from homeassistant.components import mqtt, vacuum
from homeassistant.components import vacuum
from homeassistant.components.mqtt import vacuum as mqttvacuum
from homeassistant.components.mqtt.const import CONF_COMMAND_TOPIC, CONF_STATE_TOPIC
from homeassistant.components.mqtt.const import (
CONF_COMMAND_TOPIC,
CONF_STATE_TOPIC,
DOMAIN,
)
from homeassistant.components.mqtt.vacuum import (
ALL_SERVICES,
MQTT_VACUUM_ATTRIBUTES_BLOCKED,
@@ -80,7 +84,7 @@ SEND_COMMAND_TOPIC = "vacuum/send_command"
STATE_TOPIC = "vacuum/state"
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
vacuum.DOMAIN: {
CONF_NAME: "mqtttest",
CONF_COMMAND_TOPIC: COMMAND_TOPIC,
@@ -93,7 +97,7 @@ DEFAULT_CONFIG = {
}
CONFIG_CLEAN_SEGMENTS = {
mqtt.DOMAIN: {
DOMAIN: {
vacuum.DOMAIN: {
CONF_NAME: "test",
CONF_STATE_TOPIC: STATE_TOPIC,
@@ -103,7 +107,7 @@ CONFIG_CLEAN_SEGMENTS = {
}
}
DEFAULT_CONFIG_2 = {mqtt.DOMAIN: {vacuum.DOMAIN: {"name": "test"}}}
DEFAULT_CONFIG_2 = {DOMAIN: {vacuum.DOMAIN: {"name": "test"}}}
CONFIG_ALL_SERVICES = help_custom_config(
vacuum.DOMAIN,
@@ -363,10 +367,10 @@ async def test_clean_segments_command(
mqtt_mock_entry: MqttMockHAClientGenerator,
) -> None:
"""Test cleaning segments and repair flow."""
config_entry = hass.config_entries.async_entries(mqtt.DOMAIN)[0]
config_entry = hass.config_entries.async_entries(DOMAIN)[0]
entity_registry.async_get_or_create(
vacuum.DOMAIN,
mqtt.DOMAIN,
DOMAIN,
"veryunique",
config_entry=config_entry,
suggested_object_id="test",
@@ -549,7 +553,7 @@ async def test_removing_clean_segments_command_topic_resets_feature(
"""
await mqtt_mock_entry()
config_with_clean_segments_command_topic = CONFIG_CLEAN_SEGMENTS[mqtt.DOMAIN][
config_with_clean_segments_command_topic = CONFIG_CLEAN_SEGMENTS[DOMAIN][
vacuum.DOMAIN
]
async_fire_mqtt_message(
@@ -617,7 +621,7 @@ async def test_clean_area_feature_preserved_on_config_update(
"""
await mqtt_mock_entry()
config = CONFIG_CLEAN_SEGMENTS[mqtt.DOMAIN][vacuum.DOMAIN]
config = CONFIG_CLEAN_SEGMENTS[DOMAIN][vacuum.DOMAIN]
async_fire_mqtt_message(
hass,
"homeassistant/vacuum/bla/config",
@@ -865,7 +869,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
vacuum.DOMAIN: [
{
"name": "Test 1",
@@ -1036,7 +1040,7 @@ async def test_publishing_with_custom_encoding(
"""Test publishing MQTT payload with different encoding."""
domain = vacuum.DOMAIN
config = deepcopy(DEFAULT_CONFIG)
config[mqtt.DOMAIN][domain]["supported_features"] = [
config[DOMAIN][domain]["supported_features"] = [
"clean_spot",
"fan_speed",
"locate",
@@ -1101,7 +1105,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
vacuum.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][vacuum.DOMAIN],
DEFAULT_CONFIG[DOMAIN][vacuum.DOMAIN],
topic,
value,
attribute,
tests/components/mqtt/test_valve.py
+27 -26
View File
@@ -5,7 +5,8 @@ from unittest.mock import patch
import pytest
from homeassistant.components import mqtt, valve
from homeassistant.components import valve
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.valve import (
MQTT_VALVE_ATTRIBUTES_BLOCKED,
ValveEntityFeature,
@@ -62,7 +63,7 @@ from tests.common import async_fire_mqtt_message
from tests.typing import MqttMockHAClientGenerator, MqttMockPahoClient
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"command_topic": "command-topic",
"state_topic": "test-topic",
@@ -72,7 +73,7 @@ DEFAULT_CONFIG = {
}
DEFAULT_CONFIG_REPORTS_POSITION = {
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -87,7 +88,7 @@ DEFAULT_CONFIG_REPORTS_POSITION = {
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -138,7 +139,7 @@ async def test_state_via_state_topic_no_position(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -181,7 +182,7 @@ async def test_state_via_state_topic_with_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -227,7 +228,7 @@ async def test_state_via_state_topic_with_position_template(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -292,7 +293,7 @@ async def test_state_via_state_topic_through_position(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -347,7 +348,7 @@ async def test_opening_closing_state_is_reset(
[
(
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -361,7 +362,7 @@ async def test_opening_closing_state_is_reset(
),
(
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -402,7 +403,7 @@ async def test_invalid_state_updates(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -464,7 +465,7 @@ async def test_state_via_state_trough_position_with_alt_range(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -611,7 +612,7 @@ async def test_open_close_payload_config_not_allowed(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -622,7 +623,7 @@ async def test_open_close_payload_config_not_allowed(
}
},
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -671,7 +672,7 @@ async def test_controlling_valve_by_state_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -722,7 +723,7 @@ async def test_controlling_valve_by_position(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -769,7 +770,7 @@ async def test_controlling_valve_by_set_valve_position(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -824,7 +825,7 @@ async def test_controlling_valve_optimistic_by_set_valve_position(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -873,7 +874,7 @@ async def test_controlling_valve_with_alt_range_by_set_valve_position(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -924,7 +925,7 @@ async def test_controlling_valve_with_alt_range_by_position(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -936,7 +937,7 @@ async def test_controlling_valve_with_alt_range_by_position(
}
},
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"command_topic": "command-topic",
@@ -989,7 +990,7 @@ async def test_controlling_valve_by_position_optimistic(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"state_topic": "state-topic",
@@ -1085,7 +1086,7 @@ async def test_custom_availability_payload(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"device_class": "water",
@@ -1109,7 +1110,7 @@ async def test_valid_device_class(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: {
"name": "test",
"device_class": "abc123",
@@ -1193,7 +1194,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
valve.DOMAIN: [
{
"name": "Test 1",
@@ -1396,7 +1397,7 @@ async def test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
valve.DOMAIN,
DEFAULT_CONFIG[mqtt.DOMAIN][valve.DOMAIN],
DEFAULT_CONFIG[DOMAIN][valve.DOMAIN],
topic,
value,
attribute,
tests/components/mqtt/test_water_heater.py
+10 -9
View File
@@ -8,7 +8,8 @@ from unittest.mock import call, patch
import pytest
import voluptuous as vol
from homeassistant.components import mqtt, water_heater
from homeassistant.components import water_heater
from homeassistant.components.mqtt.const import DOMAIN
from homeassistant.components.mqtt.water_heater import (
MQTT_WATER_HEATER_ATTRIBUTES_BLOCKED,
)
@@ -96,7 +97,7 @@ _DEFAULT_MAX_TEMP_CELSIUS = round(
DEFAULT_CONFIG = {
mqtt.DOMAIN: {
DOMAIN: {
water_heater.DOMAIN: {
"name": "test",
"mode_command_topic": "mode-topic",
@@ -553,7 +554,7 @@ async def test_custom_availability_payload(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
water_heater.DOMAIN: {
"name": "test",
"mode_command_topic": "mode-topic",
@@ -618,7 +619,7 @@ async def test_get_with_templates(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
water_heater.DOMAIN: {
"name": "test",
"mode_command_topic": "mode-topic",
@@ -867,7 +868,7 @@ async def test_discovery_update_attr(
"hass_config",
[
{
mqtt.DOMAIN: {
DOMAIN: {
water_heater.DOMAIN: [
{
"name": "Test 1",
@@ -910,7 +911,7 @@ async def test_encoding_subscribable_topics(
attribute_value: Any,
) -> None:
"""Test handling of incoming encoded payload."""
config = copy.deepcopy(DEFAULT_CONFIG[mqtt.DOMAIN][water_heater.DOMAIN])
config = copy.deepcopy(DEFAULT_CONFIG[DOMAIN][water_heater.DOMAIN])
await help_test_encoding_subscribable_topics(
hass,
mqtt_mock_entry,
@@ -927,7 +928,7 @@ async def test_discovery_removal_water_heater(
hass: HomeAssistant, mqtt_mock_entry: MqttMockHAClientGenerator
) -> None:
"""Test removal of discovered water heater."""
data = json.dumps(DEFAULT_CONFIG[mqtt.DOMAIN][water_heater.DOMAIN])
data = json.dumps(DEFAULT_CONFIG[DOMAIN][water_heater.DOMAIN])
await help_test_discovery_removal(hass, mqtt_mock_entry, water_heater.DOMAIN, data)
@@ -1008,7 +1009,7 @@ async def test_entity_id_update_subscriptions(
) -> None:
"""Test MQTT subscriptions are managed when entity_id is updated."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
water_heater.DOMAIN: {
"name": "test",
"mode_state_topic": "test-topic",
@@ -1035,7 +1036,7 @@ async def test_entity_debug_info_message(
) -> None:
"""Test MQTT debug info."""
config = {
mqtt.DOMAIN: {
DOMAIN: {
water_heater.DOMAIN: {
"name": "test",
"mode_command_topic": "command-topic",