evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Prepare for refactoring of MQTT related tests (#88557)

Browse source 
* Update mqtt_mock

* Tests manual_mqtt

* Tests mqtt_json

* Tests mqtt_room
This commit is contained in:
Jan Bouwhuis 2023-02-27 13:54:56 +01:00 • committed by GitHub
parent b542f6b3ac
commit b25f6e3ffc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 43 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

65
tests/components/manual_mqtt/test_alarm_control_panel.py
View file

@ -33,13 +33,13 @@ from tests.common import (
async_fire_time_changed, async_fire_time_changed,
) )
from tests.components.alarm_control_panel import common from tests.components.alarm_control_panel import common
from tests.typing import MqttMockHAClientGenerator from tests.typing import MqttMockHAClient
CODE = "HELLO_CODE" CODE = "HELLO_CODE"
async def test_fail_setup_without_state_topic( async def test_fail_setup_without_state_topic(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test for failing with no state topic.""" """Test for failing with no state topic."""
with assert_setup_component(0, alarm_control_panel.DOMAIN) as config: with assert_setup_component(0, alarm_control_panel.DOMAIN) as config:
@ -57,7 +57,7 @@ async def test_fail_setup_without_state_topic(
async def test_fail_setup_without_command_topic( async def test_fail_setup_without_command_topic(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test failing with no command topic.""" """Test failing with no command topic."""
with assert_setup_component(0, alarm_control_panel.DOMAIN): with assert_setup_component(0, alarm_control_panel.DOMAIN):
@ -87,7 +87,7 @@ async def test_no_pending(
hass: HomeAssistant, hass: HomeAssistant,
service, service,
expected_state, expected_state,
mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator, mqtt_mock: MqttMockHAClient,
) -> None: ) -> None:
"""Test arm method.""" """Test arm method."""
assert await async_setup_component( assert await async_setup_component(
@ -135,7 +135,7 @@ async def test_no_pending_when_code_not_req(
hass: HomeAssistant, hass: HomeAssistant,
service, service,
expected_state, expected_state,
mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator, mqtt_mock: MqttMockHAClient,
) -> None: ) -> None:
"""Test arm method.""" """Test arm method."""
assert await async_setup_component( assert await async_setup_component(
@ -184,7 +184,7 @@ async def test_with_pending(
hass: HomeAssistant, hass: HomeAssistant,
service, service,
expected_state, expected_state,
mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator, mqtt_mock: MqttMockHAClient,
) -> None: ) -> None:
"""Test arm method.""" """Test arm method."""
assert await async_setup_component( assert await async_setup_component(
@ -256,7 +256,7 @@ async def test_with_invalid_code(
hass: HomeAssistant, hass: HomeAssistant,
service, service,
expected_state, expected_state,
mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator, mqtt_mock: MqttMockHAClient,
) -> None: ) -> None:
"""Attempt to arm without a valid code.""" """Attempt to arm without a valid code."""
assert await async_setup_component( assert await async_setup_component(
@ -304,7 +304,7 @@ async def test_with_template_code(
hass: HomeAssistant, hass: HomeAssistant,
service, service,
expected_state, expected_state,
mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator, mqtt_mock: MqttMockHAClient,
) -> None: ) -> None:
"""Attempt to arm with a template-based code.""" """Attempt to arm with a template-based code."""
assert await async_setup_component( assert await async_setup_component(
@ -353,7 +353,7 @@ async def test_with_specific_pending(
hass: HomeAssistant, hass: HomeAssistant,
service, service,
expected_state, expected_state,
mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator, mqtt_mock: MqttMockHAClient,
) -> None: ) -> None:
"""Test arm method.""" """Test arm method."""
assert await async_setup_component( assert await async_setup_component(
@ -395,7 +395,7 @@ async def test_with_specific_pending(
async def test_trigger_no_pending( async def test_trigger_no_pending(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test triggering when no pending submitted method.""" """Test triggering when no pending submitted method."""
assert await async_setup_component( assert await async_setup_component(
@ -435,7 +435,7 @@ async def test_trigger_no_pending(
async def test_trigger_with_delay( async def test_trigger_with_delay(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test trigger method and switch from pending to triggered.""" """Test trigger method and switch from pending to triggered."""
assert await async_setup_component( assert await async_setup_component(
@ -483,7 +483,7 @@ async def test_trigger_with_delay(
async def test_trigger_zero_trigger_time( async def test_trigger_zero_trigger_time(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test disabled trigger.""" """Test disabled trigger."""
assert await async_setup_component( assert await async_setup_component(
@ -513,7 +513,7 @@ async def test_trigger_zero_trigger_time(
async def test_trigger_zero_trigger_time_with_pending( async def test_trigger_zero_trigger_time_with_pending(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test disabled trigger.""" """Test disabled trigger."""
assert await async_setup_component( assert await async_setup_component(
@ -543,7 +543,7 @@ async def test_trigger_zero_trigger_time_with_pending(
async def test_trigger_with_pending( async def test_trigger_with_pending(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test arm home method.""" """Test arm home method."""
assert await async_setup_component( assert await async_setup_component(
@ -596,7 +596,7 @@ async def test_trigger_with_pending(
async def test_trigger_with_disarm_after_trigger( async def test_trigger_with_disarm_after_trigger(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test disarm after trigger.""" """Test disarm after trigger."""
assert await async_setup_component( assert await async_setup_component(
@ -636,7 +636,7 @@ async def test_trigger_with_disarm_after_trigger(
async def test_trigger_with_zero_specific_trigger_time( async def test_trigger_with_zero_specific_trigger_time(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test trigger method.""" """Test trigger method."""
assert await async_setup_component( assert await async_setup_component(
@ -667,7 +667,7 @@ async def test_trigger_with_zero_specific_trigger_time(
async def test_trigger_with_unused_zero_specific_trigger_time( async def test_trigger_with_unused_zero_specific_trigger_time(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test disarm after trigger.""" """Test disarm after trigger."""
assert await async_setup_component( assert await async_setup_component(
@ -708,7 +708,7 @@ async def test_trigger_with_unused_zero_specific_trigger_time(
async def test_trigger_with_specific_trigger_time( async def test_trigger_with_specific_trigger_time(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test disarm after trigger.""" """Test disarm after trigger."""
assert await async_setup_component( assert await async_setup_component(
@ -748,7 +748,7 @@ async def test_trigger_with_specific_trigger_time(
async def test_back_to_back_trigger_with_no_disarm_after_trigger( async def test_back_to_back_trigger_with_no_disarm_after_trigger(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test no disarm after back to back trigger.""" """Test no disarm after back to back trigger."""
assert await async_setup_component( assert await async_setup_component(
@ -806,7 +806,7 @@ async def test_back_to_back_trigger_with_no_disarm_after_trigger(
async def test_disarm_while_pending_trigger( async def test_disarm_while_pending_trigger(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test disarming while pending state.""" """Test disarming while pending state."""
assert await async_setup_component( assert await async_setup_component(
@ -849,7 +849,7 @@ async def test_disarm_while_pending_trigger(
async def test_disarm_during_trigger_with_invalid_code( async def test_disarm_during_trigger_with_invalid_code(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test disarming while code is invalid.""" """Test disarming while code is invalid."""
assert await async_setup_component( assert await async_setup_component(
@ -897,7 +897,7 @@ async def test_disarm_during_trigger_with_invalid_code(
async def test_trigger_with_unused_specific_delay( async def test_trigger_with_unused_specific_delay(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test trigger method and switch from pending to triggered.""" """Test trigger method and switch from pending to triggered."""
assert await async_setup_component( assert await async_setup_component(
@ -946,7 +946,7 @@ async def test_trigger_with_unused_specific_delay(
async def test_trigger_with_specific_delay( async def test_trigger_with_specific_delay(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test trigger method and switch from pending to triggered.""" """Test trigger method and switch from pending to triggered."""
assert await async_setup_component( assert await async_setup_component(
@ -995,7 +995,7 @@ async def test_trigger_with_specific_delay(
async def test_trigger_with_pending_and_delay( async def test_trigger_with_pending_and_delay(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test trigger method and switch from pending to triggered.""" """Test trigger method and switch from pending to triggered."""
assert await async_setup_component( assert await async_setup_component(
@ -1056,7 +1056,7 @@ async def test_trigger_with_pending_and_delay(
async def test_trigger_with_pending_and_specific_delay( async def test_trigger_with_pending_and_specific_delay(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test trigger method and switch from pending to triggered.""" """Test trigger method and switch from pending to triggered."""
assert await async_setup_component( assert await async_setup_component(
@ -1118,7 +1118,7 @@ async def test_trigger_with_pending_and_specific_delay(
async def test_trigger_with_specific_pending( async def test_trigger_with_specific_pending(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test arm home method.""" """Test arm home method."""
assert await async_setup_component( assert await async_setup_component(
@ -1167,7 +1167,7 @@ async def test_trigger_with_specific_pending(
async def test_trigger_with_no_disarm_after_trigger( async def test_trigger_with_no_disarm_after_trigger(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test disarm after trigger.""" """Test disarm after trigger."""
assert await async_setup_component( assert await async_setup_component(
@ -1212,7 +1212,7 @@ async def test_trigger_with_no_disarm_after_trigger(
async def test_arm_away_after_disabled_disarmed( async def test_arm_away_after_disabled_disarmed(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test pending state with and without zero trigger time.""" """Test pending state with and without zero trigger time."""
assert await async_setup_component( assert await async_setup_component(
@ -1278,7 +1278,7 @@ async def test_arm_away_after_disabled_disarmed(
async def test_disarm_with_template_code( async def test_disarm_with_template_code(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Attempt to disarm with a valid or invalid template-based code.""" """Attempt to disarm with a valid or invalid template-based code."""
assert await async_setup_component( assert await async_setup_component(
@ -1332,7 +1332,7 @@ async def test_arm_via_command_topic(
hass: HomeAssistant, hass: HomeAssistant,
config, config,
expected_state, expected_state,
mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator, mqtt_mock: MqttMockHAClient,
) -> None: ) -> None:
"""Test arming via command topic.""" """Test arming via command topic."""
command = config[8:].upper() command = config[8:].upper()
@ -1374,7 +1374,7 @@ async def test_arm_via_command_topic(
async def test_disarm_pending_via_command_topic( async def test_disarm_pending_via_command_topic(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test disarming pending alarm via command topic.""" """Test disarming pending alarm via command topic."""
assert await async_setup_component( assert await async_setup_component(
@ -1410,7 +1410,7 @@ async def test_disarm_pending_via_command_topic(
async def test_state_changes_are_published_to_mqtt( async def test_state_changes_are_published_to_mqtt(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test publishing of MQTT messages when state changes.""" """Test publishing of MQTT messages when state changes."""
assert await async_setup_component( assert await async_setup_component(
@ -1431,7 +1431,6 @@ async def test_state_changes_are_published_to_mqtt(
# Component should send disarmed alarm state on startup # Component should send disarmed alarm state on startup
await hass.async_block_till_done() await hass.async_block_till_done()
mqtt_mock = await mqtt_mock_entry_with_yaml_config()
mqtt_mock.async_publish.assert_called_once_with( mqtt_mock.async_publish.assert_called_once_with(
"alarm/state", STATE_ALARM_DISARMED, 0, True "alarm/state", STATE_ALARM_DISARMED, 0, True
) )

6
tests/components/mqtt_json/test_device_tracker.py
View file

@ -1,4 +1,5 @@
"""The tests for the JSON MQTT device tracker platform.""" """The tests for the JSON MQTT device tracker platform."""
from collections.abc import Generator
import json import json
import logging import logging
import os import os
@ -15,6 +16,7 @@ from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component from homeassistant.setup import async_setup_component
from tests.common import async_fire_mqtt_message from tests.common import async_fire_mqtt_message
from tests.typing import MqttMockHAClient
LOCATION_MESSAGE = { LOCATION_MESSAGE = {
"longitude": 1.0, "longitude": 1.0,
@ -27,7 +29,9 @@ LOCATION_MESSAGE_INCOMPLETE = {"longitude": 2.0}
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
async def setup_comp(hass, mqtt_mock_entry_with_yaml_config): async def setup_comp(
hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> Generator[None, None, None]:
"""Initialize components.""" """Initialize components."""
yaml_devices = hass.config.path(YAML_DEVICES) yaml_devices = hass.config.path(YAML_DEVICES)
yield yield

8
tests/components/mqtt_room/test_sensor.py
View file

@ -18,7 +18,7 @@ from homeassistant.setup import async_setup_component
from homeassistant.util import dt from homeassistant.util import dt
from tests.common import async_fire_mqtt_message from tests.common import async_fire_mqtt_message
from tests.typing import MqttMockHAClientGenerator from tests.typing import MqttMockHAClient
DEVICE_ID = "123TESTMAC" DEVICE_ID = "123TESTMAC"
NAME = "test_device" NAME = "test_device"
@ -56,9 +56,7 @@ async def assert_distance(hass, distance):
assert state.attributes.get("distance") == distance assert state.attributes.get("distance") == distance
async def test_room_update( async def test_room_update(hass: HomeAssistant, mqtt_mock: MqttMockHAClient) -> None:
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator
) -> None:
"""Test the updating between rooms.""" """Test the updating between rooms."""
assert await async_setup_component( assert await async_setup_component(
hass, hass,
@ -96,7 +94,7 @@ async def test_room_update(
async def test_unique_id_is_set( async def test_unique_id_is_set(
hass: HomeAssistant, mqtt_mock_entry_with_yaml_config: MqttMockHAClientGenerator hass: HomeAssistant, mqtt_mock: MqttMockHAClient
) -> None: ) -> None:
"""Test the updating between rooms.""" """Test the updating between rooms."""
unique_name = "my_unique_name_0123456789" unique_name = "my_unique_name_0123456789"

2
tests/conftest.py
View file

@ -846,11 +846,13 @@ def mqtt_client_mock(hass: HomeAssistant) -> Generator[MqttMockPahoClient, None,
@pytest.fixture @pytest.fixture
async def mqtt_mock( async def mqtt_mock(
hass: HomeAssistant, hass: HomeAssistant,
mock_hass_config: None,
mqtt_client_mock: MqttMockPahoClient, mqtt_client_mock: MqttMockPahoClient,
mqtt_config_entry_data: dict[str, Any] | None, mqtt_config_entry_data: dict[str, Any] | None,
mqtt_mock_entry_no_yaml_config: MqttMockHAClientGenerator, mqtt_mock_entry_no_yaml_config: MqttMockHAClientGenerator,
) -> AsyncGenerator[MqttMockHAClient, None]: ) -> AsyncGenerator[MqttMockHAClient, None]:
"""Fixture to mock MQTT component.""" """Fixture to mock MQTT component."""
with patch("homeassistant.components.mqtt.PLATFORMS", []):
return await mqtt_mock_entry_no_yaml_config() return await mqtt_mock_entry_no_yaml_config()