pytest-homeassistant-custom-component 0.13.289__tar.gz → 0.13.305__tar.gz

{pytest_homeassistant_custom_component-0.13.289/src/pytest_homeassistant_custom_component.egg-info → pytest_homeassistant_custom_component-0.13.305}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-homeassistant-custom-component
-Version: 0.13.289
+Version: 0.13.305
 Summary: Experimental package to automatically extract test plugins for Home Assistant custom components
 Home-page: https://github.com/MatthewFlamm/pytest-homeassistant-custom-component
 Author: Matthew Flamm
@@ -20,13 +20,12 @@ License-File: LICENSE_HA_CORE.md
 Requires-Dist: sqlalchemy
 Requires-Dist: coverage==7.10.6
 Requires-Dist: freezegun==1.5.2
-Requires-Dist: go2rtc-client==0.2.1
 Requires-Dist: license-expression==30.4.3
 Requires-Dist: mock-open==1.4.0
-Requires-Dist: pydantic==2.11.9
+Requires-Dist: pydantic==2.12.2
 Requires-Dist: pylint-per-file-ignores==1.4.0
 Requires-Dist: pipdeptree==2.26.1
-Requires-Dist: pytest-asyncio==1.2.0
+Requires-Dist: pytest-asyncio==1.3.0
 Requires-Dist: pytest-aiohttp==1.1.0
 Requires-Dist: pytest-cov==7.0.0
 Requires-Dist: pytest-freezer==0.4.9
@@ -37,12 +36,12 @@ Requires-Dist: pytest-timeout==2.4.0
 Requires-Dist: pytest-unordered==0.7.0
 Requires-Dist: pytest-picked==0.5.1
 Requires-Dist: pytest-xdist==3.8.0
-Requires-Dist: pytest==8.4.2
+Requires-Dist: pytest==9.0.0
 Requires-Dist: requests-mock==1.12.1
 Requires-Dist: respx==0.22.0
-Requires-Dist: syrupy==4.9.1
+Requires-Dist: syrupy==5.0.0
 Requires-Dist: tqdm==4.67.1
-Requires-Dist: homeassistant==2025.10.4
+Requires-Dist: homeassistant==2026.1.0
 Requires-Dist: SQLAlchemy==2.0.41
 Requires-Dist: paho-mqtt==2.1.0
 Requires-Dist: numpy==2.3.2
@@ -60,9 +59,7 @@ Dynamic: summary
 
 # pytest-homeassistant-custom-component
 
-![HA core version](https://img.shields.io/static/v1?label=HA+core+version&message=2025.10.4&labelColor=blue)
-
-[![Open in Gitpod](https://gitpod.io/button/open-in-gitpod.svg)](https://gitpod.io/#https://github.com/MatthewFlamm/pytest-homeassistant-custom-component)
+![HA core version](https://img.shields.io/static/v1?label=HA+core+version&message=2026.1.0&labelColor=blue)
 
 Package to automatically extract testing plugins from Home Assistant for custom component testing.
 The goal is to provide the same functionality as the tests in home-assistant/core.

{pytest_homeassistant_custom_component-0.13.289 → pytest_homeassistant_custom_component-0.13.305}/README.md

@@ -1,8 +1,6 @@
 # pytest-homeassistant-custom-component
 
-![HA core version](https://img.shields.io/static/v1?label=HA+core+version&message=2025.10.4&labelColor=blue)
-
-[![Open in Gitpod](https://gitpod.io/button/open-in-gitpod.svg)](https://gitpod.io/#https://github.com/MatthewFlamm/pytest-homeassistant-custom-component)
+![HA core version](https://img.shields.io/static/v1?label=HA+core+version&message=2026.1.0&labelColor=blue)
 
 Package to automatically extract testing plugins from Home Assistant for custom component testing.
 The goal is to provide the same functionality as the tests in home-assistant/core.

{pytest_homeassistant_custom_component-0.13.289 → pytest_homeassistant_custom_component-0.13.305}/src/pytest_homeassistant_custom_component/common.py

@@ -1,5 +1,5 @@
 """
-Test the helper method for writing .
+Test the helper method for writing tests.
 
 This file is originally from homeassistant/core and modified by pytest-homeassistant-custom-component.
 """
@@ -204,14 +204,14 @@ class StoreWithoutWriteLoad[_T: (Mapping[str, Any] | Sequence[Any])](storage.Sto
     async def async_save(self, *args: Any, **kwargs: Any) -> None:
         """Save the data.
 
-        This function is mocked out in .
+        This function is mocked out in tests.
         """
 
     @callback
     def async_save_delay(self, *args: Any, **kwargs: Any) -> None:
         """Save data with an optional delay.
 
-        This function is mocked out in .
+        This function is mocked out in tests.
         """
 
 
@@ -1148,7 +1148,7 @@ class MockConfigEntry(config_entries.ConfigEntry):
         state: config_entries.ConfigEntryState,
         reason: str | None = None,
     ) -> None:
-        """Mock the state of a config entry to be used in .
+        """Mock the state of a config entry to be used in tests.
 
         Currently this is a wrapper around _async_set_state, but it may
         change in the future.
@@ -1159,7 +1159,7 @@ class MockConfigEntry(config_entries.ConfigEntry):
 
         When in doubt, this helper should not be used in new code
         and is only intended for backwards compatibility with existing
-        .
+        tests.
         """
         self._async_set_state(hass, state, reason)
 
@@ -1539,7 +1539,7 @@ def mock_storage(data: dict[str, Any] | None = None) -> Generator[dict[str, Any]
         return loaded
 
     async def mock_write_data(
-        store: storage.Store, path: str, data_to_write: dict[str, Any]
+        store: storage.Store, data_to_write: dict[str, Any]
     ) -> None:
         """Mock version of write data."""
         # To ensure that the data can be serialized
@@ -1616,12 +1616,16 @@ def mock_integration(
     top_level_files: set[str] | None = None,
 ) -> loader.Integration:
     """Mock an integration."""
-    integration = loader.Integration(
-        hass,
+    path = (
         f"{loader.PACKAGE_BUILTIN}.{module.DOMAIN}"
         if built_in
-        else f"{loader.PACKAGE_CUSTOM_COMPONENTS}.{module.DOMAIN}",
-        pathlib.Path(""),
+        else f"{loader.PACKAGE_CUSTOM_COMPONENTS}.{module.DOMAIN}"
+    )
+
+    integration = loader.Integration(
+        hass,
+        path,
+        pathlib.Path(path.replace(".", "/")),
         module.mock_manifest(),
         top_level_files,
     )
@@ -1914,7 +1918,7 @@ def setup_test_component_platform(
     from_config_entry: bool = False,
     built_in: bool = True,
 ) -> MockPlatform:
-    """Mock a test component platform for ."""
+    """Mock a test component platform for tests."""
 
     async def _async_setup_platform(
         hass: HomeAssistant,

pytest_homeassistant_custom_component-0.13.305/src/pytest_homeassistant_custom_component/components/__init__.py

@@ -0,0 +1,399 @@
+"""
+The tests for components.
+
+This file is originally from homeassistant/core and modified by pytest-homeassistant-custom-component.
+"""
+
+from collections.abc import Iterable
+from enum import StrEnum
+import itertools
+from typing import TypedDict
+
+from homeassistant.const import (
+    ATTR_AREA_ID,
+    ATTR_DEVICE_ID,
+    ATTR_FLOOR_ID,
+    ATTR_LABEL_ID,
+    CONF_ENTITY_ID,
+    CONF_OPTIONS,
+    CONF_PLATFORM,
+    CONF_TARGET,
+    STATE_UNAVAILABLE,
+    STATE_UNKNOWN,
+)
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import (
+    area_registry as ar,
+    device_registry as dr,
+    entity_registry as er,
+    floor_registry as fr,
+    label_registry as lr,
+)
+from homeassistant.setup import async_setup_component
+
+from ..common import MockConfigEntry, mock_device_registry
+
+
+async def target_entities(
+    hass: HomeAssistant, domain: str
+) -> tuple[list[str], list[str]]:
+    """Create multiple entities associated with different targets.
+
+    Returns a dict with the following keys:
+    - included: List of entity_ids meant to be targeted.
+    - excluded: List of entity_ids not meant to be targeted.
+    """
+    config_entry = MockConfigEntry(domain="test")
+    config_entry.add_to_hass(hass)
+
+    floor_reg = fr.async_get(hass)
+    floor = floor_reg.async_get_floor_by_name("Test Floor") or floor_reg.async_create(
+        "Test Floor"
+    )
+
+    area_reg = ar.async_get(hass)
+    area = area_reg.async_get_area_by_name("Test Area") or area_reg.async_create(
+        "Test Area", floor_id=floor.floor_id
+    )
+
+    label_reg = lr.async_get(hass)
+    label = label_reg.async_get_label_by_name("Test Label") or label_reg.async_create(
+        "Test Label"
+    )
+
+    device = dr.DeviceEntry(id="test_device", area_id=area.id, labels={label.label_id})
+    mock_device_registry(hass, {device.id: device})
+
+    entity_reg = er.async_get(hass)
+    # Entities associated with area
+    entity_area = entity_reg.async_get_or_create(
+        domain=domain,
+        platform="test",
+        unique_id=f"{domain}_area",
+        suggested_object_id=f"area_{domain}",
+    )
+    entity_reg.async_update_entity(entity_area.entity_id, area_id=area.id)
+    entity_area_excluded = entity_reg.async_get_or_create(
+        domain=domain,
+        platform="test",
+        unique_id=f"{domain}_area_excluded",
+        suggested_object_id=f"area_{domain}_excluded",
+    )
+    entity_reg.async_update_entity(entity_area_excluded.entity_id, area_id=area.id)
+
+    # Entities associated with device
+    entity_reg.async_get_or_create(
+        domain=domain,
+        platform="test",
+        unique_id=f"{domain}_device",
+        suggested_object_id=f"device_{domain}",
+        device_id=device.id,
+    )
+    entity_reg.async_get_or_create(
+        domain=domain,
+        platform="test",
+        unique_id=f"{domain}_device_excluded",
+        suggested_object_id=f"device_{domain}_excluded",
+        device_id=device.id,
+    )
+
+    # Entities associated with label
+    entity_label = entity_reg.async_get_or_create(
+        domain=domain,
+        platform="test",
+        unique_id=f"{domain}_label",
+        suggested_object_id=f"label_{domain}",
+    )
+    entity_reg.async_update_entity(entity_label.entity_id, labels={label.label_id})
+    entity_label_excluded = entity_reg.async_get_or_create(
+        domain=domain,
+        platform="test",
+        unique_id=f"{domain}_label_excluded",
+        suggested_object_id=f"label_{domain}_excluded",
+    )
+    entity_reg.async_update_entity(
+        entity_label_excluded.entity_id, labels={label.label_id}
+    )
+
+    # Return all available entities
+    return {
+        "included": [
+            f"{domain}.standalone_{domain}",
+            f"{domain}.label_{domain}",
+            f"{domain}.area_{domain}",
+            f"{domain}.device_{domain}",
+        ],
+        "excluded": [
+            f"{domain}.standalone_{domain}_excluded",
+            f"{domain}.label_{domain}_excluded",
+            f"{domain}.area_{domain}_excluded",
+            f"{domain}.device_{domain}_excluded",
+        ],
+    }
+
+
+def parametrize_target_entities(domain: str) -> list[tuple[dict, str, int]]:
+    """Parametrize target entities for different target types.
+
+    Meant to be used with target_entities.
+    """
+    return [
+        (
+            {CONF_ENTITY_ID: f"{domain}.standalone_{domain}"},
+            f"{domain}.standalone_{domain}",
+            1,
+        ),
+        ({ATTR_LABEL_ID: "test_label"}, f"{domain}.label_{domain}", 2),
+        ({ATTR_AREA_ID: "test_area"}, f"{domain}.area_{domain}", 2),
+        ({ATTR_FLOOR_ID: "test_floor"}, f"{domain}.area_{domain}", 2),
+        ({ATTR_LABEL_ID: "test_label"}, f"{domain}.device_{domain}", 2),
+        ({ATTR_AREA_ID: "test_area"}, f"{domain}.device_{domain}", 2),
+        ({ATTR_FLOOR_ID: "test_floor"}, f"{domain}.device_{domain}", 2),
+        ({ATTR_DEVICE_ID: "test_device"}, f"{domain}.device_{domain}", 1),
+    ]
+
+
+class _StateDescription(TypedDict):
+    """Test state and expected service call count."""
+
+    state: str | None
+    attributes: dict
+
+
+class StateDescription(TypedDict):
+    """Test state and expected service call count."""
+
+    included: _StateDescription
+    excluded: _StateDescription
+    count: int
+
+
+def parametrize_trigger_states(
+    *,
+    trigger: str,
+    target_states: list[str | None | tuple[str | None, dict]],
+    other_states: list[str | None | tuple[str | None, dict]],
+    additional_attributes: dict | None = None,
+    trigger_from_none: bool = True,
+    retrigger_on_target_state: bool = False,
+) -> list[tuple[str, list[StateDescription]]]:
+    """Parametrize states and expected service call counts.
+
+    The target_states and other_states iterables are either iterables of
+    states or iterables of (state, attributes) tuples.
+
+    Set `trigger_from_none` to False if the trigger is not expected to fire
+    when the initial state is None.
+
+    Set `retrigger_on_target_state` to True if the trigger is expected to fire
+    when the state changes to another target state.
+
+    Returns a list of tuples with (trigger, list of states),
+    where states is a list of StateDescription dicts.
+    """
+
+    additional_attributes = additional_attributes or {}
+
+    def state_with_attributes(
+        state: str | None | tuple[str | None, dict], count: int
+    ) -> dict:
+        """Return (state, attributes) dict."""
+        if isinstance(state, str) or state is None:
+            return {
+                "included": {
+                    "state": state,
+                    "attributes": additional_attributes,
+                },
+                "excluded": {
+                    "state": state,
+                    "attributes": {},
+                },
+                "count": count,
+            }
+        return {
+            "included": {
+                "state": state[0],
+                "attributes": state[1] | additional_attributes,
+            },
+            "excluded": {
+                "state": state[0],
+                "attributes": state[1],
+            },
+            "count": count,
+        }
+
+    tests = [
+        # Initial state None
+        (
+            trigger,
+            list(
+                itertools.chain.from_iterable(
+                    (
+                        state_with_attributes(None, 0),
+                        state_with_attributes(target_state, 0),
+                        state_with_attributes(other_state, 0),
+                        state_with_attributes(
+                            target_state, 1 if trigger_from_none else 0
+                        ),
+                    )
+                    for target_state in target_states
+                    for other_state in other_states
+                )
+            ),
+        ),
+        # Initial state different from target state
+        (
+            trigger,
+            # other_state,
+            list(
+                itertools.chain.from_iterable(
+                    (
+                        state_with_attributes(other_state, 0),
+                        state_with_attributes(target_state, 1),
+                        state_with_attributes(other_state, 0),
+                        state_with_attributes(target_state, 1),
+                    )
+                    for target_state in target_states
+                    for other_state in other_states
+                )
+            ),
+        ),
+        # Initial state same as target state
+        (
+            trigger,
+            list(
+                itertools.chain.from_iterable(
+                    (
+                        state_with_attributes(target_state, 0),
+                        state_with_attributes(target_state, 0),
+                        state_with_attributes(other_state, 0),
+                        state_with_attributes(target_state, 1),
+                        # Repeat target state to test retriggering
+                        state_with_attributes(target_state, 0),
+                        state_with_attributes(STATE_UNAVAILABLE, 0),
+                    )
+                    for target_state in target_states
+                    for other_state in other_states
+                )
+            ),
+        ),
+        # Initial state unavailable / unknown
+        (
+            trigger,
+            list(
+                itertools.chain.from_iterable(
+                    (
+                        state_with_attributes(STATE_UNAVAILABLE, 0),
+                        state_with_attributes(target_state, 0),
+                        state_with_attributes(other_state, 0),
+                        state_with_attributes(target_state, 1),
+                    )
+                    for target_state in target_states
+                    for other_state in other_states
+                )
+            ),
+        ),
+        (
+            trigger,
+            list(
+                itertools.chain.from_iterable(
+                    (
+                        state_with_attributes(STATE_UNKNOWN, 0),
+                        state_with_attributes(target_state, 0),
+                        state_with_attributes(other_state, 0),
+                        state_with_attributes(target_state, 1),
+                    )
+                    for target_state in target_states
+                    for other_state in other_states
+                )
+            ),
+        ),
+    ]
+
+    if len(target_states) > 1:
+        # If more than one target state, test state change between target states
+        tests.append(
+            (
+                trigger,
+                list(
+                    itertools.chain.from_iterable(
+                        (
+                            state_with_attributes(target_states[idx - 1], 0),
+                            state_with_attributes(
+                                target_state, 1 if retrigger_on_target_state else 0
+                            ),
+                            state_with_attributes(other_state, 0),
+                            state_with_attributes(target_states[idx - 1], 1),
+                            state_with_attributes(
+                                target_state, 1 if retrigger_on_target_state else 0
+                            ),
+                            state_with_attributes(STATE_UNAVAILABLE, 0),
+                        )
+                        for idx, target_state in enumerate(target_states[1:], start=1)
+                        for other_state in other_states
+                    )
+                ),
+            ),
+        )
+
+    return tests
+
+
+async def arm_trigger(
+    hass: HomeAssistant,
+    trigger: str,
+    trigger_options: dict | None,
+    trigger_target: dict,
+) -> None:
+    """Arm the specified trigger, call service test.automation when it triggers."""
+
+    # Local include to avoid importing the automation component unnecessarily
+    from homeassistant.components import automation  # noqa: PLC0415
+
+    options = {CONF_OPTIONS: {**trigger_options}} if trigger_options is not None else {}
+
+    await async_setup_component(
+        hass,
+        automation.DOMAIN,
+        {
+            automation.DOMAIN: {
+                "trigger": {
+                    CONF_PLATFORM: trigger,
+                    CONF_TARGET: {**trigger_target},
+                }
+                | options,
+                "action": {
+                    "service": "test.automation",
+                    "data_template": {CONF_ENTITY_ID: "{{ trigger.entity_id }}"},
+                },
+            }
+        },
+    )
+
+
+def set_or_remove_state(
+    hass: HomeAssistant,
+    entity_id: str,
+    state: StateDescription,
+) -> None:
+    """Set or remove the state of an entity."""
+    if state["state"] is None:
+        hass.states.async_remove(entity_id)
+    else:
+        hass.states.async_set(
+            entity_id, state["state"], state["attributes"], force_update=True
+        )
+
+
+def other_states(state: StrEnum | Iterable[StrEnum]) -> list[str]:
+    """Return a sorted list with all states except the specified one."""
+    if isinstance(state, StrEnum):
+        excluded_values = {state.value}
+        enum_class = state.__class__
+    else:
+        if len(state) == 0:
+            raise ValueError("state iterable must not be empty")
+        excluded_values = {s.value for s in state}
+        enum_class = list(state)[0].__class__
+
+    return sorted({s.value for s in enum_class} - excluded_values)

{pytest_homeassistant_custom_component-0.13.289 → pytest_homeassistant_custom_component-0.13.305}/src/pytest_homeassistant_custom_component/components/recorder/common.py

@@ -15,10 +15,12 @@ from functools import partial
 import importlib
 import sys
 import time
+from types import ModuleType
 from typing import Any, Literal, cast
 from unittest.mock import MagicMock, patch, sentinel
 
 from freezegun import freeze_time
+import pytest
 from sqlalchemy import create_engine, event as sqlalchemy_event
 from sqlalchemy.orm.session import Session
 
@@ -32,18 +34,25 @@ from homeassistant.components.recorder import (
     statistics,
 )
 from homeassistant.components.recorder.db_schema import (
+    EventData,
     Events,
     EventTypes,
     RecorderRuns,
+    StateAttributes,
     States,
     StatesMeta,
 )
+from homeassistant.components.recorder.models import (
+    bytes_to_ulid_or_none,
+    bytes_to_uuid_hex_or_none,
+)
 from homeassistant.components.recorder.tasks import RecorderTask, StatisticsTask
 from homeassistant.components.sensor import SensorDeviceClass, SensorStateClass
 from homeassistant.const import DEGREE, UnitOfTemperature
 from homeassistant.core import Event, HomeAssistant, State
 from homeassistant.helpers import recorder as recorder_helper
 from homeassistant.util import dt as dt_util
+from homeassistant.util.json import json_loads, json_loads_object
 
 from . import db_schema_0
 
@@ -200,7 +209,7 @@ def statistics_during_period(
     types: set[Literal["last_reset", "max", "mean", "min", "state", "sum"]]
     | None = None,
 ) -> dict[str, list[dict[str, Any]]]:
-    """Call statistics_during_period with defaults for simpler ..."""
+    """Call statistics_during_period with defaults for simpler tests."""
     if statistic_ids is not None and not isinstance(statistic_ids, set):
         statistic_ids = set(statistic_ids)
     if types is None:
@@ -453,7 +462,14 @@ def create_engine_test_for_schema_version_postfix(
 
 def get_schema_module_path(schema_version_postfix: str) -> str:
     """Return the path to the schema module."""
-    return f"...components.recorder.db_schema_{schema_version_postfix}"
+    return f"tests.components.recorder.db_schema_{schema_version_postfix}"
+
+
+def get_patched_live_version(old_db_schema: ModuleType) -> int:
+    """Return the patched live migration version."""
+    return min(
+        migration.LIVE_MIGRATION_MIN_SCHEMA_VERSION, old_db_schema.SCHEMA_VERSION
+    )
 
 
 @contextmanager
@@ -466,6 +482,11 @@ def old_db_schema(hass: HomeAssistant, schema_version_postfix: str) -> Iterator[
     with (
         patch.object(recorder, "db_schema", old_db_schema),
         patch.object(migration, "SCHEMA_VERSION", old_db_schema.SCHEMA_VERSION),
+        patch.object(
+            migration,
+            "LIVE_MIGRATION_MIN_SCHEMA_VERSION",
+            get_patched_live_version(old_db_schema),
+        ),
         patch.object(migration, "non_live_data_migration_needed", return_value=False),
         patch.object(core, "StatesMeta", old_db_schema.StatesMeta),
         patch.object(core, "EventTypes", old_db_schema.EventTypes),
@@ -496,3 +517,97 @@ async def async_attach_db_engine(hass: HomeAssistant) -> None:
             )
 
     await instance.async_add_executor_job(_mock_setup_recorder_connection)
+
+
+EVENT_ORIGIN_ORDER = [ha.EventOrigin.local, ha.EventOrigin.remote]
+
+
+def db_event_to_native(event: Events, validate_entity_id: bool = True) -> Event | None:
+    """Convert to a native HA Event."""
+    context = ha.Context(
+        id=bytes_to_ulid_or_none(event.context_id_bin),
+        user_id=bytes_to_uuid_hex_or_none(event.context_user_id_bin),
+        parent_id=bytes_to_ulid_or_none(event.context_parent_id_bin),
+    )
+    return Event(
+        event.event_type or "",
+        json_loads_object(event.event_data) if event.event_data else {},
+        ha.EventOrigin(event.origin)
+        if event.origin
+        else EVENT_ORIGIN_ORDER[event.origin_idx or 0],
+        event.time_fired_ts or 0,
+        context=context,
+    )
+
+
+def db_event_data_to_native(event_data: EventData) -> dict[str, Any]:
+    """Convert to an event data dictionary."""
+    shared_data = event_data.shared_data
+    if shared_data is None:
+        return {}
+    return cast(dict[str, Any], json_loads(shared_data))
+
+
+def db_state_to_native(state: States, validate_entity_id: bool = True) -> State | None:
+    """Convert to an HA state object."""
+    context = ha.Context(
+        id=bytes_to_ulid_or_none(state.context_id_bin),
+        user_id=bytes_to_uuid_hex_or_none(state.context_user_id_bin),
+        parent_id=bytes_to_ulid_or_none(state.context_parent_id_bin),
+    )
+    attrs = json_loads_object(state.attributes) if state.attributes else {}
+    last_updated = dt_util.utc_from_timestamp(state.last_updated_ts or 0)
+    if state.last_changed_ts is None or state.last_changed_ts == state.last_updated_ts:
+        last_changed = dt_util.utc_from_timestamp(state.last_updated_ts or 0)
+    else:
+        last_changed = dt_util.utc_from_timestamp(state.last_changed_ts or 0)
+    if (
+        state.last_reported_ts is None
+        or state.last_reported_ts == state.last_updated_ts
+    ):
+        last_reported = dt_util.utc_from_timestamp(state.last_updated_ts or 0)
+    else:
+        last_reported = dt_util.utc_from_timestamp(state.last_reported_ts or 0)
+    return State(
+        state.entity_id or "",
+        state.state,  # type: ignore[arg-type]
+        # Join the state_attributes table on attributes_id to get the attributes
+        # for newer states
+        attrs,
+        last_changed=last_changed,
+        last_reported=last_reported,
+        last_updated=last_updated,
+        context=context,
+        validate_entity_id=validate_entity_id,
+    )
+
+
+def db_state_attributes_to_native(state_attrs: StateAttributes) -> dict[str, Any]:
+    """Convert to a state attributes dictionary."""
+    shared_attrs = state_attrs.shared_attrs
+    if shared_attrs is None:
+        return {}
+    return cast(dict[str, Any], json_loads(shared_attrs))
+
+
+async def async_drop_index(
+    recorder: Recorder, table: str, index: str, caplog: pytest.LogCaptureFixture
+) -> None:
+    """Drop an index from the database.
+
+    migration._drop_index does not return or raise, so we verify the result
+    by checking the log for success or failure messages.
+    """
+
+    finish_msg = f"Finished dropping index `{index}` from table `{table}`"
+    fail_msg = f"Failed to drop index `{index}` from table `{table}`"
+
+    count_finish = caplog.text.count(finish_msg)
+    count_fail = caplog.text.count(fail_msg)
+
+    await recorder.async_add_executor_job(
+        migration._drop_index, recorder.get_session, table, index
+    )
+
+    assert caplog.text.count(finish_msg) == count_finish + 1
+    assert caplog.text.count(fail_msg) == count_fail

{pytest_homeassistant_custom_component-0.13.289 → pytest_homeassistant_custom_component-0.13.305}/src/pytest_homeassistant_custom_component/const.py

@@ -4,9 +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 = 2025
-MINOR_VERSION: Final = 10
-PATCH_VERSION: Final = "4"
+MAJOR_VERSION: Final = 2026
+MINOR_VERSION: Final = 1
+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-0.13.289 → pytest_homeassistant_custom_component-0.13.305}/src/pytest_homeassistant_custom_component/ignore_uncaught_exceptions.py

@@ -8,25 +8,25 @@ IGNORE_UNCAUGHT_EXCEPTIONS = [
     (
         # This test explicitly throws an uncaught exception
         # and should not be removed.
-        ".test_runner",
+        "tests.test_runner",
         "test_unhandled_exception_traceback",
     ),
     (
         # This test explicitly throws an uncaught exception
         # and should not be removed.
-        ".helpers.test_event",
+        "tests.helpers.test_event",
         "test_track_point_in_time_repr",
     ),
     (
         # This test explicitly throws an uncaught exception
         # and should not be removed.
-        ".test_config_entries",
+        "tests.test_config_entries",
         "test_config_entry_unloaded_during_platform_setups",
     ),
     (
         # This test explicitly throws an uncaught exception
         # and should not be removed.
-        ".test_config_entries",
+        "tests.test_config_entries",
         "test_config_entry_unloaded_during_platform_setup",
     ),
     (
@@ -34,17 +34,17 @@ IGNORE_UNCAUGHT_EXCEPTIONS = [
         "test_homeassistant_bridge_fan_setup",
     ),
     (
-        ".components.owntracks.test_device_tracker",
+        "tests.components.owntracks.test_device_tracker",
         "test_mobile_multiple_async_enter_exit",
     ),
     (
-        ".components.smartthings.test_init",
+        "tests.components.smartthings.test_init",
         "test_event_handler_dispatches_updated_devices",
     ),
     (
-        ".components.unifi.test_controller",
+        "tests.components.unifi.test_controller",
         "test_wireless_client_event_calls_update_wireless_devices",
     ),
-    (".components.iaqualink.test_config_flow", "test_with_invalid_credentials"),
-    (".components.iaqualink.test_config_flow", "test_with_existing_config"),
+    ("tests.components.iaqualink.test_config_flow", "test_with_invalid_credentials"),
+    ("tests.components.iaqualink.test_config_flow", "test_with_existing_config"),
 ]

{pytest_homeassistant_custom_component-0.13.289 → pytest_homeassistant_custom_component-0.13.305}/src/pytest_homeassistant_custom_component/patch_time.py

@@ -30,8 +30,31 @@ def ha_datetime_to_fakedatetime(datetime) -> freezegun.api.FakeDatetime:  # type
     )
 
 
-class HAFakeDatetime(freezegun.api.FakeDatetime):  # type: ignore[name-defined]
-    """Modified to include https://github.com/spulec/freezegun/pull/424."""
+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):

{pytest_homeassistant_custom_component-0.13.289 → pytest_homeassistant_custom_component-0.13.305}/src/pytest_homeassistant_custom_component/plugins.py

@@ -131,7 +131,7 @@ if TYPE_CHECKING:
     from homeassistant.components import recorder
 
 
-pytest.register_assert_rewrite(".common")
+pytest.register_assert_rewrite("tests.common")
 
 from .common import (  # noqa: E402, isort:skip
     CLIENT_ID,
@@ -163,6 +163,7 @@ asyncio.set_event_loop_policy = lambda policy: None
 def pytest_addoption(parser: pytest.Parser) -> None:
     """Register custom pytest options."""
     parser.addoption("--dburl", action="store", default="sqlite://")
+    parser.addoption("--drop-existing-db", action="store_const", const=True)
 
 
 def pytest_configure(config: pytest.Config) -> None:
@@ -190,11 +191,13 @@ def pytest_runtest_setup() -> None:
     destinations will be allowed.
 
     freezegun:
-    Modified to include https://github.com/spulec/freezegun/pull/424
+    Modified to include https://github.com/spulec/freezegun/pull/424 and improve class str.
     """
     pytest_socket.socket_allow_hosts(["127.0.0.1"])
     pytest_socket.disable_socket(allow_unix_socket=True)
 
+    freezegun.api.FakeDate = patch_time.HAFakeDate  # type: ignore[attr-defined]
+
     freezegun.api.datetime_to_fakedatetime = patch_time.ha_datetime_to_fakedatetime  # type: ignore[attr-defined]
     freezegun.api.FakeDatetime = patch_time.HAFakeDatetime  # type: ignore[attr-defined]
 
@@ -1271,9 +1274,11 @@ def evict_faked_translations(translations_once) -> Generator[_patch]:
     component_paths = components.__path__
 
     for call in mock_component_strings.mock_calls:
+        _components: set[str] = call.args[2]
         integrations: dict[str, loader.Integration] = call.args[3]
-        for domain, integration in integrations.items():
-            if any(
+        for domain in _components:
+            # If the integration exists, don't evict from cache
+            if (integration := integrations.get(domain)) and any(
                 pathlib.Path(f"{component_path}/{domain}") == integration.file_path
                 for component_path in component_paths
             ):
@@ -1475,7 +1480,7 @@ def persistent_database() -> bool:
     When using sqlite, this uses on disk database instead of in memory database.
     This does nothing when using mysql or postgresql.
 
-    Note that the database is always destroyed in between .
+    Note that the database is always destroyed in between tests.
 
     To use a persistent database, tests can be marked with:
     @pytest.mark.parametrize("persistent_database", [True])
@@ -1494,44 +1499,58 @@ def recorder_db_url(
     assert not hass_fixture_setup
 
     db_url = cast(str, pytestconfig.getoption("dburl"))
+    drop_existing_db = pytestconfig.getoption("drop_existing_db")
+
+    def drop_db() -> None:
+        import sqlalchemy as sa  # noqa: PLC0415
+        import sqlalchemy_utils  # noqa: PLC0415
+
+        if db_url.startswith("mysql://"):
+            made_url = sa.make_url(db_url)
+            db = made_url.database
+            engine = sa.create_engine(db_url)
+            # Check for any open connections to the database before dropping it
+            # to ensure that InnoDB does not deadlock.
+            with engine.begin() as connection:
+                query = sa.text(
+                    "select id FROM information_schema.processlist WHERE db=:db and id != CONNECTION_ID()"
+                )
+                rows = connection.execute(query, parameters={"db": db}).fetchall()
+                if rows:
+                    raise RuntimeError(
+                        f"Unable to drop database {db} because it is in use by {rows}"
+                    )
+            engine.dispose()
+            sqlalchemy_utils.drop_database(db_url)
+        elif db_url.startswith("postgresql://"):
+            sqlalchemy_utils.drop_database(db_url)
+
     if db_url == "sqlite://" and persistent_database:
         tmp_path = tmp_path_factory.mktemp("recorder")
         db_url = "sqlite:///" + str(tmp_path / "pytest.db")
-    elif db_url.startswith("mysql://"):
+    elif db_url.startswith(("mysql://", "postgresql://")):
         import sqlalchemy_utils  # noqa: PLC0415
 
-        charset = "utf8mb4' COLLATE = 'utf8mb4_unicode_ci"
-        assert not sqlalchemy_utils.database_exists(db_url)
-        sqlalchemy_utils.create_database(db_url, encoding=charset)
-    elif db_url.startswith("postgresql://"):
-        import sqlalchemy_utils  # noqa: PLC0415
+        if drop_existing_db and sqlalchemy_utils.database_exists(db_url):
+            drop_db()
+
+        if sqlalchemy_utils.database_exists(db_url):
+            raise RuntimeError(
+                f"Database {db_url} already exists. Use --drop-existing-db "
+                "to automatically drop existing database before start of test."
+            )
 
-        assert not sqlalchemy_utils.database_exists(db_url)
-        sqlalchemy_utils.create_database(db_url, encoding="utf8")
+        sqlalchemy_utils.create_database(
+            db_url,
+            encoding="utf8mb4' COLLATE = 'utf8mb4_unicode_ci"
+            if db_url.startswith("mysql://")
+            else "utf8",
+        )
     yield db_url
     if db_url == "sqlite://" and persistent_database:
         rmtree(tmp_path, ignore_errors=True)
-    elif db_url.startswith("mysql://"):
-        import sqlalchemy as sa  # noqa: PLC0415
-
-        made_url = sa.make_url(db_url)
-        db = made_url.database
-        engine = sa.create_engine(db_url)
-        # Check for any open connections to the database before dropping it
-        # to ensure that InnoDB does not deadlock.
-        with engine.begin() as connection:
-            query = sa.text(
-                "select id FROM information_schema.processlist WHERE db=:db and id != CONNECTION_ID()"
-            )
-            rows = connection.execute(query, parameters={"db": db}).fetchall()
-            if rows:
-                raise RuntimeError(
-                    f"Unable to drop database {db} because it is in use by {rows}"
-                )
-        engine.dispose()
-        sqlalchemy_utils.drop_database(db_url)
-    elif db_url.startswith("postgresql://"):
-        sqlalchemy_utils.drop_database(db_url)
+    elif db_url.startswith(("mysql://", "postgresql://")):
+        drop_db()
 
 
 async def _async_init_recorder_component(
@@ -1661,10 +1680,12 @@ async def async_test_recorder(
     migrate_entity_ids = (
         migration.EntityIDMigration.migrate_data if enable_migrate_entity_ids else None
     )
-    legacy_event_id_foreign_key_exists = (
-        migration.EventIDPostMigration._legacy_event_id_foreign_key_exists
+    post_migrate_event_ids = (
+        migration.EventIDPostMigration.needs_migrate_impl
         if enable_migrate_event_ids
-        else lambda _: None
+        else lambda _1, _2, _3: migration.DataMigrationStatus(
+            needs_migrate=False, migration_done=True
+        )
     )
     with (
         patch(
@@ -1703,8 +1724,8 @@ async def async_test_recorder(
             autospec=True,
         ),
         patch(
-            "homeassistant.components.recorder.migration.EventIDPostMigration._legacy_event_id_foreign_key_exists",
-            side_effect=legacy_event_id_foreign_key_exists,
+            "homeassistant.components.recorder.migration.EventIDPostMigration.needs_migrate_impl",
+            side_effect=post_migrate_event_ids,
             autospec=True,
         ),
         patch(
@@ -2114,3 +2135,14 @@ def _dhcp_service_info_init(self: DhcpServiceInfo, *args: Any, **kwargs: Any) ->
 
 
 DhcpServiceInfo.__init__ = _dhcp_service_info_init
+
+
+@pytest.fixture(autouse=True)
+def disable_http_server() -> Generator[None]:
+    """Disable automatic start of HTTP server during tests.
+
+    This prevents the HTTP server from starting in tests that setup
+    integrations which depend on the HTTP component.
+    """
+    with patch("homeassistant.components.http.start_http_server_and_save_config"):
+        yield

{pytest_homeassistant_custom_component-0.13.289 → pytest_homeassistant_custom_component-0.13.305}/src/pytest_homeassistant_custom_component/test_util/aiohttp.py

@@ -225,8 +225,8 @@ class AiohttpClientMockResponse:
 
         if (
             self._url.scheme != url.scheme
-            or self._url.host != url.host
-            or self._url.path != url.path
+            or self._url.raw_host != url.raw_host
+            or self._url.raw_path != url.raw_path
         ):
             return False
 

{pytest_homeassistant_custom_component-0.13.289 → pytest_homeassistant_custom_component-0.13.305}/src/pytest_homeassistant_custom_component/typing.py

@@ -1,5 +1,5 @@
 """
-Typing helpers for Home Assistant .
+Typing helpers for Home Assistant tests.
 
 This file is originally from homeassistant/core and modified by pytest-homeassistant-custom-component.
 """

{pytest_homeassistant_custom_component-0.13.289 → pytest_homeassistant_custom_component-0.13.305/src/pytest_homeassistant_custom_component.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-homeassistant-custom-component
-Version: 0.13.289
+Version: 0.13.305
 Summary: Experimental package to automatically extract test plugins for Home Assistant custom components
 Home-page: https://github.com/MatthewFlamm/pytest-homeassistant-custom-component
 Author: Matthew Flamm
@@ -20,13 +20,12 @@ License-File: LICENSE_HA_CORE.md
 Requires-Dist: sqlalchemy
 Requires-Dist: coverage==7.10.6
 Requires-Dist: freezegun==1.5.2
-Requires-Dist: go2rtc-client==0.2.1
 Requires-Dist: license-expression==30.4.3
 Requires-Dist: mock-open==1.4.0
-Requires-Dist: pydantic==2.11.9
+Requires-Dist: pydantic==2.12.2
 Requires-Dist: pylint-per-file-ignores==1.4.0
 Requires-Dist: pipdeptree==2.26.1
-Requires-Dist: pytest-asyncio==1.2.0
+Requires-Dist: pytest-asyncio==1.3.0
 Requires-Dist: pytest-aiohttp==1.1.0
 Requires-Dist: pytest-cov==7.0.0
 Requires-Dist: pytest-freezer==0.4.9
@@ -37,12 +36,12 @@ Requires-Dist: pytest-timeout==2.4.0
 Requires-Dist: pytest-unordered==0.7.0
 Requires-Dist: pytest-picked==0.5.1
 Requires-Dist: pytest-xdist==3.8.0
-Requires-Dist: pytest==8.4.2
+Requires-Dist: pytest==9.0.0
 Requires-Dist: requests-mock==1.12.1
 Requires-Dist: respx==0.22.0
-Requires-Dist: syrupy==4.9.1
+Requires-Dist: syrupy==5.0.0
 Requires-Dist: tqdm==4.67.1
-Requires-Dist: homeassistant==2025.10.4
+Requires-Dist: homeassistant==2026.1.0
 Requires-Dist: SQLAlchemy==2.0.41
 Requires-Dist: paho-mqtt==2.1.0
 Requires-Dist: numpy==2.3.2
@@ -60,9 +59,7 @@ Dynamic: summary
 
 # pytest-homeassistant-custom-component
 
-![HA core version](https://img.shields.io/static/v1?label=HA+core+version&message=2025.10.4&labelColor=blue)
-
-[![Open in Gitpod](https://gitpod.io/button/open-in-gitpod.svg)](https://gitpod.io/#https://github.com/MatthewFlamm/pytest-homeassistant-custom-component)
+![HA core version](https://img.shields.io/static/v1?label=HA+core+version&message=2026.1.0&labelColor=blue)
 
 Package to automatically extract testing plugins from Home Assistant for custom component testing.
 The goal is to provide the same functionality as the tests in home-assistant/core.

{pytest_homeassistant_custom_component-0.13.289 → pytest_homeassistant_custom_component-0.13.305}/src/pytest_homeassistant_custom_component.egg-info/requires.txt

@@ -1,13 +1,12 @@
 sqlalchemy
 coverage==7.10.6
 freezegun==1.5.2
-go2rtc-client==0.2.1
 license-expression==30.4.3
 mock-open==1.4.0
-pydantic==2.11.9
+pydantic==2.12.2
 pylint-per-file-ignores==1.4.0
 pipdeptree==2.26.1
-pytest-asyncio==1.2.0
+pytest-asyncio==1.3.0
 pytest-aiohttp==1.1.0
 pytest-cov==7.0.0
 pytest-freezer==0.4.9
@@ -18,12 +17,12 @@ pytest-timeout==2.4.0
 pytest-unordered==0.7.0
 pytest-picked==0.5.1
 pytest-xdist==3.8.0
-pytest==8.4.2
+pytest==9.0.0
 requests-mock==1.12.1
 respx==0.22.0
-syrupy==4.9.1
+syrupy==5.0.0
 tqdm==4.67.1
-homeassistant==2025.10.4
+homeassistant==2026.1.0
 SQLAlchemy==2.0.41
 paho-mqtt==2.1.0
 numpy==2.3.2

pytest_homeassistant_custom_component-0.13.289/src/pytest_homeassistant_custom_component/components/__init__.py

@@ -1,5 +0,0 @@
-"""
-The tests for components.
-
-This file is originally from homeassistant/core and modified by pytest-homeassistant-custom-component.
-"""