52561ce076
* New testframework and tests for fan platform * Merge test_common_new to test_common * Add alarm_control_panel * Add binary_sensor * Add button * Add camera * Add climate * Add config_flow * Add cover * Add device_tracker_disovery * Add device_trigger * Add diagnostics * Add discovery * Add humidifier * Add init * Add lecacy_vacuum * Add light_json * Add light_template * Add light * Add lock * Add number * Add scene * Add select * Add sensor * Add siren * Add state_vacuum * Add subscription * Add switch * Add tag * Add trigger * Add missed tests * Add another missed test * Add device_tracker * Remove commented out code * Correct tests according comments * Improve mqtt_mock_entry and recover tests * Split fixtures with and without yaml setup * Update fixtures manual_mqtt * Update fixtures mqtt_json * Fix test tasmota * Update fixture mqtt_room * Revert fixture changes, improve test * re-add test
208 lines
6 KiB
Python
208 lines
6 KiB
Python
"""The tests for the MQTT subscription component."""
|
|
from unittest.mock import ANY
|
|
|
|
from homeassistant.components.mqtt.subscription import (
|
|
async_prepare_subscribe_topics,
|
|
async_subscribe_topics,
|
|
async_unsubscribe_topics,
|
|
)
|
|
from homeassistant.core import callback
|
|
|
|
from tests.common import async_fire_mqtt_message
|
|
|
|
|
|
async def test_subscribe_topics(hass, mqtt_mock_entry_no_yaml_config, caplog):
|
|
"""Test subscription to topics."""
|
|
await mqtt_mock_entry_no_yaml_config()
|
|
calls1 = []
|
|
|
|
@callback
|
|
def record_calls1(*args):
|
|
"""Record calls."""
|
|
calls1.append(args)
|
|
|
|
calls2 = []
|
|
|
|
@callback
|
|
def record_calls2(*args):
|
|
"""Record calls."""
|
|
calls2.append(args)
|
|
|
|
sub_state = None
|
|
sub_state = async_prepare_subscribe_topics(
|
|
hass,
|
|
sub_state,
|
|
{
|
|
"test_topic1": {"topic": "test-topic1", "msg_callback": record_calls1},
|
|
"test_topic2": {"topic": "test-topic2", "msg_callback": record_calls2},
|
|
},
|
|
)
|
|
await async_subscribe_topics(hass, sub_state)
|
|
|
|
async_fire_mqtt_message(hass, "test-topic1", "test-payload1")
|
|
assert len(calls1) == 1
|
|
assert calls1[0][0].topic == "test-topic1"
|
|
assert calls1[0][0].payload == "test-payload1"
|
|
assert len(calls2) == 0
|
|
|
|
async_fire_mqtt_message(hass, "test-topic2", "test-payload2")
|
|
assert len(calls1) == 1
|
|
assert len(calls2) == 1
|
|
assert calls2[0][0].topic == "test-topic2"
|
|
assert calls2[0][0].payload == "test-payload2"
|
|
|
|
async_unsubscribe_topics(hass, sub_state)
|
|
|
|
async_fire_mqtt_message(hass, "test-topic1", "test-payload")
|
|
async_fire_mqtt_message(hass, "test-topic2", "test-payload")
|
|
|
|
assert len(calls1) == 1
|
|
assert len(calls2) == 1
|
|
|
|
|
|
async def test_modify_topics(hass, mqtt_mock_entry_no_yaml_config, caplog):
|
|
"""Test modification of topics."""
|
|
await mqtt_mock_entry_no_yaml_config()
|
|
calls1 = []
|
|
|
|
@callback
|
|
def record_calls1(*args):
|
|
"""Record calls."""
|
|
calls1.append(args)
|
|
|
|
calls2 = []
|
|
|
|
@callback
|
|
def record_calls2(*args):
|
|
"""Record calls."""
|
|
calls2.append(args)
|
|
|
|
sub_state = None
|
|
sub_state = async_prepare_subscribe_topics(
|
|
hass,
|
|
sub_state,
|
|
{
|
|
"test_topic1": {"topic": "test-topic1", "msg_callback": record_calls1},
|
|
"test_topic2": {"topic": "test-topic2", "msg_callback": record_calls2},
|
|
},
|
|
)
|
|
await async_subscribe_topics(hass, sub_state)
|
|
|
|
async_fire_mqtt_message(hass, "test-topic1", "test-payload")
|
|
assert len(calls1) == 1
|
|
assert len(calls2) == 0
|
|
|
|
async_fire_mqtt_message(hass, "test-topic2", "test-payload")
|
|
assert len(calls1) == 1
|
|
assert len(calls2) == 1
|
|
|
|
sub_state = async_prepare_subscribe_topics(
|
|
hass,
|
|
sub_state,
|
|
{"test_topic1": {"topic": "test-topic1_1", "msg_callback": record_calls1}},
|
|
)
|
|
await async_subscribe_topics(hass, sub_state)
|
|
|
|
async_fire_mqtt_message(hass, "test-topic1", "test-payload")
|
|
async_fire_mqtt_message(hass, "test-topic2", "test-payload")
|
|
assert len(calls1) == 1
|
|
assert len(calls2) == 1
|
|
|
|
async_fire_mqtt_message(hass, "test-topic1_1", "test-payload")
|
|
assert len(calls1) == 2
|
|
assert calls1[1][0].topic == "test-topic1_1"
|
|
assert calls1[1][0].payload == "test-payload"
|
|
assert len(calls2) == 1
|
|
|
|
async_unsubscribe_topics(hass, sub_state)
|
|
|
|
async_fire_mqtt_message(hass, "test-topic1_1", "test-payload")
|
|
async_fire_mqtt_message(hass, "test-topic2", "test-payload")
|
|
|
|
assert len(calls1) == 2
|
|
assert len(calls2) == 1
|
|
|
|
|
|
async def test_qos_encoding_default(hass, mqtt_mock_entry_no_yaml_config, caplog):
|
|
"""Test default qos and encoding."""
|
|
mqtt_mock = await mqtt_mock_entry_no_yaml_config()
|
|
|
|
@callback
|
|
def msg_callback(*args):
|
|
"""Do nothing."""
|
|
pass
|
|
|
|
sub_state = None
|
|
sub_state = async_prepare_subscribe_topics(
|
|
hass,
|
|
sub_state,
|
|
{"test_topic1": {"topic": "test-topic1", "msg_callback": msg_callback}},
|
|
)
|
|
await async_subscribe_topics(hass, sub_state)
|
|
mqtt_mock.async_subscribe.assert_called_with("test-topic1", ANY, 0, "utf-8")
|
|
|
|
|
|
async def test_qos_encoding_custom(hass, mqtt_mock_entry_no_yaml_config, caplog):
|
|
"""Test custom qos and encoding."""
|
|
mqtt_mock = await mqtt_mock_entry_no_yaml_config()
|
|
|
|
@callback
|
|
def msg_callback(*args):
|
|
"""Do nothing."""
|
|
pass
|
|
|
|
sub_state = None
|
|
sub_state = async_prepare_subscribe_topics(
|
|
hass,
|
|
sub_state,
|
|
{
|
|
"test_topic1": {
|
|
"topic": "test-topic1",
|
|
"msg_callback": msg_callback,
|
|
"qos": 1,
|
|
"encoding": "utf-16",
|
|
}
|
|
},
|
|
)
|
|
await async_subscribe_topics(hass, sub_state)
|
|
mqtt_mock.async_subscribe.assert_called_with("test-topic1", ANY, 1, "utf-16")
|
|
|
|
|
|
async def test_no_change(hass, mqtt_mock_entry_no_yaml_config, caplog):
|
|
"""Test subscription to topics without change."""
|
|
mqtt_mock = await mqtt_mock_entry_no_yaml_config()
|
|
|
|
calls = []
|
|
|
|
@callback
|
|
def record_calls(*args):
|
|
"""Record calls."""
|
|
calls.append(args)
|
|
|
|
sub_state = None
|
|
sub_state = async_prepare_subscribe_topics(
|
|
hass,
|
|
sub_state,
|
|
{"test_topic1": {"topic": "test-topic1", "msg_callback": record_calls}},
|
|
)
|
|
await async_subscribe_topics(hass, sub_state)
|
|
subscribe_call_count = mqtt_mock.async_subscribe.call_count
|
|
|
|
async_fire_mqtt_message(hass, "test-topic1", "test-payload")
|
|
assert len(calls) == 1
|
|
|
|
sub_state = async_prepare_subscribe_topics(
|
|
hass,
|
|
sub_state,
|
|
{"test_topic1": {"topic": "test-topic1", "msg_callback": record_calls}},
|
|
)
|
|
await async_subscribe_topics(hass, sub_state)
|
|
assert subscribe_call_count == mqtt_mock.async_subscribe.call_count
|
|
|
|
async_fire_mqtt_message(hass, "test-topic1", "test-payload")
|
|
assert len(calls) == 2
|
|
|
|
async_unsubscribe_topics(hass, sub_state)
|
|
|
|
async_fire_mqtt_message(hass, "test-topic1", "test-payload")
|
|
assert len(calls) == 2
|