From 1b88fd7e3c4669467d71fe3dffae9a1ef94d7e26 Mon Sep 17 00:00:00 2001
From: David Su <dysu@google.com>
Date: Mon, 9 Mar 2020 19:38:25 -0700
Subject: [PATCH] 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 | 2 ++
 1 file changed, 2 insertions(+)

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 6ac92d409c..fee9ef026a 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 @@ package android.net.wifi.cts;
 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 @@ public class WifiLockTest extends AndroidTestCase {
         WifiLock wl = wm.createWifiLock(lockType, WIFI_TAG);
 
         wl.setReferenceCounted(true);
+        wl.setWorkSource(new WorkSource());
         assertFalse(wl.isHeld());
         wl.acquire();
         assertTrue(wl.isHeld());