From 5c52237456fb9b0303df78b7caec1928a4b7841f Mon Sep 17 00:00:00 2001
From: David Su <dysu@google.com>
Date: Tue, 19 Nov 2019 16:20:20 -0800
Subject: [PATCH] Fix broken test caused by calculateSignalLevel() migration

calculateSignalLevel() implementation used by
WifiStatusTracker has changed. Update test to mock
new logic.

Bug: 140781184
Test: atest NetworkControllerWifiTest
Change-Id: I348137f2de7944c9600d939bb86cbfef14fcd3b9
---
 .../statusbar/policy/NetworkControllerBaseTest.java       | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/NetworkControllerBaseTest.java b/packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/NetworkControllerBaseTest.java
index 4d866136c915d..61a7cc752c706 100644
--- a/packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/NetworkControllerBaseTest.java
+++ b/packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/NetworkControllerBaseTest.java
@@ -139,6 +139,14 @@ public class NetworkControllerBaseTest extends SysuiTestCase {
         when(mMockCm.getDefaultNetworkCapabilitiesForUser(0)).thenReturn(
                 new NetworkCapabilities[] { mNetCapabilities });
         when(mMockTm.createForSubscriptionId(anyInt())).thenReturn(mMockTm);
+        doAnswer(invocation -> {
+            int rssi = invocation.getArgument(0);
+            if (rssi < -88) return 0;
+            if (rssi < -77) return 1;
+            if (rssi < -66) return 2;
+            if (rssi < -55) return 3;
+            return 4;
+        }).when(mMockWm).calculateSignalLevel(anyInt());
 
         mSignalStrength = mock(SignalStrength.class);
         mServiceState = mock(ServiceState.class);