1
0
mirror of https://github.com/home-assistant/core synced 2024-07-05 17:29:15 +00:00

Improve type hints in evil_genius_labs tests (#120677)

This commit is contained in:
epenet 2024-06-27 22:11:24 +02:00 committed by GitHub
parent 440928bcbe
commit bccd5c8c35
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,36 +1,44 @@
"""Test helpers for Evil Genius Labs."""
import json
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_fixture
from tests.common import (
MockConfigEntry,
load_json_array_fixture,
load_json_object_fixture,
)
@pytest.fixture(scope="package")
def all_fixture():
def all_fixture() -> dict[str, Any]:
"""Fixture data."""
data = json.loads(load_fixture("data.json", "evil_genius_labs"))
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():
def info_fixture() -> JsonObjectType:
"""Fixture info."""
return json.loads(load_fixture("info.json", "evil_genius_labs"))
return load_json_object_fixture("info.json", "evil_genius_labs")
@pytest.fixture(scope="package")
def product_fixture():
def product_fixture() -> dict[str, str]:
"""Fixture info."""
return {"productName": "Fibonacci256"}
@pytest.fixture
def config_entry(hass):
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)
@ -39,8 +47,13 @@ def config_entry(hass):
@pytest.fixture
async def setup_evil_genius_labs(
hass, config_entry, all_fixture, info_fixture, product_fixture, platforms
):
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(