Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 0aad410e authored by Hui Peng's avatar Hui Peng
Browse files

Add ble/sc pairing tests (keyboard_only) 

Bug: 277282212
Test: atest avatar
Flag: EXEMPT, Tests
Change-Id: Ie056ee87e2697c8390c2c23efcefbb95d9bc7101
parent f1b94432

android/pandora/test/pairing/__init__.py

+3 −0
Original line number Diff line number Diff line
@@ -28,6 +28,8 @@ from pairing.ble.legacy.keyboard_input_only.tests import BLELegKbdOnlyTestClass 
from pairing.ble.sc.display_output_and_keyboard_input.tests import BLESCDisplayKbdTestClass

from pairing.ble.sc.display_output_and_yes_no_input.tests import BLESCDisplayYesNoTestClass

from pairing.ble.sc.display_output_only.tests import BLESCDisplayOnlyTestClass

from pairing.ble.sc.keyboard_input_only.tests import BLESCKbdOnlyTestClass





from pairing.smp_test import SmpTest
@@ -40,6 +42,7 @@ _test_class_list = [ 
    BLESCDisplayKbdTestClass,

    BLESCDisplayOnlyTestClass,

    BLESCDisplayYesNoTestClass,

    BLESCKbdOnlyTestClass,

    BREDRDisplayYesNoTestClass,

    BREDRDisplayOnlyTestClass,

    # BREDRKeyboardOnlyTestClass,

android/pandora/test/pairing/ble/sc/keyboard_input_only/tests.py

0 → 100644
+77 −0
Original line number Diff line number Diff line 
# Copyright 2024 Google LLC

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

#     https://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.



import logging



from mobly.asserts import assert_equal



from pairing.ble.test_base import BLEPairTestBaseWithGeneralAndDedicatedPairingTests



from pandora.security_pb2 import PairingEventAnswer





class BLESCKbdOnlyTestClass(BLEPairTestBaseWithGeneralAndDedicatedPairingTests):



    def _setup_devices(self) -> None:

        self.ref.config.setdefault('le_enabled', True)



        self.ref.config.setdefault('classic_enabled', False)

        self.ref.config.setdefault('classic_ssp_enabled', False)

        self.ref.config.setdefault('classic_sc_enabled', False)



        self.ref.config.setdefault(

            'server',

            {

                # secure connections pairing

                'pairing_sc_enable': True,

                'pairing_mitm_enable': True,

                'pairing_bonding_enable': True,

                # Android IO CAP: Display_KBD

                # Ref IO CAP:

                'io_capability': 'keyboard_input_only',

            },

        )



    async def accept_pairing(self):

        notif_expected_pairing_method = 'passkey_entry_notification'

        req_expected_pairing_method = 'passkey_entry_request'



        responder_ev = await anext(self.responder_pairing_event_stream)

        logging.debug(f'responder_ev.method_variant():{responder_ev.method_variant()}')



        if responder_ev.method_variant() == 'just_works':

            logging.debug('>>> confirming pairing on responder')

            ans = PairingEventAnswer(event=responder_ev, confirm=True)

            self.responder_pairing_event_stream.send_nowait(ans)

            responder_ev = await anext(self.responder_pairing_event_stream)



        init_ev = await anext(self.initiator_pairing_event_stream)

        logging.debug(f'init_ev.method_variant():{init_ev.method_variant()}')



        if init_ev.method_variant() == req_expected_pairing_method:

            notif_ev = responder_ev

            req_ev = init_ev

            req_stream = self.initiator_pairing_event_stream

        else:

            notif_ev = init_ev

            req_ev = responder_ev

            req_stream = self.responder_pairing_event_stream



        assert_equal(notif_ev.method_variant(), notif_expected_pairing_method)

        assert_equal(req_ev.method_variant(), req_expected_pairing_method)



        notified_passkey = notif_ev.passkey_entry_notification



        ans = PairingEventAnswer(event=req_ev, passkey=notified_passkey)

        req_stream.send_nowait(ans)