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
This commit is contained in:
David Su
2019-11-19 16:20:20 -08:00
parent 576bf9b7ce
commit 5c52237456

View File

@@ -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);