core/tests/components/teslemetry/__init__.py

"""Tests for the Teslemetry integration."""

import time
from unittest.mock import patch

from syrupy.assertion import SnapshotAssertion

from homeassistant.components.teslemetry.const import DOMAIN
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
from homeassistant.helpers import entity_registry as er

from tests.common import MockConfigEntry


def mock_config_entry() -> MockConfigEntry:
    """Create a mock config entry."""

    return MockConfigEntry(
        domain=DOMAIN,
        version=2,
        unique_id="abc-123",
        data={
            "auth_implementation": DOMAIN,
            "token": {
                "access_token": "test_access_token",
                "refresh_token": "test_refresh_token",
                "expires_at": int(time.time()) + 3600,
            },
        },
    )


async def setup_platform(
    hass: HomeAssistant,
    platforms: list[Platform] | None = None,
) -> MockConfigEntry:
    """Set up the Teslemetry platform."""

    mock_entry = mock_config_entry()
    mock_entry.add_to_hass(hass)

    if platforms is None:
        await hass.config_entries.async_setup(mock_entry.entry_id)
    else:
        with patch("homeassistant.components.teslemetry.PLATFORMS", platforms):
            await hass.config_entries.async_setup(mock_entry.entry_id)
    await hass.async_block_till_done()

    return mock_entry


async def reload_platform(
    hass: HomeAssistant, entry: MockConfigEntry, platforms: list[Platform] | None = None
):
    """Reload the Teslemetry platform."""

    if platforms is None:
        await hass.config_entries.async_reload(entry.entry_id)
    else:
        with patch("homeassistant.components.teslemetry.PLATFORMS", platforms):
            await hass.config_entries.async_reload(entry.entry_id)
    await hass.async_block_till_done()


def assert_entities(
    hass: HomeAssistant,
    entry_id: str,
    entity_registry: er.EntityRegistry,
    snapshot: SnapshotAssertion,
) -> None:
    """Test that all entities match their snapshot."""

    entity_entries = er.async_entries_for_config_entry(entity_registry, entry_id)

    assert entity_entries
    for entity_entry in entity_entries:
        assert entity_entry == snapshot(name=f"{entity_entry.entity_id}-entry")
        assert (state := hass.states.get(entity_entry.entity_id))
        assert state == snapshot(name=f"{entity_entry.entity_id}-state")


def assert_entities_alt(
    hass: HomeAssistant,
    entry_id: str,
    entity_registry: er.EntityRegistry,
    snapshot: SnapshotAssertion,
) -> None:
    """Test that all entities match their alt snapshot.

    The `_alt` test variants use VEHICLE_DATA_ALT fixture data to verify
    entity behavior with alternative vehicle state values (different charge
    levels, door states, climate settings, etc.). This ensures entities
    handle varied data correctly.
    """
    entity_entries = er.async_entries_for_config_entry(entity_registry, entry_id)

    assert entity_entries
    for entity_entry in entity_entries:
        assert (state := hass.states.get(entity_entry.entity_id))
        assert state == snapshot(name=f"{entity_entry.entity_id}-statealt")