| #!/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. |
| """ |
| Test Script for Telephony Pre Flight check. |
| """ |
| |
| from acts import signals |
| from acts import utils |
| |
| from acts.controllers.android_device import get_info |
| from acts.libs.ota import ota_updater |
| from acts.test_decorators import test_tracker_info |
| from acts.libs.utils.multithread import multithread_func |
| from acts_contrib.test_utils.tel.TelephonyBaseTest import TelephonyBaseTest |
| from acts_contrib.test_utils.tel.tel_defines import NETWORK_SERVICE_DATA |
| from acts_contrib.test_utils.tel.tel_defines import RAT_FAMILY_WLAN |
| from acts_contrib.test_utils.tel.tel_defines import WFC_MODE_CELLULAR_PREFERRED |
| from acts_contrib.test_utils.tel.tel_defines import WFC_MODE_WIFI_PREFERRED |
| from acts_contrib.test_utils.tel.tel_data_utils import wait_for_wifi_data_connection |
| from acts_contrib.test_utils.tel.tel_ims_utils import wait_for_wfc_enabled |
| from acts_contrib.test_utils.tel.tel_phone_setup_utils import ensure_phones_default_state |
| from acts_contrib.test_utils.tel.tel_phone_setup_utils import ensure_phone_subscription |
| from acts_contrib.test_utils.tel.tel_phone_setup_utils import phone_setup_iwlan |
| from acts_contrib.test_utils.tel.tel_phone_setup_utils import phone_setup_iwlan_cellular_preferred |
| from acts_contrib.test_utils.tel.tel_phone_setup_utils import wait_for_network_rat |
| from acts_contrib.test_utils.tel.tel_test_utils import abort_all_tests |
| from acts_contrib.test_utils.tel.tel_test_utils import get_user_config_profile |
| from acts_contrib.test_utils.tel.tel_test_utils import is_sim_locked |
| from acts_contrib.test_utils.tel.tel_test_utils import unlock_sim |
| from acts_contrib.test_utils.tel.tel_test_utils import verify_internet_connection |
| from acts_contrib.test_utils.tel.tel_voice_utils import call_setup_teardown |
| from acts_contrib.test_utils.tel.tel_voice_utils import is_phone_in_call_iwlan |
| from acts_contrib.test_utils.tel.tel_voice_utils import is_phone_in_call_volte |
| from acts_contrib.test_utils.tel.tel_wifi_utils import ensure_wifi_connected |
| from acts_contrib.test_utils.tel.tel_wifi_utils import wifi_toggle_state |
| |
| |
| class TelLivePreflightTest(TelephonyBaseTest): |
| def setup_class(self): |
| super().setup_class() |
| self.user_params["telephony_auto_rerun"] = 0 |
| |
| def _check_wfc_enabled(self, ad): |
| if not wait_for_wifi_data_connection(self.log, ad, True): |
| ad.log.error("Failed to connect to WIFI") |
| return False |
| if not wait_for_wfc_enabled(self.log, ad): |
| ad.log.error("WFC is not enabled") |
| return False |
| if not wait_for_network_rat( |
| self.log, ad, RAT_FAMILY_WLAN, |
| voice_or_data=NETWORK_SERVICE_DATA): |
| ad.log.info("Data rat can not go to iwlan mode successfully") |
| return False |
| return True |
| |
| def _get_call_verification_function(self, ad): |
| user_profile = get_user_config_profile(ad) |
| if user_profile.get("WFC Enabled", False): |
| return is_phone_in_call_iwlan |
| if user_profile.get("VoLTE Enabled", False): |
| return is_phone_in_call_volte |
| return None |
| |
| def _set_user_profile_before_ota(self): |
| ad = self.android_devices[0] |
| if not phone_setup_iwlan_cellular_preferred( |
| self.log, ad, self.wifi_network_ssid, self.wifi_network_pass): |
| ad.log.error("Failed to setup WFC to %s.", |
| WFC_MODE_CELLULAR_PREFERRED) |
| if len(self.android_devices) > 1: |
| ad = self.android_devices[1] |
| if not phone_setup_iwlan( |
| self.log, ad, True, WFC_MODE_WIFI_PREFERRED, |
| self.wifi_network_ssid, self.wifi_network_pass): |
| ad.log.error("Failed to setup WFC to %s.", |
| WFC_MODE_WIFI_PREFERRED) |
| |
| if ad.droid.imsIsEnhanced4gLteModeSettingEnabledByPlatform(): |
| ad.droid.imsSetEnhanced4gMode(False) |
| state = ad.droid.imsIsEnhanced4gLteModeSettingEnabledByUser() |
| ad.log.info("Enhanced 4G LTE Setting is %s", "on" |
| if state else "off") |
| if state: |
| ad.log.error("Expecting Enhanced 4G LTE Setting off") |
| |
| def _ota_upgrade(self, ad): |
| result = True |
| self._set_user_profile_before_ota() |
| config_profile_before = get_user_config_profile(ad) |
| ad.log.info("Before OTA user config is: %s", config_profile_before) |
| try: |
| ota_updater.update(ad) |
| except Exception as e: |
| ad.log.error("OTA upgrade failed with %s", e) |
| raise |
| if is_sim_locked(ad): |
| ad.log.info("After OTA, SIM keeps the locked state") |
| elif getattr(ad, "is_sim_locked", False): |
| ad.log.error("After OTA, SIM loses the locked state") |
| result = False |
| if not unlock_sim(ad): |
| ad.log.error(ad.log, "unable to unlock SIM") |
| if not ad.droid.connectivityCheckAirplaneMode(): |
| if not ensure_phone_subscription(self.log, ad): |
| ad.log.error("Subscription check failed") |
| result = False |
| if config_profile_before.get("WFC Enabled", False): |
| self._check_wfc_enabled(ad) |
| result = False |
| config_profile_after = get_user_config_profile(ad) |
| ad.log.info("After OTA user config is: %s", config_profile_after) |
| if config_profile_before != config_profile_after: |
| ad.log.error("Before: %s, After: %s", config_profile_before, |
| config_profile_after) |
| ad.log.error("User config profile changed after OTA") |
| result = False |
| return result |
| |
| """ Tests Begin """ |
| |
| @test_tracker_info(uuid="cb897221-99e1-4697-927e-02d92d969440") |
| @TelephonyBaseTest.tel_test_wrap |
| def test_ota_upgrade(self): |
| ota_package = self.user_params.get("ota_package") |
| if isinstance(ota_package, list): |
| ota_package = ota_package[0] |
| if ota_package and "dev/null" not in ota_package: |
| self.log.info("Upgrade with ota_package %s", ota_package) |
| self.log.info("Before OTA upgrade: %s", |
| get_info(self.android_devices)) |
| else: |
| raise signals.TestSkip("No ota_package is defined") |
| ota_util = self.user_params.get("ota_util") |
| if isinstance(ota_util, list): |
| ota_util = ota_util[0] |
| if ota_util: |
| if "update_engine_client.zip" in ota_util: |
| self.user_params["UpdateDeviceOtaTool"] = ota_util |
| self.user_params["ota_tool"] = "UpdateDeviceOtaTool" |
| else: |
| self.user_params["AdbSideloadOtaTool"] = ota_util |
| self.user_params["ota_tool"] = "AdbSideloadOtaTool" |
| self.log.info("OTA upgrade with %s by %s", ota_package, |
| self.user_params["ota_tool"]) |
| ota_updater.initialize(self.user_params, self.android_devices) |
| tasks = [(self._ota_upgrade, [ad]) for ad in self.android_devices] |
| try: |
| result = multithread_func(self.log, tasks) |
| except Exception as err: |
| abort_all_tests(self.log, "Unable to do ota upgrade: %s" % err) |
| device_info = get_info(self.android_devices) |
| self.log.info("After OTA upgrade: %s", device_info) |
| self.results.add_controller_info("AndroidDevice", device_info) |
| if len(self.android_devices) > 1: |
| caller = self.android_devices[0] |
| callee = self.android_devices[1] |
| return call_setup_teardown( |
| self.log, caller, callee, caller, |
| self._get_call_verification_function(caller), |
| self._get_call_verification_function(callee)) and result |
| return result |
| |
| @test_tracker_info(uuid="8390a2eb-a744-4cda-bade-f94a2cc83f02") |
| @TelephonyBaseTest.tel_test_wrap |
| def test_check_environment(self): |
| ad = self.android_devices[0] |
| # Check WiFi environment. |
| # 1. Connect to WiFi. |
| # 2. Check WiFi have Internet access. |
| try: |
| if not ensure_wifi_connected(self.log, ad, self.wifi_network_ssid, |
| self.wifi_network_pass): |
| abort_all_tests(ad.log, "WiFi connect fail") |
| if (not wait_for_wifi_data_connection(self.log, ad, True) |
| or not verify_internet_connection(self.log, ad)): |
| abort_all_tests(ad.log, "Data not available on WiFi") |
| finally: |
| wifi_toggle_state(self.log, ad, False) |
| # TODO: add more environment check here. |
| return True |
| |
| @test_tracker_info(uuid="7bb23ac7-6b7b-4d5e-b8d6-9dd10032b9ad") |
| @TelephonyBaseTest.tel_test_wrap |
| def test_pre_flight_check(self): |
| return ensure_phones_default_state(self.log, self.android_devices) |
| |
| @test_tracker_info(uuid="1070b160-902b-43bf-92a0-92cc2d05bb13") |
| @TelephonyBaseTest.tel_test_wrap |
| def test_check_crash(self): |
| result = True |
| begin_time = None |
| for ad in self.android_devices: |
| output = ad.adb.shell("cat /proc/uptime") |
| epoch_up_time = utils.get_current_epoch_time() - 1000 * float( |
| output.split(" ")[0]) |
| ad.crash_report_preflight = ad.check_crash_report( |
| self.test_name, |
| begin_time=epoch_up_time, |
| log_crash_report=True) |
| if ad.crash_report_preflight: |
| msg = "Find crash reports %s before test starts" % ( |
| ad.crash_report_preflight) |
| ad.log.warn(msg) |
| result = False |
| return result |