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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

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)