commit 1b88fd7e3c4669467d71fe3dffae9a1ef94d7e26
author David Su <dysu@google.com> Mon Mar 09 19:38:25 2020 -0700
committer David Su <dysu@google.com> Tue Mar 10 02:43:05 2020 +0000
tree 55bfeaea3058440b1bf254c0184b678bd77796ff
parent 1fca50969b633deea96a4350e1be0315424d42f3

WifiLockTest: Test WifiLock.setWorkSource()

Bug: 150236894
Test: atest android.net.wifi.cts.WifiLockTest
Change-Id: I1a6975590c2a4b885c90690e4269ba35bcaf34ff

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

diff --git a/tests/cts/net/src/android/net/wifi/cts/WifiLockTest.java b/tests/cts/net/src/android/net/wifi/cts/WifiLockTest.java
index 6ac92d4..fee9ef0 100644
--- a/tests/cts/net/src/android/net/wifi/cts/WifiLockTest.java
+++ b/tests/cts/net/src/android/net/wifi/cts/WifiLockTest.java
@@ -19,6 +19,7 @@
 import android.content.Context;
 import android.net.wifi.WifiManager;
 import android.net.wifi.WifiManager.WifiLock;
+import android.os.WorkSource;
 import android.platform.test.annotations.AppModeFull;
 import android.test.AndroidTestCase;
 
@@ -50,6 +51,7 @@
         WifiLock wl = wm.createWifiLock(lockType, WIFI_TAG);
 
         wl.setReferenceCounted(true);
+        wl.setWorkSource(new WorkSource());
         assertFalse(wl.isHeld());
         wl.acquire();
         assertTrue(wl.isHeld());