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

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

Add ble/legacy pairing tests (keyboard_only) 

Bug: 277282212
Test: atest avatar
Flag: EXEMPT, Tests
Change-Id: Ibf1336b2e5a7023d8dba15686f7656c0b88fb20c
parent 5be4a9d5

android/pandora/test/pairing/__init__.py

+3 −0
Original line number Diff line number Diff line
@@ -23,6 +23,8 @@ 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.legacy.display_output_only.tests import BLELegDisplayOnlyTestClass

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
@@ -34,6 +36,7 @@ _test_class_list = [ 
    BLELegDisplayKbdTestClass,

    BLELegDisplayOnlyTestClass,

    BLELegDisplayYesNoTestClass,

    BLELegKbdOnlyTestClass,

    BLESCDisplayKbdTestClass,

    BLESCDisplayOnlyTestClass,

    BLESCDisplayYesNoTestClass,

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

0 → 100644
+79 −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 BLELegKbdOnlyTestClass(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': False,

                '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()[1]:{responder_ev.method_variant()}')



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

            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()}')



        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)