From c7dda3037d11cc273e34df32e52126cf5edbfd86 Mon Sep 17 00:00:00 2001 From: yinxu Date: Wed, 28 Apr 2021 16:02:19 -0700 Subject: [PATCH] Mock the makeCopy() method in the unit test Bug: 186599784 Test: atest NetworkControllerWifiTest Change-Id: I3794af59da52d0fa800deb0fe67d8ff1a9f5e496 --- .../systemui/statusbar/policy/NetworkControllerWifiTest.java | 3 +++ 1 file changed, 3 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 ab7cbf77f5eed..bfb98deda4529 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 @@ -4,6 +4,7 @@ import static junit.framework.Assert.assertEquals; import static org.junit.Assert.assertTrue; import static org.mockito.ArgumentMatchers.anyLong; +import static org.mockito.Matchers.anyLong; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; @@ -309,6 +310,7 @@ public class NetworkControllerWifiTest extends NetworkControllerBaseTest { // Put RSSI in the middle of the range. rssi += amountPerLevel / 2; when(mVcnTransportInfo.getWifiInfo()).thenReturn(mWifiInfo); + when(mVcnTransportInfo.makeCopy(anyLong())).thenReturn(mVcnTransportInfo); when(mWifiInfo.getRssi()).thenReturn(rssi); when(mWifiInfo.isCarrierMerged()).thenReturn(true); when(mWifiInfo.getSubscriptionId()).thenReturn(1); @@ -318,6 +320,7 @@ public class NetworkControllerWifiTest extends NetworkControllerBaseTest { protected void setWifiStateForVcn(boolean connected, String ssid) { when(mVcnTransportInfo.getWifiInfo()).thenReturn(mWifiInfo); + when(mVcnTransportInfo.makeCopy(anyLong())).thenReturn(mVcnTransportInfo); when(mWifiInfo.getSSID()).thenReturn(ssid); when(mWifiInfo.isCarrierMerged()).thenReturn(true); when(mWifiInfo.getSubscriptionId()).thenReturn(1);