github/ha-core
1
Fork 0
mirror of https://github.com/home-assistant/core synced 2024-09-15 17:29:45 +02:00
Code Releases Activity

Add missing mock in airvisual_pro config flow tests (#89362)

Browse Source
This commit is contained in:
epenet 2023-03-08 16:18:24 +01:00 committed by GitHub
parent 23698eb99f
commit 3a40f5f35b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/components/airvisual_pro/conftest.py
View File

@ -1,4 +1,5 @@
"""Define test fixtures for AirVisual Pro."""
from collections.abc import Generator
import json
from unittest.mock import AsyncMock, Mock, patch
@ -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.airvisual_pro.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):
"""Define a config entry fixture."""

2
tests/components/airvisual_pro/test_config_flow.py
View File

@ -14,6 +14,8 @@ from homeassistant.config_entries import SOURCE_IMPORT, SOURCE_REAUTH, SOURCE_US
from homeassistant.const import CONF_IP_ADDRESS, CONF_PASSWORD
from homeassistant.core import HomeAssistant
pytestmark = pytest.mark.usefixtures("mock_setup_entry")
@pytest.mark.parametrize(
("connect_mock", "connect_errors"),