auracaster/bumble_mirror
3
0
Fork 1
mirror of https://github.com/google/bumble.git synced 2026-06-04 08:07:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use with-patch to mock SMP session

Browse Source
This commit is contained in:
Josh Wu
2023-07-26 14:13:02 +08:00
committed by Lucas Abel
parent dbf878dc3f
commit 43234d7c3e
1 changed files with 14 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/self_test.py
+14 -15
View File
@@ -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()
# -----------------------------------------------------------------------------