#
# Copyright 2014 - 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 time
import acts.base_test
import acts.test_utils.wifi.wifi_test_utils as wutils
from acts import asserts
WifiEnums = wutils.WifiEnums
WifiEventNames = wutils.WifiEventNames
class WifiPnoTest(acts.base_test.BaseTestClass):
def __init__(self, controllers):
acts.base_test.BaseTestClass.__init__(self, controllers)
self.tests = (
"test_simple_pno_connection",
"test_pno_connection_with_multiple_saved_networks",
)
def setup_class(self):
self.dut = self.android_devices[0]
wutils.wifi_test_device_init(self.dut)
req_params = (
"attn_vals",
"pno_network_a",
"pno_network_b",
"pno_interval"
)
self.unpack_userparams(req_params)
self.attn_a = self.attenuators[0]
self.attn_b = self.attenuators[1]
self.set_attns("default")
def setup_test(self):
self.dut.droid.wifiStartTrackingStateChange()
self.dut.droid.wakeLockAcquireBright()
self.dut.droid.wakeUpNow()
wutils.reset_wifi(self.dut)
self.dut.ed.clear_all_events()
def teardown_test(self):
self.dut.droid.wifiStopTrackingStateChange()
self.dut.droid.wakeLockRelease()
self.dut.droid.goToSleepNow()
wutils.reset_wifi(self.dut)
self.dut.ed.clear_all_events()
self.set_attns("default")
def on_fail(self, test_name, begin_time):
self.dut.take_bug_report(test_name, begin_time)
self.dut.cat_adb_log(test_name, begin_time)
"""Helper Functions"""
def set_attns(self, attn_val_name):
"""Sets attenuation values on attenuators used in this test.
Args:
attn_val_name: Name of the attenuation value pair to use.
"""
msg = "Set attenuation values to %s" % self.attn_vals[attn_val_name]
self.log.info(msg)
try:
self.attn_a.set_atten(self.attn_vals[attn_val_name][0])
self.attn_b.set_atten(self.attn_vals[attn_val_name][1])
except:
msg = "Failed to set attenuation values %s." % attn_val_name
self.log.error(msg)
raise
def trigger_pno_and_assert_connect(self, attn_val_name, expected_con):
"""Sets attenuators to disconnect current connection and power-off the
screen to trigger PNO. Validate that the DUT connected to the new SSID
as expected after PNO.
Args:
attn_val_name: Name of the attenuation value pair to use.
expected_con: The expected info of the network to we expect the DUT
to roam to.
"""
connection_info = self.dut.droid.wifiGetConnectionInfo()
self.log.info("Triggering PNO connect from %s to %s" %
(connection_info[WifiEnums.SSID_KEY],
expected_con[WifiEnums.SSID_KEY]))
self.dut.droid.goToSleepNow()
self.set_attns(attn_val_name)
self.log.info("Wait %ss for PNO to trigger." % self.pno_interval)
time.sleep(self.pno_interval)
try:
self.dut.droid.wakeLockAcquireBright()
self.dut.droid.wakeUpNow()
expected_ssid = expected_con[WifiEnums.SSID_KEY]
verify_con = { WifiEnums.SSID_KEY : expected_ssid }
wutils.verify_wifi_connection_info(self.dut, verify_con)
self.log.info("Connected to %s successfully after PNO" %
expected_ssid)
finally:
self.dut.droid.wifiLockRelease()
self.dut.droid.goToSleepNow()
def add_dummy_networks(self, num_networks):
"""Add some dummy networks to the device.
Args:
num_networks: Number of networks to add.
"""
ssid_name_base = "pno_dummy_network_"
for i in range(0, num_networks) :
network = {}
network[WifiEnums.SSID_KEY] = ssid_name_base + str(i)
network[WifiEnums.PWD_KEY] = "pno_dummy";
asserts.assert_true(self.dut.droid.wifiAddNetwork(network) != -1,
"Add network %r failed" % network)
""" Tests Begin """
def test_simple_pno_connection(self):
"""Test PNO triggered autoconnect to a network.
Steps:
1. Save 2 valid network configurations (a & b) in the device.
2. Attenuate network b.
3. Connect the device to network a.
4. Switch off the screen on the device.
5. Attenuate network a and remove attenuation on network b and wait for
a few seconds to trigger PNO.
6. Check the device connected to network b automatically.
7. Switch off the screen on the device.
8. Attenuate network b and remove attenuation on network a and wait for
a few seconds to trigger PNO.
9. Check the device connected to network a automatically.
"""
asserts.assert_true(
self.dut.droid.wifiAddNetwork(self.pno_network_a) != -1,
"Add network %r failed" % self.pno_network_a)
asserts.assert_true(
self.dut.droid.wifiAddNetwork(self.pno_network_b) != -1,
"Add network %r failed" % self.pno_network_b)
self.set_attns("a_on_b_off")
wutils.wifi_connect(self.dut, self.pno_network_a),
self.trigger_pno_and_assert_connect("b_on_a_off", self.pno_network_b)
self.trigger_pno_and_assert_connect("a_on_b_off", self.pno_network_a)
def test_pno_connection_with_multiple_saved_networks(self):
"""Test PNO triggered autoconnect to a network when there are more
than 16 networks saved in the device.
16 is the max list size of PNO watch list for most devices. The device
should automatically pick the 16 latest added networks in the list.
So add 16 dummy networks and then add 2 valid networks.
Steps:
1. Save 16 dummy network configurations in the device.
2. Run the simple pno test.
"""
self.add_dummy_networks(16)
self.test_simple_pno_connection()
""" Tests End """