From dac3c7179fb7c972beb5f38d24e8db938ca86351 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 23 Feb 2023 16:22:39 +0100 Subject: [PATCH] Add missing async_setup_entry mock in openuv (#88661) --- tests/components/openuv/conftest.py | 10 ++++++++++ tests/components/openuv/test_config_flow.py | 3 +++ 2 files changed, 13 insertions(+) diff --git a/tests/components/openuv/conftest.py b/tests/components/openuv/conftest.py index 7d9e8b9a4fd..0f59c6279fb 100644 --- a/tests/components/openuv/conftest.py +++ b/tests/components/openuv/conftest.py @@ -1,4 +1,5 @@ """Define test fixtures for OpenUV.""" +from collections.abc import Generator import json from unittest.mock import AsyncMock, Mock, patch @@ -20,6 +21,15 @@ TEST_LATITUDE = 51.528308 TEST_LONGITUDE = -0.3817765 +@pytest.fixture +def mock_setup_entry() -> Generator[AsyncMock, None, None]: + """Override async_setup_entry.""" + with patch( + "homeassistant.components.openuv.async_setup_entry", return_value=True + ) as mock_setup_entry: + yield mock_setup_entry + + @pytest.fixture(name="client") def client_fixture(data_protection_window, data_uv_index): """Define a mock Client object.""" diff --git a/tests/components/openuv/test_config_flow.py b/tests/components/openuv/test_config_flow.py index 7b5a76c9ace..ddc7d3ce85d 100644 --- a/tests/components/openuv/test_config_flow.py +++ b/tests/components/openuv/test_config_flow.py @@ -2,6 +2,7 @@ from unittest.mock import AsyncMock, patch from pyopenuv.errors import InvalidApiKeyError +import pytest import voluptuous as vol from homeassistant import data_entry_flow @@ -17,6 +18,8 @@ from homeassistant.core import HomeAssistant from .conftest import TEST_API_KEY, TEST_ELEVATION, TEST_LATITUDE, TEST_LONGITUDE +pytestmark = pytest.mark.usefixtures("mock_setup_entry") + async def test_create_entry(hass: HomeAssistant, client, config, mock_pyopenuv) -> None: """Test creating an entry."""