From 5e6f5fca90cc9b050fcd010c56aa540f6097f594 Mon Sep 17 00:00:00 2001 From: Erik Montnemery Date: Fri, 4 Dec 2020 04:39:49 +0100 Subject: [PATCH] Don't send MQTT birth message in tests (#43917) --- tests/components/mqtt/test_init.py | 12 ++++++++++++ tests/conftest.py | 2 +- 2 files changed, 13 insertions(+), 1 deletion(-) diff --git a/tests/components/mqtt/test_init.py b/tests/components/mqtt/test_init.py index 86b905f2b0f..82a88de918b 100644 --- a/tests/components/mqtt/test_init.py +++ b/tests/components/mqtt/test_init.py @@ -781,6 +781,18 @@ async def test_custom_birth_message(hass, mqtt_client_mock, mqtt_mock): mqtt_client_mock.publish.assert_called_with("birth", "birth", 0, False) +@pytest.mark.parametrize( + "mqtt_config", + [ + { + mqtt.CONF_BROKER: "mock-broker", + mqtt.CONF_BIRTH_MESSAGE: { + mqtt.ATTR_TOPIC: "homeassistant/status", + mqtt.ATTR_PAYLOAD: "online", + }, + } + ], +) async def test_default_birth_message(hass, mqtt_client_mock, mqtt_mock): """Test sending birth message.""" birth = asyncio.Event() diff --git a/tests/conftest.py b/tests/conftest.py index fa390f9bf3b..d8fb9f2914b 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -400,7 +400,7 @@ def mqtt_client_mock(hass): async def mqtt_mock(hass, mqtt_client_mock, mqtt_config): """Fixture to mock MQTT component.""" if mqtt_config is None: - mqtt_config = {mqtt.CONF_BROKER: "mock-broker"} + mqtt_config = {mqtt.CONF_BROKER: "mock-broker", mqtt.CONF_BIRTH_MESSAGE: {}} result = await async_setup_component(hass, mqtt.DOMAIN, {mqtt.DOMAIN: mqtt_config}) assert result