| #!/usr/bin/env python3 |
| # |
| # Copyright (C) 2019 The Android Open Source Project |
| # |
| # 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 |
| # |
| # http://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 acts_contrib.test_utils.bt.bt_test_utils as btutils |
| from acts_contrib.test_utils.bt.bt_constants import adv_succ |
| from acts_contrib.test_utils.bt.A2dpBaseTest import A2dpBaseTest |
| from acts_contrib.test_utils.bt.bt_constants import bt_default_timeout |
| from acts_contrib.test_utils.bt.bt_constants import ble_advertise_settings_modes |
| from acts_contrib.test_utils.bt.bt_constants import ble_advertise_settings_tx_powers |
| from acts_contrib.test_utils.bt.bt_test_utils import BtTestUtilsError |
| from queue import Empty |
| from acts_contrib.test_utils.bt.bt_test_utils import generate_ble_advertise_objects |
| from acts_contrib.test_utils.bt.bt_test_utils import setup_multiple_devices_for_bt_test |
| |
| INIT_ATTEN = 0 |
| |
| |
| class BtA2dpRangeWithBleAdvTest(A2dpBaseTest): |
| """User can generate test case with below format. |
| test_bt_a2dp_range_codec_"Codec"_adv_mode_"Adv Mode"_adv_tx_power_"Adv Tx Power" |
| |
| Below are the list of test cases: |
| test_bt_a2dp_range_codec_AAC_adv_mode_low_power_adv_tx_power_ultra_low |
| test_bt_a2dp_range_codec_AAC_adv_mode_low_power_adv_tx_power_low |
| test_bt_a2dp_range_codec_AAC_adv_mode_low_power_adv_tx_power_medium |
| test_bt_a2dp_range_codec_AAC_adv_mode_low_power_adv_tx_power_high |
| test_bt_a2dp_range_codec_AAC_adv_mode_balanced_adv_tx_power_ultra_low |
| test_bt_a2dp_range_codec_AAC_adv_mode_balanced_adv_tx_power_low |
| test_bt_a2dp_range_codec_AAC_adv_mode_balanced_adv_tx_power_medium |
| test_bt_a2dp_range_codec_AAC_adv_mode_balanced_adv_tx_power_high |
| test_bt_a2dp_range_codec_AAC_adv_mode_low_latency_adv_tx_power_ultra_low |
| test_bt_a2dp_range_codec_AAC_adv_mode_low_latency_adv_tx_power_low |
| test_bt_a2dp_range_codec_AAC_adv_mode_low_latency_adv_tx_power_medium |
| test_bt_a2dp_range_codec_AAC_adv_mode_low_latency_adv_tx_power_high |
| test_bt_a2dp_range_codec_SBC_adv_mode_low_power_adv_tx_power_ultra_low |
| test_bt_a2dp_range_codec_SBC_adv_mode_low_power_adv_tx_power_low |
| test_bt_a2dp_range_codec_SBC_adv_mode_low_power_adv_tx_power_medium |
| test_bt_a2dp_range_codec_SBC_adv_mode_low_power_adv_tx_power_high |
| test_bt_a2dp_range_codec_SBC_adv_mode_balanced_adv_tx_power_ultra_low |
| test_bt_a2dp_range_codec_SBC_adv_mode_balanced_adv_tx_power_low |
| test_bt_a2dp_range_codec_SBC_adv_mode_balanced_adv_tx_power_medium |
| test_bt_a2dp_range_codec_SBC_adv_mode_balanced_adv_tx_power_high |
| test_bt_a2dp_range_codec_SBC_adv_mode_low_latency_adv_tx_power_ultra_low |
| test_bt_a2dp_range_codec_SBC_adv_mode_low_latency_adv_tx_power_low |
| test_bt_a2dp_range_codec_SBC_adv_mode_low_latency_adv_tx_power_medium |
| test_bt_a2dp_range_codec_SBC_adv_mode_low_latency_adv_tx_power_high |
| """ |
| |
| def __init__(self, configs): |
| super().__init__(configs) |
| req_params = ['attenuation_vector', 'codecs'] |
| opt_params = ['gain_mismatch', 'dual_chain'] |
| #'attenuation_vector' is a dict containing: start, stop and step of attenuation changes |
| #'codecs' is a list containing all codecs required in the tests |
| #'gain_mismatch' is an offset value between the BT two chains |
| #'dual_chain' set to 1 enable sweeping attenuation for BT two chains |
| self.unpack_userparams(req_params) |
| self.unpack_userparams(opt_params, dual_chian=None, gain_mismatch=None) |
| |
| def setup_generated_tests(self): |
| for codec_config in self.codecs: |
| for adv_mode in ble_advertise_settings_modes.items(): |
| for adv_power_level in ble_advertise_settings_tx_powers.items(): |
| arg_set = [(codec_config, adv_mode, adv_power_level)] |
| self.generate_tests( |
| test_logic=self.BtA2dp_with_ble_adv_test_logic, |
| name_func=self.create_test_name, |
| arg_sets=arg_set) |
| |
| def setup_class(self): |
| super().setup_class() |
| #Enable BQR on all android devices |
| btutils.enable_bqr(self.android_devices) |
| if hasattr(self, 'dual_chain') and self.dual_chain == 1: |
| self.atten_c0 = self.attenuators[0] |
| self.atten_c1 = self.attenuators[1] |
| self.atten_c0.set_atten(INIT_ATTEN) |
| self.atten_c1.set_atten(INIT_ATTEN) |
| return setup_multiple_devices_for_bt_test(self.android_devices) |
| |
| def teardown_class(self): |
| super().teardown_class() |
| if hasattr(self, 'atten_c0') and hasattr(self, 'atten_c1'): |
| self.atten_c0.set_atten(INIT_ATTEN) |
| self.atten_c1.set_atten(INIT_ATTEN) |
| |
| def BtA2dp_with_ble_adv_test_logic(self, codec_config, adv_mode, |
| adv_power_level): |
| adv_callback = self.start_ble_adv(adv_mode[1], adv_power_level[1]) |
| self.run_a2dp_to_max_range(codec_config) |
| self.dut.droid.bleStopBleAdvertising(adv_callback) |
| self.log.info("Advertisement stopped Successfully") |
| |
| def create_test_name(self, codec_config, adv_mode, adv_power_level): |
| if hasattr(self, 'dual_chain') and self.dual_chain == 1: |
| test_case_name = 'test_dual_bt_a2dp_range_codec_{}_gainmismatch_{}dB_adv_mode_{}_adv_tx_power_{}'.format( |
| codec_config['codec_type'], self.gain_mismatch, adv_mode[0], |
| adv_power_level[0]) |
| else: |
| test_case_name = 'test_bt_a2dp_range_codec_{}_adv_mode_{}_adv_tx_power_{}'.format( |
| codec_config['codec_type'], adv_mode[0], adv_power_level[0]) |
| return test_case_name |
| |
| def start_ble_adv(self, adv_mode, adv_power_level): |
| """Function to start an LE advertisement |
| Steps: |
| 1. Create a advertise data object |
| 2. Create a advertise settings object. |
| 3. Create a advertise callback object. |
| 4. Start an LE advertising using the objects created in steps 1-3. |
| 5. Find the onSuccess advertisement event. |
| |
| Expected Result: |
| Advertisement is successfully advertising. |
| |
| Returns: |
| Returns advertise call back""" |
| |
| self.dut.droid.bleSetAdvertiseDataIncludeDeviceName(True) |
| self.dut.droid.bleSetAdvertiseSettingsAdvertiseMode(adv_mode) |
| self.dut.droid.bleSetAdvertiseSettingsIsConnectable(True) |
| self.dut.droid.bleSetAdvertiseSettingsTxPowerLevel(adv_power_level) |
| advertise_callback, advertise_data, advertise_settings = ( |
| generate_ble_advertise_objects(self.dut.droid)) |
| self.dut.droid.bleStartBleAdvertising(advertise_callback, |
| advertise_data, |
| advertise_settings) |
| try: |
| self.dut.ed.pop_event(adv_succ.format(advertise_callback), |
| bt_default_timeout) |
| self.log.info("Advertisement started successfully") |
| except Empty as err: |
| raise BtTestUtilsError( |
| "Advertiser did not start successfully {}".format(err)) |
| return advertise_callback |
| |
