| # Copyright 2020 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| """A Bluetooth adapter MTBF test of some Bluetooth use cases. |
| |
| To add a new use case we just need to inherit from the existing test class |
| and then call the desired test methods in the batch method below. This allows |
| the test case to be used as both part of a MTBF batch and a normal batch. |
| """ |
| |
| import time |
| |
| from autotest_lib.server.cros.bluetooth.bluetooth_adapter_quick_tests import \ |
| BluetoothAdapterQuickTests |
| from autotest_lib.server.cros.bluetooth.bluetooth_adapter_better_together \ |
| import BluetoothAdapterBetterTogether |
| from autotest_lib.server.cros.bluetooth.bluetooth_adapter_hidreports_tests \ |
| import BluetoothAdapterHIDReportTests |
| |
| SHORT_SUSPEND = 10 |
| ACTION_TIMEOUT = 10 |
| SLEEP_FOR_DISCONNECTION = 10 |
| |
| class bluetooth_AdapterMTBF(BluetoothAdapterBetterTogether, |
| BluetoothAdapterHIDReportTests): |
| """A Batch of Bluetooth adapter tests for MTBF. This test is written |
| as a batch of tests in order to reduce test time, since auto-test ramp up |
| time is costly. The batch is using BluetoothAdapterQuickTests wrapper |
| methods to start and end a test and a batch of tests. |
| |
| This class can be called to run the entire test batch or to run a |
| specific test only |
| """ |
| |
| MTBF_TIMEOUT_MINS = 300 |
| batch_wrapper = BluetoothAdapterQuickTests.quick_test_batch_decorator |
| mtbf_wrapper = BluetoothAdapterQuickTests.quick_test_mtbf_decorator |
| test_wrapper = BluetoothAdapterQuickTests.quick_test_test_decorator |
| |
| def test_suspend_resume(self, device): |
| """Test the device can connect after suspending and resuming""" |
| boot_id = self.host.get_boot_id() |
| suspend = self.suspend_async( |
| suspend_time=SHORT_SUSPEND, allow_early_resume=True) |
| |
| self.test_suspend_and_wait_for_sleep( |
| suspend, sleep_timeout=ACTION_TIMEOUT) |
| self.test_wait_for_resume( |
| boot_id, suspend, resume_timeout=ACTION_TIMEOUT) |
| |
| # LE can't reconnect without advertising/discoverable |
| self.test_device_set_discoverable(device, True) |
| self.test_connection_by_device(device) |
| |
| |
| @mtbf_wrapper(timeout_mins=MTBF_TIMEOUT_MINS, test_name='typical_use_cases') |
| def run_typical_use_cases(self, phone, mouse): |
| """Run typical MTBF test scenarios""" |
| |
| self.test_smart_unlock(address=phone.address) |
| self.test_suspend_resume(device=mouse) |
| self.run_mouse_tests(device=mouse) |
| |
| |
| @test_wrapper( |
| 'MTBF Typical Use Cases', devices={'BLE_PHONE': 1, 'BLE_MOUSE': 1}) |
| def typical_use_cases_test(self): |
| """Do some initialization work then start the typical MTBF test loop""" |
| |
| mouse = self.devices['BLE_MOUSE'][0] |
| phone = self.devices['BLE_PHONE'][0] |
| |
| phone.RemoveDevice(self.bluetooth_facade.address) |
| |
| # Pair the mouse first |
| # The steps were copied from bluetooth_AdapterMDSanity |
| self.test_discover_device(mouse.address) |
| self.test_stop_discovery() |
| time.sleep(self.TEST_SLEEP_SECS) |
| self.test_pairing(mouse.address, mouse.pin, trusted=True) |
| time.sleep(self.TEST_SLEEP_SECS) |
| self.test_connection_by_adapter(mouse.address) |
| |
| self.run_typical_use_cases(phone, mouse) |
| |
| |
| @mtbf_wrapper( |
| timeout_mins=MTBF_TIMEOUT_MINS, test_name='better_together_stress') |
| def run_better_together_stress(self, address): |
| """Run better together stress test""" |
| |
| self.test_smart_unlock(address) |
| |
| |
| @test_wrapper('MTBF Better Together Stress', devices={'BLE_PHONE': 1}) |
| def better_together_stress_test(self): |
| """Run better together stress test""" |
| |
| phone = self.devices['BLE_PHONE'][0] |
| phone.RemoveDevice(self.bluetooth_facade.address) |
| self.run_better_together_stress(address=phone.address) |
| |
| |
| @batch_wrapper('Adapter MTBF') |
| def mtbf_batch_run(self, num_iterations=1, test_name=None): |
| """Run the Bluetooth MTBF test batch or a specific |
| given test. The wrapper of this method is implemented in |
| batch_decorator. Using the decorator a test batch method can |
| implement the only its core tests invocations and let the |
| decorator handle the wrapper, which is taking care for whether to |
| run a specific test or the batch as a whole, and running the batch |
| in iterations |
| |
| @param num_iterations: how many iterations to run |
| @param test_name: specific test to run otherwise None to run the |
| whole batch |
| """ |
| # TODO: finalize the test cases that need to be run as MTBF |
| self.typical_use_cases_test() |
| |
| |
| def run_once(self, host, num_iterations=1, test_name=None, args_dict=None): |
| """Run the batch of Bluetooth MTBF tests |
| |
| @param host: the DUT, usually a chromebook |
| @param num_iterations: the number of rounds to execute the test |
| @test_name: the test to run, or None for all tests |
| """ |
| |
| # Initialize and run the test batch or the requested specific test |
| self.set_fail_fast(args_dict, True) |
| self.quick_test_init(host, use_btpeer=True) |
| self.mtbf_batch_run(num_iterations, test_name) |
| self.quick_test_cleanup() |