From b61ad43144d86f3518fff513f5ccdaa2bb11cf7e Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 8 Mar 2023 16:17:45 +0100 Subject: [PATCH] Add missing mock in amber config flow tests (#89358) --- tests/components/amberelectric/conftest.py | 14 ++++++++++++++ tests/components/amberelectric/test_config_flow.py | 2 ++ 2 files changed, 16 insertions(+) create mode 100644 tests/components/amberelectric/conftest.py diff --git a/tests/components/amberelectric/conftest.py b/tests/components/amberelectric/conftest.py new file mode 100644 index 00000000000..f7d7d6623e1 --- /dev/null +++ b/tests/components/amberelectric/conftest.py @@ -0,0 +1,14 @@ +"""Provide common Amber fixtures.""" +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.amberelectric.async_setup_entry", return_value=True + ) as mock_setup_entry: + yield mock_setup_entry diff --git a/tests/components/amberelectric/test_config_flow.py b/tests/components/amberelectric/test_config_flow.py index 2be77f19bf1..6325282aff8 100644 --- a/tests/components/amberelectric/test_config_flow.py +++ b/tests/components/amberelectric/test_config_flow.py @@ -20,6 +20,8 @@ from homeassistant.core import HomeAssistant API_KEY = "psk_123456789" +pytestmark = pytest.mark.usefixtures("mock_setup_entry") + @pytest.fixture(name="invalid_key_api") def mock_invalid_key_api() -> Generator: