Refactor EZVIZ config flow tests (#136434)

tests/components/ezviz/__init__.py

@@ -1,102 +1,13 @@
 """Tests for the EZVIZ integration."""
 
-from unittest.mock import _patch, patch
-
-from homeassistant.components.ezviz.const import (
-    ATTR_SERIAL,
-    ATTR_TYPE_CAMERA,
-    ATTR_TYPE_CLOUD,
-    CONF_FFMPEG_ARGUMENTS,
-    CONF_RFSESSION_ID,
-    CONF_SESSION_ID,
-    DEFAULT_FFMPEG_ARGUMENTS,
-    DEFAULT_TIMEOUT,
-    DOMAIN,
-)
-from homeassistant.const import (
-    CONF_IP_ADDRESS,
-    CONF_PASSWORD,
-    CONF_TIMEOUT,
-    CONF_TYPE,
-    CONF_URL,
-    CONF_USERNAME,
-)
 from homeassistant.core import HomeAssistant
 
 from tests.common import MockConfigEntry
 
-ENTRY_CONFIG = {
-    CONF_SESSION_ID: "test-username",
-    CONF_RFSESSION_ID: "test-password",
-    CONF_URL: "apiieu.ezvizlife.com",
-    CONF_TYPE: ATTR_TYPE_CLOUD,
-}
 
-ENTRY_OPTIONS = {
-    CONF_FFMPEG_ARGUMENTS: DEFAULT_FFMPEG_ARGUMENTS,
-    CONF_TIMEOUT: DEFAULT_TIMEOUT,
-}
-
-USER_INPUT_VALIDATE = {
-    CONF_USERNAME: "test-username",
-    CONF_PASSWORD: "test-password",
-    CONF_URL: "apiieu.ezvizlife.com",
-}
-
-USER_INPUT = {
-    CONF_USERNAME: "test-username",
-    CONF_PASSWORD: "test-password",
-    CONF_URL: "apiieu.ezvizlife.com",
-    CONF_TYPE: ATTR_TYPE_CLOUD,
-}
-
-USER_INPUT_CAMERA_VALIDATE = {
-    ATTR_SERIAL: "C666666",
-    CONF_PASSWORD: "test-password",
-    CONF_USERNAME: "test-username",
-}
-
-USER_INPUT_CAMERA = {
-    CONF_PASSWORD: "test-password",
-    CONF_USERNAME: "test-username",
-    CONF_TYPE: ATTR_TYPE_CAMERA,
-}
-
-DISCOVERY_INFO = {
-    ATTR_SERIAL: "C666666",
-    CONF_USERNAME: None,
-    CONF_PASSWORD: None,
-    CONF_IP_ADDRESS: "127.0.0.1",
-}
-
-TEST = {
-    CONF_USERNAME: None,
-    CONF_PASSWORD: None,
-    CONF_IP_ADDRESS: "127.0.0.1",
-}
-
-API_LOGIN_RETURN_VALIDATE = {
-    CONF_SESSION_ID: "fake_token",
-    CONF_RFSESSION_ID: "fake_rf_token",
-    CONF_URL: "apiieu.ezvizlife.com",
-    CONF_TYPE: ATTR_TYPE_CLOUD,
-}
-
-
-def patch_async_setup_entry() -> _patch:
-    """Patch async_setup_entry."""
-    return patch(
-        "homeassistant.components.ezviz.async_setup_entry",
-        return_value=True,
-    )
-
-
-async def init_integration(hass: HomeAssistant) -> MockConfigEntry:
+async def setup_integration(hass: HomeAssistant, entry: MockConfigEntry) -> None:
     """Set up the EZVIZ integration in Home Assistant."""
-    entry = MockConfigEntry(domain=DOMAIN, data=ENTRY_CONFIG, options=ENTRY_OPTIONS)
     entry.add_to_hass(hass)
 
     await hass.config_entries.async_setup(entry.entry_id)
     await hass.async_block_till_done()
-
-    return entry

tests/components/ezviz/conftest.py

@@ -1,19 +1,30 @@
 """Define pytest.fixtures available for all tests."""
 
 from collections.abc import Generator
-from unittest.mock import MagicMock, patch
+from unittest.mock import AsyncMock, MagicMock, patch
 
-from pyezviz import EzvizClient
-from pyezviz.test_cam_rtsp import TestRTSPAuth
 import pytest
 
+from homeassistant.components.ezviz import (
+    ATTR_TYPE_CAMERA,
+    ATTR_TYPE_CLOUD,
+    CONF_RFSESSION_ID,
+    CONF_SESSION_ID,
+    DOMAIN,
+)
+from homeassistant.const import CONF_PASSWORD, CONF_TYPE, CONF_URL, CONF_USERNAME
 from homeassistant.core import HomeAssistant
 
-ezviz_login_token_return = {
-    "session_id": "fake_token",
-    "rf_session_id": "fake_rf_token",
-    "api_url": "apiieu.ezvizlife.com",
-}
+from tests.common import MockConfigEntry
+
+
+@pytest.fixture
+def mock_setup_entry() -> Generator[AsyncMock]:
+    """Mock setting up a config entry."""
+    with patch(
+        "homeassistant.components.ezviz.async_setup_entry", return_value=True
+    ) as setup_entry_mock:
+        yield setup_entry_mock
 
 
 @pytest.fixture(autouse=True)
@@ -23,40 +34,67 @@ def mock_ffmpeg(hass: HomeAssistant) -> None:
 
 
 @pytest.fixture
-def ezviz_test_rtsp_config_flow() -> Generator[MagicMock]:
+def mock_config_entry() -> MockConfigEntry:
+    """Return the default mocked config entry."""
+    return MockConfigEntry(
+        domain=DOMAIN,
+        unique_id="test-username",
+        title="test-username",
+        data={
+            CONF_SESSION_ID: "test-username",
+            CONF_RFSESSION_ID: "test-password",
+            CONF_URL: "apiieu.ezvizlife.com",
+            CONF_TYPE: ATTR_TYPE_CLOUD,
+        },
+    )
+
+
+@pytest.fixture
+def mock_camera_config_entry() -> MockConfigEntry:
+    """Return the default mocked config entry."""
+    return MockConfigEntry(
+        domain=DOMAIN,
+        unique_id="C666666",
+        title="Camera 1",
+        data={
+            CONF_TYPE: ATTR_TYPE_CAMERA,
+            CONF_USERNAME: "test-username",
+            CONF_PASSWORD: "test-password",
+        },
+    )
+
+
+@pytest.fixture
+def mock_ezviz_client() -> Generator[AsyncMock]:
+    """Mock the EzvizAPI for easier testing."""
+    with (
+        patch(
+            "homeassistant.components.ezviz.EzvizClient", autospec=True
+        ) as mock_ezviz,
+        patch("homeassistant.components.ezviz.config_flow.EzvizClient", new=mock_ezviz),
+    ):
+        instance = mock_ezviz.return_value
+
+        instance.login.return_value = {
+            "session_id": "fake_token",
+            "rf_session_id": "fake_rf_token",
+            "api_url": "apiieu.ezvizlife.com",
+        }
+        instance.get_detection_sensibility.return_value = True
+
+        yield instance
+
+
+@pytest.fixture
+def mock_test_rtsp_auth() -> Generator[MagicMock]:
     """Mock the EzvizApi for easier testing."""
     with (
-        patch.object(TestRTSPAuth, "main", return_value=True),
         patch(
             "homeassistant.components.ezviz.config_flow.TestRTSPAuth"
         ) as mock_ezviz_test_rtsp,
     ):
-        instance = mock_ezviz_test_rtsp.return_value = TestRTSPAuth(
-            "test-ip",
-            "test-username",
-            "test-password",
-        )
+        instance = mock_ezviz_test_rtsp.return_value
 
-        instance.main = MagicMock(return_value=True)
+        instance.main.return_value = True
 
-        yield mock_ezviz_test_rtsp
-
-
-@pytest.fixture
-def ezviz_config_flow() -> Generator[MagicMock]:
-    """Mock the EzvizAPI for easier config flow testing."""
-    with (
-        patch.object(EzvizClient, "login", return_value=True),
-        patch("homeassistant.components.ezviz.config_flow.EzvizClient") as mock_ezviz,
-    ):
-        instance = mock_ezviz.return_value = EzvizClient(
-            "test-username",
-            "test-password",
-            "local.host",
-            "1",
-        )
-
-        instance.login = MagicMock(return_value=ezviz_login_token_return)
-        instance.get_detection_sensibility = MagicMock(return_value=True)
-
-        yield mock_ezviz
+        yield instance