From d9852d619927c879fd89ae4723422aa341bdf7b6 Mon Sep 17 00:00:00 2001
From: Roshan Pius <rpius@google.com>
Date: Wed, 13 Jan 2021 10:25:41 -0800
Subject: [PATCH] NetworkControllerWifiTest: Fix unit tests

Mock WifiInfo.makeCopy() since that is used by NetworkCapabilities for
copies now.

Bug: 177390648
Test: atest com.android.systemui.statusbar.policy.NetworkControllerWifiTest
Change-Id: I71e5c757b68bcbd70fb89cf94ecbb4b7269cad19
---
 .../statusbar/policy/NetworkControllerWifiTest.java        | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/NetworkControllerWifiTest.java b/packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/NetworkControllerWifiTest.java
index 9c9d6ea2d1aee..477756eb7a32b 100644
--- a/packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/NetworkControllerWifiTest.java
+++ b/packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/NetworkControllerWifiTest.java
@@ -20,6 +20,7 @@ import android.testing.TestableLooper.RunWithLooper;
 
 import com.android.systemui.statusbar.policy.NetworkController.IconState;
 
+import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.ArgumentCaptor;
@@ -34,6 +35,12 @@ public class NetworkControllerWifiTest extends NetworkControllerBaseTest {
     private static final int MAX_RSSI = -55;
     private WifiInfo mWifiInfo = mock(WifiInfo.class);
 
+    @Before
+    public void setUp() throws Exception {
+        super.setUp();
+        when(mWifiInfo.makeCopy(anyBoolean())).thenReturn(mWifiInfo);
+    }
+
     @Test
     public void testWifiIcon() {
         String testSsid = "Test SSID";