Reset Reolink host Mock for every test

tests/components/reolink/conftest.py

@@ -71,6 +71,16 @@ def reolink_connect_class() -> Generator[MagicMock]:
         ) as host_mock_class,
     ):
         host_mock = host_mock_class.return_value
+        host_mock.baichuan = create_autospec(Baichuan)
+        yield host_mock_class
+
+
+@pytest.fixture
+def reolink_connect(
+    reolink_connect_class: MagicMock,
+) -> Generator[MagicMock]:
+    """Mock reolink connection."""
+    host_mock = reolink_connect_class.return_value
     host_mock.get_host_data.return_value = None
     host_mock.get_states.return_value = None
     host_mock.supported.return_value = True
@@ -142,7 +152,6 @@ def reolink_connect_class() -> Generator[MagicMock]:
     host_mock.recording_packing_time = "60 Minutes"
 
     # Baichuan
-        host_mock.baichuan = create_autospec(Baichuan)
     host_mock.baichuan_only = False
     # Disable tcp push by default for tests
     host_mock.baichuan.port = TEST_BC_PORT
@@ -161,16 +170,7 @@ def reolink_connect_class() -> Generator[MagicMock]:
     host_mock.baichuan.smart_ai_type_list.return_value = ["people"]
     host_mock.baichuan.smart_ai_index.return_value = 1
     host_mock.baichuan.smart_ai_name.return_value = "zone1"
-
-        yield host_mock_class
-
-
-@pytest.fixture
-def reolink_connect(
-    reolink_connect_class: MagicMock,
-) -> Generator[MagicMock]:
-    """Mock reolink connection."""
-    return reolink_connect_class.return_value
+    return host_mock
 
 
 @pytest.fixture

tests/components/reolink/test_media_source.py

@@ -141,6 +141,7 @@ async def test_browsing(
     entry_id = config_entry.entry_id
     reolink_connect.supported.return_value = 1
     reolink_connect.model = "Reolink TrackMix PoE"
+    reolink_connect.is_nvr = False
 
     with patch("homeassistant.components.reolink.PLATFORMS", [Platform.CAMERA]):
         assert await hass.config_entries.async_setup(entry_id) is True