pytest-homeassistant-custom-component 0.13.296__tar.gz → 0.13.298__tar.gz

{pytest_homeassistant_custom_component-0.13.296/src/pytest_homeassistant_custom_component.egg-info → pytest_homeassistant_custom_component-0.13.298}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-homeassistant-custom-component
-Version: 0.13.296
+Version: 0.13.298
 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,14 +20,14 @@ 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: go2rtc-client==0.3.0
 Requires-Dist: librt==0.2.1
 Requires-Dist: license-expression==30.4.3
 Requires-Dist: mock-open==1.4.0
 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
@@ -38,12 +38,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==5.0.0
 Requires-Dist: tqdm==4.67.1
-Requires-Dist: homeassistant==2025.11.2
+Requires-Dist: homeassistant==2025.12.0
 Requires-Dist: SQLAlchemy==2.0.41
 Requires-Dist: paho-mqtt==2.1.0
 Requires-Dist: numpy==2.3.2
@@ -61,7 +61,7 @@ Dynamic: summary
 
 # pytest-homeassistant-custom-component
 
-![HA core version](https://img.shields.io/static/v1?label=HA+core+version&message=2025.11.2&labelColor=blue)
+![HA core version](https://img.shields.io/static/v1?label=HA+core+version&message=2025.12.0&labelColor=blue)
 
 [![Open in Gitpod](https://gitpod.io/button/open-in-gitpod.svg)](https://gitpod.io/#https://github.com/MatthewFlamm/pytest-homeassistant-custom-component)
 

{pytest_homeassistant_custom_component-0.13.296 → pytest_homeassistant_custom_component-0.13.298}/README.md

@@ -1,6 +1,6 @@
 # pytest-homeassistant-custom-component
 
-![HA core version](https://img.shields.io/static/v1?label=HA+core+version&message=2025.11.2&labelColor=blue)
+![HA core version](https://img.shields.io/static/v1?label=HA+core+version&message=2025.12.0&labelColor=blue)
 
 [![Open in Gitpod](https://gitpod.io/button/open-in-gitpod.svg)](https://gitpod.io/#https://github.com/MatthewFlamm/pytest-homeassistant-custom-component)
 

{pytest_homeassistant_custom_component-0.13.296 → pytest_homeassistant_custom_component-0.13.298}/src/pytest_homeassistant_custom_component/common.py

@@ -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,
     )

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

@@ -0,0 +1,356 @@
+"""
+The tests for components.
+
+This file is originally from homeassistant/core and modified by pytest-homeassistant-custom-component.
+"""
+
+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.
+    """
+    await async_setup_component(hass, domain, {})
+
+    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,
+) -> 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.
+
+    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,
+        }
+
+    return [
+        # 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),
+                    )
+                    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
+                )
+            ),
+        ),
+    ]
+
+
+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) -> list[str]:
+    """Return a sorted list with all states except the specified one."""
+    return sorted({s.value for s in state.__class__} - {state.value})

{pytest_homeassistant_custom_component-0.13.296 → pytest_homeassistant_custom_component-0.13.298}/src/pytest_homeassistant_custom_component/const.py

@@ -5,8 +5,8 @@ This file is originally from homeassistant/core and modified by pytest-homeassis
 """
 from typing import TYPE_CHECKING, Final
 MAJOR_VERSION: Final = 2025
-MINOR_VERSION: Final = 11
-PATCH_VERSION: Final = "2"
+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-0.13.296 → pytest_homeassistant_custom_component-0.13.298}/src/pytest_homeassistant_custom_component/plugins.py

@@ -2133,3 +2133,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 .
+
+    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.296 → pytest_homeassistant_custom_component-0.13.298}/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.296 → pytest_homeassistant_custom_component-0.13.298/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.296
+Version: 0.13.298
 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,14 +20,14 @@ 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: go2rtc-client==0.3.0
 Requires-Dist: librt==0.2.1
 Requires-Dist: license-expression==30.4.3
 Requires-Dist: mock-open==1.4.0
 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
@@ -38,12 +38,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==5.0.0
 Requires-Dist: tqdm==4.67.1
-Requires-Dist: homeassistant==2025.11.2
+Requires-Dist: homeassistant==2025.12.0
 Requires-Dist: SQLAlchemy==2.0.41
 Requires-Dist: paho-mqtt==2.1.0
 Requires-Dist: numpy==2.3.2
@@ -61,7 +61,7 @@ Dynamic: summary
 
 # pytest-homeassistant-custom-component
 
-![HA core version](https://img.shields.io/static/v1?label=HA+core+version&message=2025.11.2&labelColor=blue)
+![HA core version](https://img.shields.io/static/v1?label=HA+core+version&message=2025.12.0&labelColor=blue)
 
 [![Open in Gitpod](https://gitpod.io/button/open-in-gitpod.svg)](https://gitpod.io/#https://github.com/MatthewFlamm/pytest-homeassistant-custom-component)
 

{pytest_homeassistant_custom_component-0.13.296 → pytest_homeassistant_custom_component-0.13.298}/src/pytest_homeassistant_custom_component.egg-info/requires.txt

@@ -1,14 +1,14 @@
 sqlalchemy
 coverage==7.10.6
 freezegun==1.5.2
-go2rtc-client==0.2.1
+go2rtc-client==0.3.0
 librt==0.2.1
 license-expression==30.4.3
 mock-open==1.4.0
 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
@@ -19,12 +19,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==5.0.0
 tqdm==4.67.1
-homeassistant==2025.11.2
+homeassistant==2025.12.0
 SQLAlchemy==2.0.41
 paho-mqtt==2.1.0
 numpy==2.3.2

pytest_homeassistant_custom_component-0.13.296/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.
-"""