diff --git a/tests/components/ipma/conftest.py b/tests/components/ipma/conftest.py index 7f3e82a8819..8f2a017dcb8 100644 --- a/tests/components/ipma/conftest.py +++ b/tests/components/ipma/conftest.py @@ -14,7 +14,7 @@ from tests.common import MockConfigEntry @pytest.fixture -def config_entry(hass): +def config_entry(hass: HomeAssistant) -> MockConfigEntry: """Define a config entry fixture.""" entry = MockConfigEntry( domain=DOMAIN, diff --git a/tests/components/ipma/test_config_flow.py b/tests/components/ipma/test_config_flow.py index eba11b02ca0..2a4c3517b2a 100644 --- a/tests/components/ipma/test_config_flow.py +++ b/tests/components/ipma/test_config_flow.py @@ -14,6 +14,8 @@ from homeassistant.data_entry_flow import FlowResultType from . import MockLocation +from tests.common import MockConfigEntry + @pytest.fixture(name="ipma_setup", autouse=True) def ipma_setup_fixture() -> Generator[None]: @@ -93,7 +95,9 @@ async def test_config_flow_failures(hass: HomeAssistant) -> None: } -async def test_flow_entry_already_exists(hass: HomeAssistant, init_integration) -> None: +async def test_flow_entry_already_exists( + hass: HomeAssistant, init_integration: MockConfigEntry +) -> None: """Test user input for config_entry that already exists. Test when the form should show when user puts existing location diff --git a/tests/components/ipma/test_diagnostics.py b/tests/components/ipma/test_diagnostics.py index b7d421a2ee5..26e54454947 100644 --- a/tests/components/ipma/test_diagnostics.py +++ b/tests/components/ipma/test_diagnostics.py @@ -4,6 +4,7 @@ from syrupy.assertion import SnapshotAssertion from homeassistant.core import HomeAssistant +from tests.common import MockConfigEntry from tests.components.diagnostics import get_diagnostics_for_config_entry from tests.typing import ClientSessionGenerator @@ -11,7 +12,7 @@ from tests.typing import ClientSessionGenerator async def test_diagnostics( hass: HomeAssistant, hass_client: ClientSessionGenerator, - init_integration, + init_integration: MockConfigEntry, snapshot: SnapshotAssertion, ) -> None: """Test diagnostics."""