Skip to content

Commit 211b51c

Browse files
committed
Speed up unit tests
1 parent 30875bd commit 211b51c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/application/test_joining.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -200,7 +200,7 @@ async def test_on_zdo_device_join(device, make_application, mocker):
200200
app, znp_server = await make_application(server_cls=device)
201201
await app.startup(auto_form=False)
202202

203-
mocker.patch.object(app, "handle_join")
203+
mocker.patch.object(app, "handle_join", wraps=app.handle_join)
204204
mocker.patch("zigpy_znp.zigbee.application.DEVICE_JOIN_MAX_DELAY", new=0)
205205

206206
nwk = 0x1234
@@ -220,7 +220,7 @@ async def test_on_zdo_device_join_and_announce_fast(device, make_application, mo
220220
app, znp_server = await make_application(server_cls=device)
221221
await app.startup(auto_form=False)
222222

223-
mocker.patch.object(app, "handle_join")
223+
mocker.patch.object(app, "handle_join", wraps=app.handle_join)
224224
mocker.patch("zigpy_znp.zigbee.application.DEVICE_JOIN_MAX_DELAY", new=0.5)
225225

226226
nwk = 0x1234
@@ -282,7 +282,7 @@ async def test_on_zdo_device_join_and_announce_slow(device, make_application, mo
282282
responses=[c.ZDO.ExtRouteDisc.Rsp(Status=t.Status.SUCCESS)],
283283
)
284284

285-
mocker.patch.object(app, "handle_join")
285+
mocker.patch.object(app, "handle_join", wraps=app.handle_join)
286286
mocker.patch("zigpy_znp.zigbee.application.DEVICE_JOIN_MAX_DELAY", new=0.1)
287287

288288
nwk = 0x1234

0 commit comments

Comments
 (0)