pytest-homeassistant-custom-component 0.13.163__py3-none-any.whl → 0.13.298__py3-none-any.whl

pytest_homeassistant_custom_component/const.py

@@ -4,8 +4,9 @@ Constants used by Home Assistant components.
 This file is originally from homeassistant/core and modified by pytest-homeassistant-custom-component.
 """
 from typing import TYPE_CHECKING, Final
-MAJOR_VERSION: Final = 2024
-MINOR_VERSION: Final = 9
-PATCH_VERSION: Final = "2"
+MAJOR_VERSION: Final = 2025
+MINOR_VERSION: Final = 12
+PATCH_VERSION: Final = "0"
 __short_version__: Final = f"{MAJOR_VERSION}.{MINOR_VERSION}"
 __version__: Final = f"{__short_version__}.{PATCH_VERSION}"
+CONF_API_VERSION: Final = "api_version"

pytest_homeassistant_custom_component/patch_json.py

@@ -0,0 +1,41 @@
+"""
+Patch JSON related functions.
+
+This file is originally from homeassistant/core and modified by pytest-homeassistant-custom-component.
+"""
+
+from __future__ import annotations
+
+import functools
+from typing import Any
+from unittest import mock
+
+import orjson
+
+from homeassistant.helpers import json as json_helper
+
+real_json_encoder_default = json_helper.json_encoder_default
+
+mock_objects = []
+
+
+def json_encoder_default(obj: Any) -> Any:
+    """Convert Home Assistant objects.
+
+    Hand other objects to the original method.
+    """
+    if isinstance(obj, mock.Base):
+        mock_objects.append(obj)
+        raise TypeError(f"Attempting to serialize mock object {obj}")
+    return real_json_encoder_default(obj)
+
+
+json_helper.json_encoder_default = json_encoder_default
+json_helper.json_bytes = functools.partial(
+    orjson.dumps, option=orjson.OPT_NON_STR_KEYS, default=json_encoder_default
+)
+json_helper.json_bytes_sorted = functools.partial(
+    orjson.dumps,
+    option=orjson.OPT_NON_STR_KEYS | orjson.OPT_SORT_KEYS,
+    default=json_encoder_default,
+)

pytest_homeassistant_custom_component/patch_recorder.py

@@ -10,7 +10,7 @@ from contextlib import contextmanager
 import sys
 
 # Patch recorder util session scope
-from homeassistant.helpers import recorder as recorder_helper  # noqa: E402
+from homeassistant.helpers import recorder as recorder_helper
 
 # Make sure homeassistant.components.recorder.util is not already imported
 assert "homeassistant.components.recorder.util" not in sys.modules

pytest_homeassistant_custom_component/patch_time.py

@@ -9,6 +9,72 @@ from __future__ import annotations
 import datetime
 import time
 
+import freezegun
+
+
+def ha_datetime_to_fakedatetime(datetime) -> freezegun.api.FakeDatetime:  # type: ignore[name-defined]
+    """Convert datetime to FakeDatetime.
+
+    Modified to include https://github.com/spulec/freezegun/pull/424.
+    """
+    return freezegun.api.FakeDatetime(  # type: ignore[attr-defined]
+        datetime.year,
+        datetime.month,
+        datetime.day,
+        datetime.hour,
+        datetime.minute,
+        datetime.second,
+        datetime.microsecond,
+        datetime.tzinfo,
+        fold=datetime.fold,
+    )
+
+
+class HAFakeDateMeta(freezegun.api.FakeDateMeta):
+    """Modified to override the string representation."""
+
+    def __str__(cls) -> str:  # noqa: N805 (ruff doesn't know this is a metaclass)
+        """Return the string representation of the class."""
+        return "<class 'datetime.date'>"
+
+
+class HAFakeDate(freezegun.api.FakeDate, metaclass=HAFakeDateMeta):  # type: ignore[name-defined]
+    """Modified to improve class str."""
+
+
+class HAFakeDatetimeMeta(freezegun.api.FakeDatetimeMeta):
+    """Modified to override the string representation."""
+
+    def __str__(cls) -> str:  # noqa: N805 (ruff doesn't know this is a metaclass)
+        """Return the string representation of the class."""
+        return "<class 'datetime.datetime'>"
+
+
+class HAFakeDatetime(freezegun.api.FakeDatetime, metaclass=HAFakeDatetimeMeta):  # type: ignore[name-defined]
+    """Modified to include basic fold support and improve class str.
+
+    Fold support submitted to upstream in https://github.com/spulec/freezegun/pull/424.
+    """
+
+    @classmethod
+    def now(cls, tz=None):
+        """Return frozen now."""
+        now = cls._time_to_freeze() or freezegun.api.real_datetime.now()
+        if tz:
+            result = tz.fromutc(now.replace(tzinfo=tz))
+        else:
+            result = now
+
+        # Add the _tz_offset only if it's non-zero to preserve fold
+        if cls._tz_offset():
+            result += cls._tz_offset()
+
+        return ha_datetime_to_fakedatetime(result)
+
+
+# Needed by Mashumaro
+datetime.HAFakeDatetime = HAFakeDatetime
+
 # Do not add any Home Assistant import here