30 lines
832 B
Python
30 lines
832 B
Python
"""Patch time related functions."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import datetime
|
|
import time
|
|
|
|
from homeassistant import runner, util
|
|
from homeassistant.helpers import event as event_helper
|
|
from homeassistant.util import dt as dt_util
|
|
|
|
|
|
def _utcnow() -> datetime.datetime:
|
|
"""Make utcnow patchable by freezegun."""
|
|
return datetime.datetime.now(datetime.UTC)
|
|
|
|
|
|
def _monotonic() -> float:
|
|
"""Make monotonic patchable by freezegun."""
|
|
return time.monotonic()
|
|
|
|
|
|
# Replace partial functions which are not found by freezegun
|
|
dt_util.utcnow = _utcnow # type: ignore[assignment]
|
|
event_helper.time_tracker_utcnow = _utcnow # type: ignore[assignment]
|
|
util.utcnow = _utcnow # type: ignore[assignment]
|
|
|
|
# Replace bound methods which are not found by freezegun
|
|
runner.monotonic = _monotonic # type: ignore[assignment]
|