From 23698eb99fcbf091bab82cbfb22cb3dfea9e17d9 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 8 Mar 2023 16:18:17 +0100 Subject: [PATCH] Add missing mock in agent_dvr config flow tests (#89361) --- tests/components/agent_dvr/conftest.py | 14 ++++++++++++++ tests/components/agent_dvr/test_config_flow.py | 4 ++++ 2 files changed, 18 insertions(+) create mode 100644 tests/components/agent_dvr/conftest.py diff --git a/tests/components/agent_dvr/conftest.py b/tests/components/agent_dvr/conftest.py new file mode 100644 index 00000000000..da2cd90ed18 --- /dev/null +++ b/tests/components/agent_dvr/conftest.py @@ -0,0 +1,14 @@ +"""Test fixtures for Agent DVR.""" +from collections.abc import Generator +from unittest.mock import AsyncMock, patch + +import pytest + + +@pytest.fixture +def mock_setup_entry() -> Generator[AsyncMock, None, None]: + """Override async_setup_entry.""" + with patch( + "homeassistant.components.agent_dvr.async_setup_entry", return_value=True + ) as mock_setup_entry: + yield mock_setup_entry diff --git a/tests/components/agent_dvr/test_config_flow.py b/tests/components/agent_dvr/test_config_flow.py index b36044e45b1..9d1be278ada 100644 --- a/tests/components/agent_dvr/test_config_flow.py +++ b/tests/components/agent_dvr/test_config_flow.py @@ -1,4 +1,6 @@ """Tests for the Agent DVR config flow.""" +import pytest + from homeassistant import data_entry_flow from homeassistant.components.agent_dvr import config_flow from homeassistant.components.agent_dvr.const import SERVER_URL @@ -11,6 +13,8 @@ from . import init_integration from tests.common import load_fixture from tests.test_util.aiohttp import AiohttpClientMocker +pytestmark = pytest.mark.usefixtures("mock_setup_entry") + async def test_show_user_form(hass: HomeAssistant) -> None: """Test that the user set up form is served."""