jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-15 17:27:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve type hints in rainforest_eagle tests (#123880)

Browse Source
This commit is contained in:
epenet 2024-08-14 12:58:49 +02:00 committed by GitHub
parent ac223e64f9
commit 8117532cc7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/components/rainforest_eagle/conftest.py
View File

@ -1,6 +1,7 @@
"""Conftest for rainforest_eagle."""
from unittest.mock import AsyncMock, Mock, patch
from collections.abc import AsyncGenerator
from unittest.mock import AsyncMock, MagicMock, Mock, patch
import pytest
@ -13,6 +14,7 @@ from homeassistant.components.rainforest_eagle.const import (
TYPE_EAGLE_200,
)
from homeassistant.const import CONF_HOST, CONF_TYPE
from homeassistant.core import HomeAssistant
from homeassistant.setup import async_setup_component
from . import MOCK_200_RESPONSE_WITHOUT_PRICE, MOCK_CLOUD_ID
@ -21,7 +23,7 @@ from tests.common import MockConfigEntry
@pytest.fixture
def config_entry_200(hass):
def config_entry_200(hass: HomeAssistant) -> MockConfigEntry:
"""Return a config entry."""
entry = MockConfigEntry(
domain="rainforest_eagle",
@ -38,7 +40,9 @@ def config_entry_200(hass):
@pytest.fixture
async def setup_rainforest_200(hass, config_entry_200):
async def setup_rainforest_200(
hass: HomeAssistant, config_entry_200: MockConfigEntry
) -> AsyncGenerator[Mock]:
"""Set up rainforest."""
with patch(
"aioeagle.ElectricMeter.create_instance",
@ -53,7 +57,7 @@ async def setup_rainforest_200(hass, config_entry_200):
@pytest.fixture
async def setup_rainforest_100(hass):
async def setup_rainforest_100(hass: HomeAssistant) -> AsyncGenerator[MagicMock]:
"""Set up rainforest."""
MockConfigEntry(
domain="rainforest_eagle",