Improve type hints in sma tests (#123889)

This commit is contained in:
epenet
2024-08-14 12:49:53 +02:00
committed by GitHub
parent 57902fed22
commit f8879a51fe

View File

@@ -9,6 +9,7 @@ import pytest
from homeassistant import config_entries
from homeassistant.components.sma.const import DOMAIN
from homeassistant.core import HomeAssistant
from . import MOCK_DEVICE, MOCK_USER_INPUT
@@ -16,7 +17,7 @@ from tests.common import MockConfigEntry
@pytest.fixture
def mock_config_entry():
def mock_config_entry() -> MockConfigEntry:
"""Return the default mocked config entry."""
return MockConfigEntry(
domain=DOMAIN,
@@ -28,7 +29,9 @@ def mock_config_entry():
@pytest.fixture
async def init_integration(hass, mock_config_entry):
async def init_integration(
hass: HomeAssistant, mock_config_entry: MockConfigEntry
) -> MockConfigEntry:
"""Create a fake SMA Config Entry."""
mock_config_entry.add_to_hass(hass)