mirror of
https://github.com/home-assistant/core.git
synced 2025-07-23 05:07:41 +00:00
Speed up slow tests in Husqvarna Automower (#122854)
This commit is contained in:
parent
d9e996def5
commit
a5136a1021
@ -1,13 +1,18 @@
|
||||
"""Tests for number platform."""
|
||||
|
||||
from datetime import timedelta
|
||||
from unittest.mock import AsyncMock, patch
|
||||
|
||||
from aioautomower.exceptions import ApiException
|
||||
from aioautomower.utils import mower_list_to_dictionary_dataclass
|
||||
from freezegun.api import FrozenDateTimeFactory
|
||||
import pytest
|
||||
from syrupy import SnapshotAssertion
|
||||
|
||||
from homeassistant.components.husqvarna_automower.const import DOMAIN
|
||||
from homeassistant.components.husqvarna_automower.const import (
|
||||
DOMAIN,
|
||||
EXECUTION_TIME_DELAY,
|
||||
)
|
||||
from homeassistant.const import Platform
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.exceptions import HomeAssistantError
|
||||
@ -16,7 +21,12 @@ from homeassistant.helpers import entity_registry as er
|
||||
from . import setup_integration
|
||||
from .const import TEST_MOWER_ID
|
||||
|
||||
from tests.common import MockConfigEntry, load_json_value_fixture, snapshot_platform
|
||||
from tests.common import (
|
||||
MockConfigEntry,
|
||||
async_fire_time_changed,
|
||||
load_json_value_fixture,
|
||||
snapshot_platform,
|
||||
)
|
||||
|
||||
|
||||
@pytest.mark.usefixtures("entity_registry_enabled_by_default")
|
||||
@ -57,6 +67,7 @@ async def test_number_workarea_commands(
|
||||
hass: HomeAssistant,
|
||||
mock_automower_client: AsyncMock,
|
||||
mock_config_entry: MockConfigEntry,
|
||||
freezer: FrozenDateTimeFactory,
|
||||
) -> None:
|
||||
"""Test number commands."""
|
||||
entity_id = "number.test_mower_1_front_lawn_cutting_height"
|
||||
@ -75,8 +86,11 @@ async def test_number_workarea_commands(
|
||||
service="set_value",
|
||||
target={"entity_id": entity_id},
|
||||
service_data={"value": "75"},
|
||||
blocking=True,
|
||||
blocking=False,
|
||||
)
|
||||
freezer.tick(timedelta(seconds=EXECUTION_TIME_DELAY))
|
||||
async_fire_time_changed(hass)
|
||||
await hass.async_block_till_done()
|
||||
mocked_method.assert_called_once_with(TEST_MOWER_ID, 75, 123456)
|
||||
state = hass.states.get(entity_id)
|
||||
assert state.state is not None
|
||||
|
@ -1,5 +1,6 @@
|
||||
"""Tests for switch platform."""
|
||||
|
||||
from datetime import timedelta
|
||||
from unittest.mock import AsyncMock, patch
|
||||
|
||||
from aioautomower.exceptions import ApiException
|
||||
@ -9,7 +10,10 @@ from freezegun.api import FrozenDateTimeFactory
|
||||
import pytest
|
||||
from syrupy import SnapshotAssertion
|
||||
|
||||
from homeassistant.components.husqvarna_automower.const import DOMAIN
|
||||
from homeassistant.components.husqvarna_automower.const import (
|
||||
DOMAIN,
|
||||
EXECUTION_TIME_DELAY,
|
||||
)
|
||||
from homeassistant.components.husqvarna_automower.coordinator import SCAN_INTERVAL
|
||||
from homeassistant.const import Platform
|
||||
from homeassistant.core import HomeAssistant
|
||||
@ -109,6 +113,7 @@ async def test_stay_out_zone_switch_commands(
|
||||
excepted_state: str,
|
||||
mock_automower_client: AsyncMock,
|
||||
mock_config_entry: MockConfigEntry,
|
||||
freezer: FrozenDateTimeFactory,
|
||||
) -> None:
|
||||
"""Test switch commands."""
|
||||
entity_id = "switch.test_mower_1_avoid_danger_zone"
|
||||
@ -124,8 +129,11 @@ async def test_stay_out_zone_switch_commands(
|
||||
domain="switch",
|
||||
service=service,
|
||||
service_data={"entity_id": entity_id},
|
||||
blocking=True,
|
||||
blocking=False,
|
||||
)
|
||||
freezer.tick(timedelta(seconds=EXECUTION_TIME_DELAY))
|
||||
async_fire_time_changed(hass)
|
||||
await hass.async_block_till_done()
|
||||
mocked_method.assert_called_once_with(TEST_MOWER_ID, TEST_ZONE_ID, boolean)
|
||||
state = hass.states.get(entity_id)
|
||||
assert state is not None
|
||||
|
Loading…
x
Reference in New Issue
Block a user