Try to improve joining test reliability

zigpy · Oct 27, 2024 · 3b9eff7 · 3b9eff7
1 parent 7a390ca
commit 3b9eff7
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/application/test_joining.py b/tests/application/test_joining.py
@@ -184,6 +184,10 @@ async def test_permit_join_with_key(device, permit_result, make_application, moc
     await app.shutdown()
 
 
+@mock.patch(
+    "zigpy.device.Device._initialize",
+    new=zigpy.device.Device._initialize.__wrapped__,  # to disable retries
+)
 @pytest.mark.parametrize("device", FORMED_DEVICES)
 async def test_on_zdo_device_join(device, make_application, mocker):
     app, znp_server = make_application(server_cls=device)
@@ -204,6 +208,10 @@ async def test_on_zdo_device_join(device, make_application, mocker):
     await app.shutdown()
 
 
+@mock.patch(
+    "zigpy.device.Device._initialize",
+    new=zigpy.device.Device._initialize.__wrapped__,  # to disable retries
+)
 @pytest.mark.parametrize("device", FORMED_DEVICES)
 async def test_on_zdo_device_join_and_announce_fast(device, make_application, mocker):
     app, znp_server = make_application(server_cls=device)