| #!/usr/bin/env python3.4 |
| # |
| # Copyright 2016 - Google |
| # |
| # 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. |
| """ |
| Base Class for Defining Common Telephony Test Functionality |
| """ |
| |
| import os |
| import time |
| import traceback |
| |
| import acts.controllers.diag_logger |
| |
| from acts.base_test import BaseTestClass |
| from acts.signals import TestSignal |
| from acts import utils |
| |
| from acts.test_utils.tel.tel_subscription_utils import \ |
| get_subid_from_slot_index |
| from acts.test_utils.tel.tel_subscription_utils import \ |
| initial_set_up_for_subid_infomation |
| from acts.test_utils.tel.tel_subscription_utils import set_subid_for_data |
| from acts.test_utils.tel.tel_subscription_utils import \ |
| set_subid_for_message |
| from acts.test_utils.tel.tel_subscription_utils import \ |
| set_subid_for_outgoing_call |
| from acts.test_utils.tel.tel_test_utils import ensure_phones_default_state |
| from acts.test_utils.tel.tel_test_utils import \ |
| reset_preferred_network_type_to_allowable_range |
| from acts.test_utils.tel.tel_test_utils import set_phone_screen_on |
| from acts.test_utils.tel.tel_test_utils import set_phone_silent_mode |
| from acts.test_utils.tel.tel_test_utils import setup_droid_properties |
| from acts.test_utils.tel.tel_test_utils import refresh_droid_config |
| from acts.test_utils.tel.tel_defines import PRECISE_CALL_STATE_LISTEN_LEVEL_FOREGROUND |
| from acts.test_utils.tel.tel_defines import PRECISE_CALL_STATE_LISTEN_LEVEL_RINGING |
| from acts.test_utils.tel.tel_defines import PRECISE_CALL_STATE_LISTEN_LEVEL_BACKGROUND |
| from acts.test_utils.tel.tel_defines import WIFI_VERBOSE_LOGGING_ENABLED |
| from acts.test_utils.tel.tel_defines import WIFI_VERBOSE_LOGGING_DISABLED |
| from acts.utils import force_airplane_mode |
| |
| |
| class TelephonyBaseTest(BaseTestClass): |
| def __init__(self, controllers): |
| BaseTestClass.__init__(self, controllers) |
| self.logger_sessions = [] |
| |
| # Use for logging in the test cases to facilitate |
| # faster log lookup and reduce ambiguity in logging. |
| def tel_test_wrap(fn): |
| def _safe_wrap_test_case(self, *args, **kwargs): |
| test_id = "{}:{}:{}".format(self.__class__.__name__, fn.__name__, |
| time.time()) |
| log_string = "[Test ID] {}".format(test_id) |
| self.log.info(log_string) |
| try: |
| for ad in self.android_devices: |
| ad.droid.logI("Started " + log_string) |
| # TODO: b/19002120 start QXDM Logging |
| result = fn(self, *args, **kwargs) |
| if result is not True and "telephony_auto_rerun" in self.user_params: |
| self.teardown_test() |
| # re-run only once, if re-run pass, mark as pass |
| log_string = "[Rerun Test ID] {}. 1st run failed.".format( |
| test_id) |
| self.log.info(log_string) |
| self.setup_test() |
| for ad in self.android_devices: |
| ad.droid.logI("Rerun Started " + log_string) |
| result = fn(self, *args, **kwargs) |
| if result is True: |
| self.log.info("Rerun passed.") |
| elif result is False: |
| self.log.info("Rerun failed.") |
| else: |
| # In the event that we have a non-bool or null |
| # retval, we want to clearly distinguish this in the |
| # logs from an explicit failure, though the test will |
| # still be considered a failure for reporting purposes. |
| self.log.info("Rerun indeterminate.") |
| result = False |
| return result |
| except TestSignal: |
| raise |
| except Exception as e: |
| self.log.error(traceback.format_exc()) |
| self.log.error(str(e)) |
| return False |
| finally: |
| # TODO: b/19002120 stop QXDM Logging |
| for ad in self.android_devices: |
| try: |
| ad.adb.wait_for_device() |
| ad.droid.logI("Finished " + log_string) |
| except Exception as e: |
| self.log.error(str(e)) |
| |
| return _safe_wrap_test_case |
| |
| def setup_class(self): |
| setattr(self, "diag_logger", |
| self.register_controller(acts.controllers.diag_logger, |
| required=False)) |
| for ad in self.android_devices: |
| setup_droid_properties(self.log, ad, |
| self.user_params["sim_conf_file"]) |
| if not set_phone_screen_on(self.log, ad): |
| self.log.error("Failed to set phone screen-on time.") |
| return False |
| if not set_phone_silent_mode(self.log, ad): |
| self.log.error("Failed to set phone silent mode.") |
| return False |
| |
| ad.droid.telephonyAdjustPreciseCallStateListenLevel( |
| PRECISE_CALL_STATE_LISTEN_LEVEL_FOREGROUND, True) |
| ad.droid.telephonyAdjustPreciseCallStateListenLevel( |
| PRECISE_CALL_STATE_LISTEN_LEVEL_RINGING, True) |
| ad.droid.telephonyAdjustPreciseCallStateListenLevel( |
| PRECISE_CALL_STATE_LISTEN_LEVEL_BACKGROUND, True) |
| |
| if "enable_wifi_verbose_logging" in self.user_params: |
| ad.droid.wifiEnableVerboseLogging(WIFI_VERBOSE_LOGGING_ENABLED) |
| |
| # Reset preferred network type. |
| reset_preferred_network_type_to_allowable_range(self.log, ad) |
| |
| # Sub ID setup |
| for ad in self.android_devices: |
| initial_set_up_for_subid_infomation(self.log, ad) |
| return True |
| |
| def teardown_class(self): |
| try: |
| ensure_phones_default_state(self.log, self.android_devices) |
| |
| for ad in self.android_devices: |
| ad.droid.telephonyAdjustPreciseCallStateListenLevel( |
| PRECISE_CALL_STATE_LISTEN_LEVEL_FOREGROUND, False) |
| ad.droid.telephonyAdjustPreciseCallStateListenLevel( |
| PRECISE_CALL_STATE_LISTEN_LEVEL_RINGING, False) |
| ad.droid.telephonyAdjustPreciseCallStateListenLevel( |
| PRECISE_CALL_STATE_LISTEN_LEVEL_BACKGROUND, False) |
| if "enable_wifi_verbose_logging" in self.user_params: |
| ad.droid.wifiEnableVerboseLogging( |
| WIFI_VERBOSE_LOGGING_DISABLED) |
| finally: |
| for ad in self.android_devices: |
| try: |
| ad.droid.connectivityToggleAirplaneMode(True) |
| except BrokenPipeError: |
| # Broken Pipe, can not call SL4A API to turn on Airplane Mode. |
| # Use adb command to turn on Airplane Mode. |
| if not force_airplane_mode(ad, True): |
| self.log.error( |
| "Can not turn on airplane mode on:{}".format( |
| ad.serial)) |
| return True |
| |
| def setup_test(self): |
| for ad in self.android_devices: |
| refresh_droid_config(self.log, ad) |
| |
| if getattr(self, "diag_logger", None): |
| for logger in self.diag_logger: |
| self.log.info("Starting a diagnostic session {}".format( |
| logger)) |
| self.logger_sessions.append((logger, logger.start())) |
| |
| return ensure_phones_default_state(self.log, self.android_devices) |
| |
| def teardown_test(self): |
| for (logger, session) in self.logger_sessions: |
| self.log.info("Resetting a diagnostic session {},{}".format( |
| logger, session)) |
| logger.reset() |
| self.logger_sessions = [] |
| return True |
| |
| def on_exception(self, test_name, begin_time): |
| self._pull_diag_logs(test_name, begin_time) |
| return self._take_bug_report(test_name, begin_time) |
| |
| def on_fail(self, test_name, begin_time): |
| self._pull_diag_logs(test_name, begin_time) |
| return self._take_bug_report(test_name, begin_time) |
| |
| def _pull_diag_logs(self, test_name, begin_time): |
| for (logger, session) in self.logger_sessions: |
| self.log.info("Pulling diagnostic session {}".format(logger)) |
| logger.stop(session) |
| diag_path = os.path.join(self.log_path, begin_time) |
| utils.create_dir(diag_path) |
| logger.pull(session, diag_path) |
| |
| def _take_bug_report(self, test_name, begin_time): |
| if "no_bug_report_on_fail" in self.user_params: |
| return |
| |
| # magical sleep to ensure the runtime restart or reboot begins |
| time.sleep(1) |
| for ad in self.android_devices: |
| try: |
| ad.adb.wait_for_device() |
| ad.take_bug_report(test_name, begin_time) |
| tombstone_path = os.path.join( |
| ad.log_path, "BugReports", |
| "{},{}".format(begin_time, ad.serial).replace(' ', '_')) |
| utils.create_dir(tombstone_path) |
| ad.adb.pull('/data/tombstones/', tombstone_path) |
| except: |
| ad.log.error("Failed to take a bug report for {}, {}" |
| .format(ad.serial, test_name)) |