Clean up self.droid and self.ed from wifi test scripts.
Bug=27231417
Change-Id: I8780ea68ee3c5388c0f72828f51cdc89d1e9fc02
diff --git a/acts/tests/google/wifi/WifiEnterpriseTest.py b/acts/tests/google/wifi/WifiEnterpriseTest.py
index 45d8137..0f13a0c 100755
--- a/acts/tests/google/wifi/WifiEnterpriseTest.py
+++ b/acts/tests/google/wifi/WifiEnterpriseTest.py
@@ -124,24 +124,24 @@
self.config_passpoint_ttls.update(self.config_passpoint)
del self.config_passpoint_ttls[WifiEnums.SSID_KEY]
# Set screen lock password so ConfigStore is unlocked.
- self.droid.setDevicePassword(self.device_password)
+ self.dut.droid.setDevicePassword(self.device_password)
def teardown_class(self):
wutils.reset_wifi(self.dut)
- self.droid.disableDevicePassword()
- self.ed.clear_all_events()
+ self.dut.droid.disableDevicePassword()
+ self.dut.ed.clear_all_events()
def setup_test(self):
- self.droid.wifiStartTrackingStateChange()
- self.droid.wakeLockAcquireBright()
- self.droid.wakeUpNow()
+ self.dut.droid.wifiStartTrackingStateChange()
+ self.dut.droid.wakeLockAcquireBright()
+ self.dut.droid.wakeUpNow()
wutils.reset_wifi(self.dut)
- self.ed.clear_all_events()
+ self.dut.ed.clear_all_events()
def teardown_test(self):
- self.droid.wakeLockRelease()
- self.droid.goToSleepNow()
- self.droid.wifiStopTrackingStateChange()
+ self.dut.droid.wakeLockRelease()
+ self.dut.droid.goToSleepNow()
+ self.dut.droid.wifiStopTrackingStateChange()
"""Helper Functions"""