Merge "Fix Robolectric failures for new registerReceiver security design"

import static com.google.common.truth.Truth.assertThat;

import static org.mockito.ArgumentMatchers.any;

import static org.mockito.ArgumentMatchers.anyInt;

import static org.mockito.Mockito.doNothing;

import static org.mockito.Mockito.doReturn;

import static org.mockito.Mockito.spy;

import static org.mockito.Mockito.times;

import static org.mockito.Mockito.verify;

import android.content.BroadcastReceiver;

import android.content.Context;

import android.content.Intent;

import android.content.IntentFilter;

import android.net.ConnectivityManager;

import android.net.NetworkRequest;

import com.android.settingslib.core.instrumentation.MetricsFeatureProvider;

import org.junit.Before;

import org.junit.Ignore;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Mock;

@Config(shadows = ShadowRestrictedLockUtilsInternal.class)

public class WifiPrimarySwitchPreferenceControllerTest {

    @Mock

    private Intent mIntentReceiver;

    @Mock

    private WifiManager mWifiManager;

    @Mock

        MockitoAnnotations.initMocks(this);

        mMetricsFeatureProvider = FakeFeatureFactory.setupForTest().getMetricsFeatureProvider();

        mContext = spy(RuntimeEnvironment.application.getApplicationContext());

        doReturn(mIntentReceiver).when(mContext)

                .registerReceiver(any(BroadcastReceiver.class), any(IntentFilter.class), anyInt());

        doNothing().when(mContext).unregisterReceiver(any(BroadcastReceiver.class));

        when(mContext.getSystemService(ConnectivityManager.class)).thenReturn(mConnectivityManager);

        when(mContext.getSystemService(NetworkScoreManager.class)).thenReturn(mNetworkScoreManager);

        mController = new WifiPrimarySwitchPreferenceController(mContext, mMetricsFeatureProvider);

    }

    @Test

    @Ignore

    public void onResume_shouldRegisterCallback() {

        mController.onResume();

        verify(mContext).registerReceiver(any(BroadcastReceiver.class), any(IntentFilter.class));

        verify(mContext).registerReceiver(

                any(BroadcastReceiver.class), any(IntentFilter.class), anyInt());

        verify(mConnectivityManager).registerNetworkCallback(

                any(NetworkRequest.class),

                any(ConnectivityManager.NetworkCallback.class),

    }

    @Test

    @Ignore

    public void onPause_shouldUnregisterCallback() {

        mController.onResume();

        mController.onPause();