"""Fixtures for Tessie."""

from __future__ import annotations

from copy import deepcopy
from unittest.mock import patch

import pytest

from .common import (
    COMMAND_OK,
    LIVE_STATUS,
    PRODUCTS,
    SCOPES,
    SITE_INFO,
    TEST_STATE_OF_ALL_VEHICLES,
    TEST_VEHICLE_STATE_ONLINE,
    TEST_VEHICLE_STATUS_AWAKE,
)

# Tessie


@pytest.fixture(autouse=True)
def mock_get_state():
    """Mock get_state function."""
    with patch(
        "homeassistant.components.tessie.coordinator.get_state",
        return_value=TEST_VEHICLE_STATE_ONLINE,
    ) as mock_get_state:
        yield mock_get_state


@pytest.fixture(autouse=True)
def mock_get_status():
    """Mock get_status function."""
    with patch(
        "homeassistant.components.tessie.coordinator.get_status",
        return_value=TEST_VEHICLE_STATUS_AWAKE,
    ) as mock_get_status:
        yield mock_get_status


@pytest.fixture(autouse=True)
def mock_get_state_of_all_vehicles():
    """Mock get_state_of_all_vehicles function."""
    with patch(
        "homeassistant.components.tessie.get_state_of_all_vehicles",
        return_value=TEST_STATE_OF_ALL_VEHICLES,
    ) as mock_get_state_of_all_vehicles:
        yield mock_get_state_of_all_vehicles


# Fleet API
@pytest.fixture(autouse=True)
def mock_scopes():
    """Mock scopes function."""
    with patch(
        "homeassistant.components.tessie.Tessie.scopes",
        return_value=SCOPES,
    ) as mock_scopes:
        yield mock_scopes


@pytest.fixture(autouse=True)
def mock_products():
    """Mock Tesla Fleet Api products method."""
    with patch(
        "homeassistant.components.tessie.Tessie.products", return_value=PRODUCTS
    ) as mock_products:
        yield mock_products


@pytest.fixture(autouse=True)
def mock_request():
    """Mock Tesla Fleet API request method."""
    with patch(
        "homeassistant.components.tessie.Tessie._request",
        return_value=COMMAND_OK,
    ) as mock_request:
        yield mock_request


@pytest.fixture(autouse=True)
def mock_live_status():
    """Mock Tesla Fleet API EnergySpecific live_status method."""
    with patch(
        "homeassistant.components.tessie.EnergySpecific.live_status",
        side_effect=lambda: deepcopy(LIVE_STATUS),
    ) as mock_live_status:
        yield mock_live_status


@pytest.fixture(autouse=True)
def mock_site_info():
    """Mock Tesla Fleet API EnergySpecific site_info method."""
    with patch(
        "homeassistant.components.tessie.EnergySpecific.site_info",
        side_effect=lambda: deepcopy(SITE_INFO),
    ) as mock_live_status:
        yield mock_live_status