From acb147767366e73402a6ad4f9e8a053d26959110 Mon Sep 17 00:00:00 2001 From: Erik Montnemery Date: Thu, 13 Oct 2022 11:51:27 +0200 Subject: [PATCH] Avoid time traveling in recorder tests (#80247) --- tests/components/recorder/common.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/tests/components/recorder/common.py b/tests/components/recorder/common.py index 8d1929c7362..0ddc76e4423 100644 --- a/tests/components/recorder/common.py +++ b/tests/components/recorder/common.py @@ -3,7 +3,7 @@ from __future__ import annotations import asyncio from dataclasses import dataclass -from datetime import datetime, timedelta +from datetime import datetime import time from typing import Any, cast @@ -21,8 +21,6 @@ from homeassistant.util import dt as dt_util from . import db_schema_0 -from tests.common import async_fire_time_changed, fire_time_changed - DEFAULT_PURGE_TASKS = 3 @@ -69,9 +67,7 @@ def wait_recording_done(hass: HomeAssistant) -> None: def trigger_db_commit(hass: HomeAssistant) -> None: """Force the recorder to commit.""" - for _ in range(recorder.DEFAULT_COMMIT_INTERVAL): - # We only commit on time change - fire_time_changed(hass, dt_util.utcnow() + timedelta(seconds=1)) + recorder.get_instance(hass)._async_commit(dt_util.utcnow()) async def async_wait_recording_done(hass: HomeAssistant) -> None: @@ -100,8 +96,7 @@ async def async_wait_purge_done(hass: HomeAssistant, max: int = None) -> None: @ha.callback def async_trigger_db_commit(hass: HomeAssistant) -> None: """Force the recorder to commit. Async friendly.""" - for _ in range(recorder.DEFAULT_COMMIT_INTERVAL): - async_fire_time_changed(hass, dt_util.utcnow() + timedelta(seconds=1)) + recorder.get_instance(hass)._async_commit(dt_util.utcnow()) async def async_recorder_block_till_done(hass: HomeAssistant) -> None: