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

Commit 634a304e authored by Hui Peng's avatar Hui Peng
Browse files

Add bredr/ssp pairing tests (display_yesno)

Bug: 277282212
Test: atest avatar
Flag: EXEMPT, Tests
Change-Id: Ib5fe2130b27c769ef9953ba1fa729fc6e943812a
parent 71d71448
Loading
Loading
Loading
Loading
+2 −0
Original line number Diff line number Diff line
@@ -15,11 +15,13 @@
from pairing.br_edr.legacy.tests import BREDRLegacyTestClass
from pairing.br_edr.misc.service_access_tests import ServiceAccessTempBondingTest

from pairing.br_edr.ssp.display_output_and_yes_no_input.tests import BREDRDisplayYesNoTestClass
from pairing.br_edr.ssp.no_output_no_input.tests import BREDRNoOutputNoInputTestClass

from pairing.smp_test import SmpTest

_test_class_list = [
    BREDRDisplayYesNoTestClass,
    BREDRNoOutputNoInputTestClass,
    BREDRLegacyTestClass,
    ServiceAccessTempBondingTest,
+68 −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.br_edr.ssp.test_base import BREDRSSPPairTestBase

from pandora.security_pb2 import PairingEventAnswer


class BREDRDisplayYesNoTestClass(BREDRSSPPairTestBase):

    def _setup_devices(self) -> None:

        self.ref.config.setdefault('classic_enabled', True)
        self.ref.config.setdefault('le_enabled', False)
        self.ref.config.setdefault('classic_ssp_enabled', True)
        self.ref.config.setdefault('classic_sc_enabled', False)

        self.ref.config.setdefault(
            'server',
            {
                'pairing_sc_enable': False,
                'pairing_mitm_enable': True,
                'pairing_bonding_enable': True,
                # Android IO CAP: Display_YESNO
                # Ref IO CAP:
                'io_capability': 'display_output_and_yes_no_input',
            },
        )

    async def accept_pairing(self):
        expected_pairing_method = 'numeric_comparison'

        # initiator receives numeric_comparison
        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)
        logging.debug(f'init_ev.numeric_comparison:{init_ev.numeric_comparison}')

        # responder receives numeric_comparison
        responder_ev = await anext(self.responder_pairing_event_stream)
        logging.debug(f'responder_ev.method_variant():{responder_ev.method_variant()}')
        assert_equal(responder_ev.method_variant(), expected_pairing_method)
        logging.debug(f'responder_ev.numeric_comparison:{responder_ev.numeric_comparison}')

        confirm = responder_ev.numeric_comparison == init_ev.numeric_comparison

        logging.debug(f'confirm:{confirm}')

        init_ev_ans = PairingEventAnswer(event=init_ev, confirm=confirm)
        self.initiator_pairing_event_stream.send_nowait(init_ev_ans)

        responder_ev_ans = PairingEventAnswer(event=responder_ev, confirm=confirm)
        self.responder_pairing_event_stream.send_nowait(responder_ev_ans)