From 92d150ff57ccd4df411b5cf0effcf879912b04d3 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 13 Jun 2024 09:21:59 +0200 Subject: [PATCH] Fix dangerous-default-value warnings in integration tests (#119574) --- tests/components/integration/test_sensor.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/tests/components/integration/test_sensor.py b/tests/components/integration/test_sensor.py index 5bc87717440..500d567dca4 100644 --- a/tests/components/integration/test_sensor.py +++ b/tests/components/integration/test_sensor.py @@ -1,6 +1,7 @@ """The tests for the integration sensor platform.""" from datetime import timedelta +from typing import Any from freezegun import freeze_time import pytest @@ -33,6 +34,8 @@ from tests.common import ( mock_restore_cache_with_extra_data, ) +DEFAULT_MAX_SUB_INTERVAL = {"minutes": 1} + @pytest.mark.parametrize("method", ["trapezoidal", "left", "right"]) async def test_state(hass: HomeAssistant, method) -> None: @@ -752,7 +755,7 @@ async def test_device_id( assert integration_entity.device_id == source_entity.device_id -def _integral_sensor_config(max_sub_interval: dict[str, int] | None = {"minutes": 1}): +def _integral_sensor_config(max_sub_interval: dict[str, int] | None) -> dict[str, Any]: sensor = { "platform": "integration", "name": "integration", @@ -765,7 +768,7 @@ def _integral_sensor_config(max_sub_interval: dict[str, int] | None = {"minutes" async def _setup_integral_sensor( - hass: HomeAssistant, max_sub_interval: dict[str, int] | None = {"minutes": 1} + hass: HomeAssistant, max_sub_interval: dict[str, int] | None ) -> None: await async_setup_component( hass, "sensor", _integral_sensor_config(max_sub_interval=max_sub_interval) @@ -775,7 +778,9 @@ async def _setup_integral_sensor( async def _update_source_sensor(hass: HomeAssistant, value: int | str) -> None: hass.states.async_set( - _integral_sensor_config()["sensor"]["source"], + _integral_sensor_config(max_sub_interval=DEFAULT_MAX_SUB_INTERVAL)["sensor"][ + "source" + ], value, {ATTR_UNIT_OF_MEASUREMENT: UnitOfPower.KILO_WATT}, force_update=True, @@ -790,7 +795,7 @@ async def test_on_valid_source_expect_update_on_time( start_time = dt_util.utcnow() with freeze_time(start_time) as freezer: - await _setup_integral_sensor(hass) + await _setup_integral_sensor(hass, max_sub_interval=DEFAULT_MAX_SUB_INTERVAL) await _update_source_sensor(hass, 100) state_before_max_sub_interval_exceeded = hass.states.get("sensor.integration") @@ -816,7 +821,7 @@ async def test_on_unvailable_source_expect_no_update_on_time( start_time = dt_util.utcnow() with freeze_time(start_time) as freezer: - await _setup_integral_sensor(hass) + await _setup_integral_sensor(hass, max_sub_interval=DEFAULT_MAX_SUB_INTERVAL) await _update_source_sensor(hass, 100) freezer.tick(61) async_fire_time_changed(hass, dt_util.now()) @@ -843,7 +848,7 @@ async def test_on_statechanges_source_expect_no_update_on_time( start_time = dt_util.utcnow() with freeze_time(start_time) as freezer: - await _setup_integral_sensor(hass) + await _setup_integral_sensor(hass, max_sub_interval=DEFAULT_MAX_SUB_INTERVAL) await _update_source_sensor(hass, 100) freezer.tick(30)