78 lines
2.1 KiB
Python
78 lines
2.1 KiB
Python
"""Test helpers for Evil Genius Labs."""
|
|
|
|
from typing import Any
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from typing_extensions import AsyncGenerator
|
|
|
|
from homeassistant.const import Platform
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.setup import async_setup_component
|
|
from homeassistant.util.json import JsonObjectType
|
|
|
|
from tests.common import (
|
|
MockConfigEntry,
|
|
load_json_array_fixture,
|
|
load_json_object_fixture,
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="package")
|
|
def all_fixture() -> dict[str, Any]:
|
|
"""Fixture data."""
|
|
data = load_json_array_fixture("data.json", "evil_genius_labs")
|
|
return {item["name"]: item for item in data}
|
|
|
|
|
|
@pytest.fixture(scope="package")
|
|
def info_fixture() -> JsonObjectType:
|
|
"""Fixture info."""
|
|
return load_json_object_fixture("info.json", "evil_genius_labs")
|
|
|
|
|
|
@pytest.fixture(scope="package")
|
|
def product_fixture() -> dict[str, str]:
|
|
"""Fixture info."""
|
|
return {"productName": "Fibonacci256"}
|
|
|
|
|
|
@pytest.fixture
|
|
def config_entry(hass: HomeAssistant) -> MockConfigEntry:
|
|
"""Evil genius labs config entry."""
|
|
entry = MockConfigEntry(domain="evil_genius_labs", data={"host": "192.168.1.113"})
|
|
entry.add_to_hass(hass)
|
|
return entry
|
|
|
|
|
|
@pytest.fixture
|
|
async def setup_evil_genius_labs(
|
|
hass: HomeAssistant,
|
|
config_entry: MockConfigEntry,
|
|
all_fixture: dict[str, Any],
|
|
info_fixture: JsonObjectType,
|
|
product_fixture: dict[str, str],
|
|
platforms: list[Platform],
|
|
) -> AsyncGenerator[None]:
|
|
"""Test up Evil Genius Labs instance."""
|
|
with (
|
|
patch(
|
|
"pyevilgenius.EvilGeniusDevice.get_all",
|
|
return_value=all_fixture,
|
|
),
|
|
patch(
|
|
"pyevilgenius.EvilGeniusDevice.get_info",
|
|
return_value=info_fixture,
|
|
),
|
|
patch(
|
|
"pyevilgenius.EvilGeniusDevice.get_product",
|
|
return_value=product_fixture,
|
|
),
|
|
patch(
|
|
"homeassistant.components.evil_genius_labs.PLATFORMS",
|
|
platforms,
|
|
),
|
|
):
|
|
assert await async_setup_component(hass, "evil_genius_labs", {})
|
|
await hass.async_block_till_done()
|
|
yield
|