commit 3f9db859bc629713ec97d0b36e18395638a7fd4c
author Bindu Mahadev <bmahadev@google.com> Tue May 22 19:39:49 2018 -0700
committer Bindu Mahadev <bmahadev@google.com> Wed May 23 02:45:18 2018 +0000
tree de89ea90e867502a6d77c42afe6bc6a4b32fb629
parent 75b4e46523c68393b86409281ad8be0a8cde49f7

[Tests]Reset wifi in wifistresstest.

Bug: 80154082
Test: Local
Change-Id: Ie19a1fe0d42bb88d633d66aaedce144890b55efc

acts/tests/google/wifi/WifiStressTest.py

diff --git a/acts/tests/google/wifi/WifiStressTest.py b/acts/tests/google/wifi/WifiStressTest.py
index 17d6c75..e798223 100755
--- a/acts/tests/google/wifi/WifiStressTest.py
+++ b/acts/tests/google/wifi/WifiStressTest.py
@@ -80,6 +80,7 @@
     def teardown_test(self):
         self.dut.droid.wakeLockRelease()
         self.dut.droid.goToSleepNow()
+        wutils.reset_wifi(self.dut)
 
     def on_fail(self, test_name, begin_time):
         self.dut.take_bug_report(test_name, begin_time)
@@ -270,7 +271,7 @@
         AP1_network = self.reference_networks[0]["5g"]
         AP2_network = self.reference_networks[1]["5g"]
         wutils.set_attns(self.attenuators, "AP1_on_AP2_off")
-        wutils.wifi_connect(self.dut, AP1_network)
+        self.scan_and_connect_by_ssid(AP1_network)
         # Reduce iteration to half because each iteration does two roams.
         for count in range(self.stress_count/2):
             self.log.info("Roaming iteration %d, from %s to %s", count,