core/tests/components/mazda/__init__.py

"""Tests for the Mazda Connected Services integration."""

import json
from unittest.mock import AsyncMock, MagicMock, patch

from pymazda import Client as MazdaAPI

from homeassistant.components.mazda.const import DOMAIN
from homeassistant.const import CONF_EMAIL, CONF_PASSWORD, CONF_REGION
from homeassistant.core import HomeAssistant
from homeassistant.helpers import aiohttp_client

from tests.common import MockConfigEntry, load_fixture

FIXTURE_USER_INPUT = {
    CONF_EMAIL: "example@example.com",
    CONF_PASSWORD: "password",
    CONF_REGION: "MNAO",
}


async def init_integration(hass: HomeAssistant, use_nickname=True) -> MockConfigEntry:
    """Set up the Mazda Connected Services integration in Home Assistant."""
    get_vehicles_fixture = json.loads(load_fixture("mazda/get_vehicles.json"))
    if not use_nickname:
        get_vehicles_fixture[0].pop("nickname")

    get_vehicle_status_fixture = json.loads(
        load_fixture("mazda/get_vehicle_status.json")
    )

    config_entry = MockConfigEntry(domain=DOMAIN, data=FIXTURE_USER_INPUT)
    config_entry.add_to_hass(hass)

    client_mock = MagicMock(
        MazdaAPI(
            FIXTURE_USER_INPUT[CONF_EMAIL],
            FIXTURE_USER_INPUT[CONF_PASSWORD],
            FIXTURE_USER_INPUT[CONF_REGION],
            aiohttp_client.async_get_clientsession(hass),
        )
    )
    client_mock.get_vehicles = AsyncMock(return_value=get_vehicles_fixture)
    client_mock.get_vehicle_status = AsyncMock(return_value=get_vehicle_status_fixture)
    client_mock.lock_doors = AsyncMock()
    client_mock.unlock_doors = AsyncMock()
    client_mock.send_poi = AsyncMock()
    client_mock.start_charging = AsyncMock()
    client_mock.start_engine = AsyncMock()
    client_mock.stop_charging = AsyncMock()
    client_mock.stop_engine = AsyncMock()
    client_mock.turn_off_hazard_lights = AsyncMock()
    client_mock.turn_on_hazard_lights = AsyncMock()

    with patch(
        "homeassistant.components.mazda.config_flow.MazdaAPI",
        return_value=client_mock,
    ), patch("homeassistant.components.mazda.MazdaAPI", return_value=client_mock):
        assert await hass.config_entries.async_setup(config_entry.entry_id)
        await hass.async_block_till_done()

    return client_mock