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

Commit cd833245 authored by Hui Peng's avatar Hui Peng
Browse files

Add ble/sc pairing tests (display_keyboard) 

Bug: 277282212
Test: atest avatar
Flag: EXEMPT, Tests
Change-Id: I9821fbe9a2198466c19d6f3224b7cd5a85ecf10e
parent 184fe6ae

android/pandora/test/pairing/__init__.py

+2 −0
Original line number Diff line number Diff line
@@ -22,6 +22,7 @@ from pairing.br_edr.ssp.no_output_no_input.tests import BREDRNoOutputNoInputTest 


from pairing.ble.legacy.display_output_and_keyboard_input.tests import BLELegDisplayKbdTestClass

from pairing.ble.legacy.display_output_and_yes_no_input.tests import BLELegDisplayYesNoTestClass

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.smp_test import SmpTest
@@ -30,6 +31,7 @@ from pairing.smp_test import SmpTest 
_test_class_list = [

    BLELegDisplayKbdTestClass,

    BLELegDisplayYesNoTestClass,

    BLESCDisplayKbdTestClass,

    BLESCDisplayYesNoTestClass,

    BREDRDisplayYesNoTestClass,

    BREDRDisplayOnlyTestClass,

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

0 → 100644
+75 −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 BLESCDisplayKbdTestClass(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',

            {

                # legacy pairing

                'pairing_sc_enable': True,

                'pairing_mitm_enable': True,

                'pairing_bonding_enable': True,

                # Android IO CAP: Display_KBD

                # Ref IO CAP:

                'io_capability': 'display_output_and_keyboard_input',

            },

        )



    async def accept_pairing(self):

        expected_pairing_method = 'numeric_comparison'



        responder_ev = await anext(self.responder_pairing_event_stream)

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



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

            # when pairing is initiated from bumble,

            logging.debug('>>> confirming pairing request')

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

            self.responder_pairing_event_stream.send_nowait(ans)



            responder_ev = await anext(self.responder_pairing_event_stream)

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



        init_ev = await anext(self.initiator_pairing_event_stream)

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



        assert_equal(init_ev.method_variant(), expected_pairing_method)

        assert_equal(responder_ev.method_variant(), expected_pairing_method)



        confirm = (responder_ev.numeric_comparison == init_ev.numeric_comparison)



        init_ev_ans = PairingEventAnswer(event=init_ev, confirm=confirm)

        # respond pairing based on numeric comparison on initiator

        self.initiator_pairing_event_stream.send_nowait(init_ev_ans)



        responder_ev_ans = PairingEventAnswer(event=responder_ev, confirm=confirm)

        # respond pairing based on numeric comparison on responder

        self.responder_pairing_event_stream.send_nowait(responder_ev_ans)