Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit c7dda303 authored by yinxu's avatar yinxu
Browse files

Mock the makeCopy() method in the unit test

Bug: 186599784
Test: atest NetworkControllerWifiTest
Change-Id: I3794af59da52d0fa800deb0fe67d8ff1a9f5e496
parent d22bbdfc
Loading
Loading
Loading
Loading
+3 −0
Original line number Diff line number Diff line
@@ -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);