@@ -200,7 +200,7 @@ async def test_on_zdo_device_join(device, make_application, mocker):
200
200
app , znp_server = await make_application (server_cls = device )
201
201
await app .startup (auto_form = False )
202
202
203
- mocker .patch .object (app , "handle_join" )
203
+ mocker .patch .object (app , "handle_join" , wraps = app . handle_join )
204
204
mocker .patch ("zigpy_znp.zigbee.application.DEVICE_JOIN_MAX_DELAY" , new = 0 )
205
205
206
206
nwk = 0x1234
@@ -220,7 +220,7 @@ async def test_on_zdo_device_join_and_announce_fast(device, make_application, mo
220
220
app , znp_server = await make_application (server_cls = device )
221
221
await app .startup (auto_form = False )
222
222
223
- mocker .patch .object (app , "handle_join" )
223
+ mocker .patch .object (app , "handle_join" , wraps = app . handle_join )
224
224
mocker .patch ("zigpy_znp.zigbee.application.DEVICE_JOIN_MAX_DELAY" , new = 0.5 )
225
225
226
226
nwk = 0x1234
@@ -282,7 +282,7 @@ async def test_on_zdo_device_join_and_announce_slow(device, make_application, mo
282
282
responses = [c .ZDO .ExtRouteDisc .Rsp (Status = t .Status .SUCCESS )],
283
283
)
284
284
285
- mocker .patch .object (app , "handle_join" )
285
+ mocker .patch .object (app , "handle_join" , wraps = app . handle_join )
286
286
mocker .patch ("zigpy_znp.zigbee.application.DEVICE_JOIN_MAX_DELAY" , new = 0.1 )
287
287
288
288
nwk = 0x1234
0 commit comments