Reset Reolink host Mock for every test

This commit is contained in:
abmantis
2025-06-12 23:50:07 +01:00
parent 00199013e7
commit b06d86d085

View File

@@ -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