#!/usr/bin/env python3.4 # # Copyright 2016 - 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. from acts.test_decorators import test_tracker_info from acts.test_utils.tel.TelephonyBaseTest import TelephonyBaseTest from acts.test_utils.tel.tel_atten_utils import set_rssi from acts.test_utils.tel.tel_defines import MAX_RSSI_RESERVED_VALUE from acts.test_utils.tel.tel_defines import MIN_RSSI_RESERVED_VALUE from acts.test_utils.tel.tel_defines import MAX_WAIT_TIME_NW_SELECTION from acts.test_utils.tel.tel_defines import NETWORK_SERVICE_DATA from acts.test_utils.tel.tel_defines import GEN_4G from acts.test_utils.tel.tel_test_utils import ensure_network_generation from acts.test_utils.tel.tel_test_utils import ensure_wifi_connected from acts.test_utils.tel.tel_test_utils import toggle_airplane_mode from acts.test_utils.tel.tel_test_utils import verify_http_connection from acts.test_utils.tel.tel_test_utils import wait_for_cell_data_connection from acts.test_utils.tel.tel_test_utils import wait_for_wifi_data_connection from acts.test_utils.tel.tel_test_utils import run_multithread_func from acts.test_utils.tel.tel_test_utils import active_file_download_test from acts.test_utils.tel.tel_test_utils import get_telephony_signal_strength from acts.test_utils.tel.tel_test_utils import get_wifi_signal_strength from acts.utils import adb_shell_ping # Attenuator name ATTEN_NAME_FOR_WIFI_2G = 'wifi0' ATTEN_NAME_FOR_WIFI_5G = 'wifi1' ATTEN_NAME_FOR_CELL_3G = 'cell0' ATTEN_NAME_FOR_CELL_4G = 'cell1' DEFAULT_PING_DURATION = 120 DEFAULT_IRAT_DURATION = 60 class TelWifiDataTest(TelephonyBaseTest): def __init__(self, controllers): TelephonyBaseTest.__init__(self, controllers) self.stress_test_number = self.get_stress_test_number() self.live_network_ssid = self.user_params["wifi_network_ssid"] self.live_network_pwd = self.user_params.get("wifi_network_pass") self.attens = {} for atten in self.attenuators: self.attens[atten.path] = atten attentuator_name_list = [ ATTEN_NAME_FOR_WIFI_2G, ATTEN_NAME_FOR_WIFI_5G, ATTEN_NAME_FOR_CELL_3G, ATTEN_NAME_FOR_CELL_4G ] for atten_name in attentuator_name_list: set_rssi(self.log, self.attens[atten_name], 0, MAX_RSSI_RESERVED_VALUE) def teardown_test(self): super().teardown_test() set_rssi(self.log, self.attens[ATTEN_NAME_FOR_WIFI_2G], 0, MAX_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_WIFI_5G], 0, MAX_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_CELL_3G], 0, MAX_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_CELL_4G], 0, MAX_RSSI_RESERVED_VALUE) return True def _basic_connectivity_check(self): """ Set Attenuator Value for WiFi and Cell to 0 Make sure DUT get Cell Data coverage (LTE) Make sure DUT WiFi is connected """ ad = self.android_devices[0] toggle_airplane_mode(self.log, ad, False) if not ensure_network_generation(self.log, ad, GEN_4G, NETWORK_SERVICE_DATA): return False if not ensure_wifi_connected(self.log, ad, self.live_network_ssid, self.live_network_pwd): ad.log.error("connect WiFi failed") return False return True def _atten_setup_wifi_cell(self): set_rssi(self.log, self.attens[ATTEN_NAME_FOR_WIFI_2G], 0, MAX_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_WIFI_5G], 0, MAX_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_CELL_3G], 0, MAX_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_CELL_4G], 0, MAX_RSSI_RESERVED_VALUE) def _atten_setup_cell_only(self): set_rssi(self.log, self.attens[ATTEN_NAME_FOR_WIFI_2G], 0, MIN_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_WIFI_5G], 0, MIN_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_CELL_3G], 0, MAX_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_CELL_4G], 0, MAX_RSSI_RESERVED_VALUE) def _atten_setup_lte_only(self): set_rssi(self.log, self.attens[ATTEN_NAME_FOR_WIFI_2G], 0, MIN_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_WIFI_5G], 0, MIN_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_CELL_3G], 0, MIN_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_CELL_4G], 0, MAX_RSSI_RESERVED_VALUE) def _atten_setup_wcdma_only(self): set_rssi(self.log, self.attens[ATTEN_NAME_FOR_WIFI_2G], 0, MIN_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_WIFI_5G], 0, MIN_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_CELL_3G], 0, MAX_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_CELL_4G], 0, MIN_RSSI_RESERVED_VALUE) def _atten_setup_wifi_only(self): set_rssi(self.log, self.attens[ATTEN_NAME_FOR_WIFI_2G], 0, MAX_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_WIFI_5G], 0, MAX_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_CELL_3G], 0, MIN_RSSI_RESERVED_VALUE) set_rssi(self.log, self.attens[ATTEN_NAME_FOR_CELL_4G], 0, MIN_RSSI_RESERVED_VALUE) @TelephonyBaseTest.tel_test_wrap def _wifi_cell_irat_task(self, ad, irat_wait_time=60): """ Atten only WiFi to MIN and MAX WiFi --> Cellular """ self._atten_setup_wifi_cell() if (not wait_for_wifi_data_connection(self.log, ad, True, irat_wait_time) or not verify_http_connection(self.log, ad)): ad.log.error("Data not on WiFi") get_telephony_signal_strength(ad) get_wifi_signal_strength(ad) return False ad.log.info("Triggering WiFi to Cellular IRAT") self._atten_setup_cell_only() if (not wait_for_cell_data_connection(self.log, ad, True, irat_wait_time) or not verify_http_connection(self.log, ad)): ad.log.error("Data not on Cell") get_telephony_signal_strength(ad) get_wifi_signal_strength(ad) return False return True @test_tracker_info(uuid="b223f74b-59f4-4eec-8785-67420bd96bd1") @TelephonyBaseTest.tel_test_wrap def test_wifi_cell_irat_stress_ping_continuous(self): """Test for data switch between WiFi and Cell. DUT go in and out WiFi coverage for multiple times. Steps: 1. Set WiFi and Cellular signal to good (attenuation value to MIN). 2. Make sure DUT get Cell data coverage (LTE) and WiFi connected. 3. Set WiFi RSSI to MAX (WiFi attenuator value to MIN). 4. Verify DUT report WiFi connected and Internet access OK. 5. Set WiFi RSSI to MIN (WiFi attenuator value to MAX). 6. Verify DUT report Cellular Data connected and Internet access OK. 7. Repeat Step 3~6 for stress number. Expected Results: 4. DUT report WiFi connected and Internet access OK. 6. DUT report Cellular Data connected and Internet access OK. 7. Stress test should pass. Returns: True if Pass. False if fail. """ if not self._basic_connectivity_check(): self.log.error("Basic Connectivity Check Failed") return False total_iteration = self.stress_test_number ad = self.android_devices[0] ping_task = (adb_shell_ping, (ad, DEFAULT_PING_DURATION, "www.google.com", 200, 40)) irat_task = (self._wifi_cell_irat_task, (ad, DEFAULT_IRAT_DURATION)) current_iteration = 1 while (current_iteration <= total_iteration): self.log.info(">----Current iteration = %d/%d----<", current_iteration, total_iteration) results = run_multithread_func(self.log, [ping_task, irat_task]) if not results[1]: ad.log.error("Data IRAT failed in active ICMP transfer") break if results[0]: ad.log.info("ICMP transfer succeeded with parallel IRAT") else: ad.log.error("ICMP transfer failed with parallel IRAT") break self.log.info(">----Iteration : %d/%d succeed.----<", current_iteration, total_iteration) current_iteration += 1 if current_iteration <= total_iteration: self.log.info(">----Iteration : %d/%d failed.----<", current_iteration, total_iteration) return False else: return True @test_tracker_info(uuid="72d2aa4d-c395-417e-99c5-12dc22ea90a1") @TelephonyBaseTest.tel_test_wrap def test_wifi_cell_irat_stress_http_dl(self): """Test for data switch between WiFi and Cell. DUT go in and out WiFi coverage for multiple times. Steps: 1. Set WiFi and Cellular signal to good (attenuation value to MIN). 2. Make sure DUT get Cell data coverage (LTE) and WiFi connected. 3. Set WiFi RSSI to MAX (WiFi attenuator value to MIN). 4. Verify DUT report WiFi connected and able to download file 5. Set WiFi RSSI to MIN (WiFi attenuator value to MAX). 6. Verify DUT report Cellular Data connected and able to download file 7. Repeat Step 3~6 for stress number. Expected Results: 4. DUT report WiFi connected and able to download file 6. DUT report Cellular Data connected and able to download file 7. Stress test should pass. Returns: True if Pass. False if fail. """ ad = self.android_devices[0] if not self._basic_connectivity_check(): self.log.error("Basic Connectivity Check Failed") return False total_iteration = self.stress_test_number self.log.info("Stress test. Total iteration = %d.", total_iteration) current_iteration = 1 while (current_iteration <= total_iteration): self.log.info(">----Current iteration = %d/%d----<", current_iteration, total_iteration) self._atten_setup_wifi_cell() if (not wait_for_wifi_data_connection(self.log, ad, True)): ad.log.error("Data not on WiFi") get_telephony_signal_strength(ad) get_wifi_signal_strength(ad) break ad.on_mobile_data = False if not active_file_download_test(self.log, ad): ad.log.error("HTTP file download failed on WiFi") get_telephony_signal_strength(ad) get_wifi_signal_strength(ad) break self._atten_setup_cell_only() if (not wait_for_cell_data_connection(self.log, ad, True)): ad.log.error("Data not on Cell") get_telephony_signal_strength(ad) get_wifi_signal_strength(ad) break ad.on_mobile_data = True if not active_file_download_test(self.log, ad): ad.log.error("HTTP file download failed on cell") get_telephony_signal_strength(ad) get_wifi_signal_strength(ad) break self.log.info(">----Iteration : %d/%d succeed.----<", current_iteration, total_iteration) current_iteration += 1 if current_iteration <= total_iteration: self.log.info(">----Iteration : %d/%d failed.----<", current_iteration, total_iteration) return False else: return True @test_tracker_info(uuid="bce71469-114c-489f-b9c4-26c53c29a553") @TelephonyBaseTest.tel_test_wrap def test_wifi_cell_irat_stress_ping(self): """Test for data switch between WiFi and Cell. DUT go in and out WiFi coverage for multiple times. Steps: 1. Set WiFi and Cellular signal to good (attenuation value to MIN). 2. Make sure DUT get Cell data coverage (LTE) and WiFi connected. 3. Set WiFi RSSI to MAX (WiFi attenuator value to MIN). 4. Verify DUT report WiFi connected and Internet access OK. 5. Set WiFi RSSI to MIN (WiFi attenuator value to MAX). 6. Verify DUT report Cellular Data connected and Internet access OK. 7. Repeat Step 3~6 for stress number. Expected Results: 4. DUT report WiFi connected and Internet access OK. 6. DUT report Cellular Data connected and Internet access OK. 7. Stress test should pass. Returns: True if Pass. False if fail. """ ad = self.android_devices[0] if not self._basic_connectivity_check(): self.log.error("Basic Connectivity Check Failed") return False total_iteration = self.stress_test_number self.log.info("Stress test. Total iteration = %d.", total_iteration) current_iteration = 1 while (current_iteration <= total_iteration): self.log.info(">----Current iteration = %d/%d----<", current_iteration, total_iteration) self._atten_setup_wifi_cell() if (not wait_for_wifi_data_connection(self.log, ad, True) or not verify_http_connection(self.log, ad)): ad.log.error("Data not on WiFi") get_telephony_signal_strength(ad) get_wifi_signal_strength(ad) break self._atten_setup_cell_only() if (not wait_for_cell_data_connection(self.log, ad, True) or not verify_http_connection(self.log, ad)): ad.log.error("Data not on Cell") get_telephony_signal_strength(ad) get_wifi_signal_strength(ad) break self.log.info(">----Iteration : %d/%d succeed.----<", current_iteration, total_iteration) current_iteration += 1 if current_iteration <= total_iteration: self.log.info(">----Iteration : %d/%d failed.----<", current_iteration, total_iteration) return False else: return True @test_tracker_info(uuid="696f22ef-39cd-4e15-bbb2-f836d2ee47f1") @TelephonyBaseTest.tel_test_wrap def test_wifi_only_http_dl(self): """Test for 10MB file download on WiFi Only Steps: 1. Set WiFi atten to MIN and Cellular to MAX 2. Start downloading 1GB file from net 3. Verify is the download is successfull Expected Results: 1. File should download over WiFi Returns: True if Pass. False if fail. """ ad = self.android_devices[0] if not self._basic_connectivity_check(): self.log.error("Basic Connectivity Check Failed") return False self._atten_setup_wifi_only() if (not wait_for_wifi_data_connection(self.log, ad, True) or not verify_http_connection(self.log, ad)): ad.log.error("Data not on WiFi") get_telephony_signal_strength(ad) get_wifi_signal_strength(ad) return False ad.on_mobile_data = False if not active_file_download_test(self.log, ad, "10MB"): ad.log.error("HTTP file download failed on WiFi") get_telephony_signal_strength(ad) get_wifi_signal_strength(ad) return False return True @test_tracker_info(uuid="6c9bf89b-5469-4b08-acf4-0ef651b1a318") @TelephonyBaseTest.tel_test_wrap def test_lte_only_http_dl(self): """Test for 1GB file download on WiFi Only Steps: 1. Set WiFi atten to MIN and Cellular to MAX 2. Start downloading 1GB file from net 3. Verify is the download is successfull Expected Results: 1. File should download over WiFi Returns: True if Pass. False if fail. """ ad = self.android_devices[0] if not self._basic_connectivity_check(): self.log.error("Basic Connectivity Check Failed") return False self._atten_setup_lte_only() if (not wait_for_cell_data_connection(self.log, ad, True) or not verify_http_connection(self.log, ad)): ad.log.error("Data not on LTE") get_telephony_signal_strength(ad) get_wifi_signal_strength(ad) return False ad.on_mobile_data = True if not active_file_download_test(self.log, ad, "512MB"): ad.log.error("HTTP file download failed on LTE") get_telephony_signal_strength(ad) get_wifi_signal_strength(ad) return False return True if __name__ == "__main__": raise Exception("Cannot run this class directly")