commit eb787b81e4ae741c28ee018fa7413b4683430ccd
author Ahmed ElArabawy <arabawy@google.com> Tue Mar 05 19:57:14 2019 +0000
committer Android (Google) Code Review <android-gerrit@google.com> Tue Mar 05 19:57:14 2019 +0000
tree 853d4f07f64be20052a99f153b1c26c9a0de1d1d
parent 4e1a153776fa12e870cb74b0588ba8702fdadfbf
parent d87f0affe6219855d34807f104ab456d940b0336

Merge "[Wifi] Update CTS test for Wifi Locks"

tests/cts/net/src/android/net/wifi/cts/WifiManager_WifiLockTest.java

diff --git a/tests/cts/net/src/android/net/wifi/cts/WifiManager_WifiLockTest.java b/tests/cts/net/src/android/net/wifi/cts/WifiManager_WifiLockTest.java
index 3cdd56a..e08a972 100644
--- a/tests/cts/net/src/android/net/wifi/cts/WifiManager_WifiLockTest.java
+++ b/tests/cts/net/src/android/net/wifi/cts/WifiManager_WifiLockTest.java
@@ -25,13 +25,27 @@
 
     private static final String WIFI_TAG = "WifiManager_WifiLockTest";
 
-    public void testWifiLock() {
+    /**
+     * Verify acquire and release of High Performance wifi locks
+     */
+    public void testHiPerfWifiLock() {
+        testWifiLock(WifiManager.WIFI_MODE_FULL_HIGH_PERF);
+    }
+
+    /**
+     * Verify acquire and release of Low latency wifi locks
+     */
+    public void testLowLatencyWifiLock() {
+        testWifiLock(WifiManager.WIFI_MODE_FULL_LOW_LATENCY);
+    }
+
+    private void testWifiLock(int lockType) {
         if (!WifiFeature.isWifiSupported(getContext())) {
             // skip the test if WiFi is not supported
             return;
         }
         WifiManager wm = (WifiManager) getContext().getSystemService(Context.WIFI_SERVICE);
-        WifiLock wl = wm.createWifiLock(WIFI_TAG);
+        WifiLock wl = wm.createWifiLock(lockType, WIFI_TAG);
 
         wl.setReferenceCounted(true);
         assertFalse(wl.isHeld());
@@ -55,7 +69,7 @@
             // expected
         }
 
-        wl = wm.createWifiLock(WIFI_TAG);
+        wl = wm.createWifiLock(lockType, WIFI_TAG);
         wl.setReferenceCounted(false);
         assertFalse(wl.isHeld());
         wl.acquire();