From 43234d7c3e16b19368c226ac6efd4d7f5aec4a37 Mon Sep 17 00:00:00 2001 From: Josh Wu Date: Wed, 26 Jul 2023 14:13:02 +0800 Subject: [PATCH] Use with-patch to mock SMP session --- tests/self_test.py | 29 ++++++++++++++--------------- 1 file changed, 14 insertions(+), 15 deletions(-) diff --git a/tests/self_test.py b/tests/self_test.py index d7e15e4..a6be77f 100644 --- a/tests/self_test.py +++ b/tests/self_test.py @@ -21,7 +21,7 @@ import logging import os import pytest -from unittest.mock import MagicMock +from unittest.mock import MagicMock, patch from bumble.controller import Controller from bumble.core import BT_BR_EDR_TRANSPORT, BT_PERIPHERAL_ROLE, BT_CENTRAL_ROLE @@ -539,22 +539,21 @@ async def test_self_smp_over_classic(): two_devices.connections[1].on('pairing', lambda keys: on_pairing(1, keys)) # Mock SMP - from bumble.smp import Session as SmpSession + with patch('bumble.smp.Session', spec=True) as MockSmpSession: + MockSmpSession.send_pairing_confirm_command = MagicMock() + MockSmpSession.send_pairing_dhkey_check_command = MagicMock() + MockSmpSession.send_public_key_command = MagicMock() + MockSmpSession.send_pairing_random_command = MagicMock() - SmpSession.send_pairing_confirm_command = MagicMock() - SmpSession.send_pairing_dhkey_check_command = MagicMock() - SmpSession.send_public_key_command = MagicMock() - SmpSession.send_pairing_random_command = MagicMock() + # Start CTKD + await two_devices.connections[0].pair() + await asyncio.gather(*paired) - # Start CTKD - await two_devices.connections[0].pair() - await asyncio.gather(*paired) - - # Phase 2 commands should not be invoked - SmpSession.send_pairing_confirm_command.assert_not_called() - SmpSession.send_pairing_dhkey_check_command.assert_not_called() - SmpSession.send_public_key_command.assert_not_called() - SmpSession.send_pairing_random_command.assert_not_called() + # Phase 2 commands should not be invoked + MockSmpSession.send_pairing_confirm_command.assert_not_called() + MockSmpSession.send_pairing_dhkey_check_command.assert_not_called() + MockSmpSession.send_public_key_command.assert_not_called() + MockSmpSession.send_pairing_random_command.assert_not_called() # -----------------------------------------------------------------------------