From b6fe7460ac8d420e7df9a931901456f99acadccb Mon Sep 17 00:00:00 2001 From: Josh Wu Date: Tue, 28 Mar 2023 12:50:17 +0800 Subject: [PATCH] Add role switch test and assertion in self test --- tests/self_test.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/tests/self_test.py b/tests/self_test.py index 4ff2d43b..d6b16ec5 100644 --- a/tests/self_test.py +++ b/tests/self_test.py @@ -140,6 +140,17 @@ async def test_self_classic_connection(responder_role): assert two_devices.connections[0] is not None assert two_devices.connections[1] is not None + # Check the role + assert two_devices.connections[0].role != responder_role + assert two_devices.connections[1].role == responder_role + + # Role switch + await two_devices.connections[0].switch_role(responder_role) + + # Check the role + assert two_devices.connections[0].role == responder_role + assert two_devices.connections[1].role != responder_role + await two_devices.connections[0].disconnect()