Avoid mocking EZSP when testing multicast

zigpy · Jul 24, 2024 · 1f61799 · 1f61799
1 parent 0622d1a
commit 1f61799
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/tests/test_multicast.py b/tests/test_multicast.py
@@ -7,16 +7,20 @@
 import bellows.multicast
 import bellows.types as t
 
+from tests.common import mock_ezsp_commands
+
 CUSTOM_SIZE = 12
 
 
 @pytest.fixture
 def ezsp_f():
-    e = MagicMock()
-    e.getConfigurationValue = AsyncMock(
-        return_value=[t.EmberStatus.SUCCESS, CUSTOM_SIZE]
-    )
-    return e
+    """EZSP v8 protocol handler."""
+    ezsp = bellows.ezsp.v8.EZSPv8(MagicMock(), MagicMock())
+    mock_ezsp_commands(ezsp)
+
+    ezsp.getConfigurationValue.return_value = [t.EmberStatus.SUCCESS, CUSTOM_SIZE]
+
+    return ezsp
 
 
 @pytest.fixture