diff --git a/tests/components/guardian/conftest.py b/tests/components/guardian/conftest.py index c7bffee4fff..acf59aeea86 100644 --- a/tests/components/guardian/conftest.py +++ b/tests/components/guardian/conftest.py @@ -1,6 +1,7 @@ """Define fixtures for Elexa Guardian tests.""" +from collections.abc import Generator import json -from unittest.mock import patch +from unittest.mock import AsyncMock, patch import pytest @@ -11,6 +12,15 @@ from homeassistant.setup import async_setup_component from tests.common import MockConfigEntry, load_fixture +@pytest.fixture +def mock_setup_entry() -> Generator[AsyncMock, None, None]: + """Override async_setup_entry.""" + with patch( + "homeassistant.components.guardian.async_setup_entry", return_value=True + ) as mock_setup_entry: + yield mock_setup_entry + + @pytest.fixture(name="config_entry") def config_entry_fixture(hass, config, unique_id): """Define a config entry fixture.""" diff --git a/tests/components/guardian/test_config_flow.py b/tests/components/guardian/test_config_flow.py index 2fdacdf29a7..cb28ea22a37 100644 --- a/tests/components/guardian/test_config_flow.py +++ b/tests/components/guardian/test_config_flow.py @@ -2,6 +2,7 @@ from unittest.mock import patch from aioguardian.errors import GuardianError +import pytest from homeassistant import data_entry_flow from homeassistant.components import dhcp, zeroconf @@ -16,6 +17,8 @@ from homeassistant.core import HomeAssistant from tests.common import MockConfigEntry +pytestmark = pytest.mark.usefixtures("mock_setup_entry") + async def test_duplicate_error( hass: HomeAssistant, config, config_entry, setup_guardian