Merge "Make NetworkController tests more deterministic"

package com.android.systemui.statusbar.policy;

import static junit.framework.Assert.assertEquals;

import static org.mockito.Matchers.any;

import static org.mockito.Matchers.anyBoolean;

import static org.mockito.Matchers.anyInt;

import static org.mockito.Matchers.anyString;

import static org.mockito.Matchers.eq;

import static org.mockito.Mockito.doAnswer;

import static org.mockito.Mockito.mock;

import static org.mockito.Mockito.when;

import android.content.Intent;

import android.net.ConnectivityManager;

import android.net.NetworkCapabilities;

import android.net.wifi.WifiManager;

import android.os.Looper;

import android.provider.Settings;

import android.provider.Settings.Global;

import android.telephony.PhoneStateListener;

import android.telephony.SubscriptionInfo;

import android.telephony.SubscriptionManager;

import android.telephony.TelephonyManager;

import android.testing.TestableLooper;

import android.util.Log;

import com.android.internal.telephony.cdma.EriInfo;

import com.android.settingslib.net.DataUsageController;

import com.android.systemui.SysuiTestCase;

import com.android.systemui.statusbar.phone.SignalDrawable;

import com.android.systemui.statusbar.policy.DeviceProvisionedController.DeviceProvisionedListener;

import com.android.systemui.statusbar.policy.NetworkController.IconState;

import com.android.systemui.statusbar.policy.NetworkController.SignalCallback;

import com.android.systemui.statusbar.policy.NetworkControllerImpl.Config;

import com.android.systemui.statusbar.policy.NetworkControllerImpl.SubscriptionDefaults;

import com.android.systemui.SysuiTestCase;

import org.junit.Before;

import org.junit.Rule;

import org.junit.runner.Description;

import org.mockito.ArgumentCaptor;

import org.mockito.Mockito;

import org.mockito.invocation.InvocationOnMock;

import org.mockito.stubbing.Answer;

import java.io.PrintWriter;

import java.io.StringWriter;

import java.util.ArrayList;

import java.util.List;

import static junit.framework.Assert.assertEquals;

import static org.mockito.Matchers.any;

import static org.mockito.Matchers.anyBoolean;

import static org.mockito.Matchers.anyInt;

import static org.mockito.Matchers.anyString;

import static org.mockito.Matchers.eq;

import static org.mockito.Mockito.doAnswer;

import static org.mockito.Mockito.mock;

import static org.mockito.Mockito.when;

public class NetworkControllerBaseTest extends SysuiTestCase {

    private static final String TAG = "NetworkControllerBaseTest";

    protected static final int DEFAULT_LEVEL = 2;

            mUserCallback = (DeviceProvisionedListener) invocation.getArguments()[0];

            mUserCallback.onUserSetupChanged();

            mUserCallback.onDeviceProvisionedChanged();

            TestableLooper.get(this).processAllMessages();

            return null;

        }).when(mMockProvisionController).addCallback(any());

        mNetworkController = new NetworkControllerImpl(mContext, mMockCm, mMockTm, mMockWm, mMockSm,

                mConfig, Looper.getMainLooper(), mCallbackHandler,

                mConfig, TestableLooper.get(this).getLooper(), mCallbackHandler,

                mock(AccessPointControllerImpl.class), mock(DataUsageController.class),

                mMockSubDefaults, mMockProvisionController);

        setupNetworkController();

      when(mMockCm.isNetworkSupported(ConnectivityManager.TYPE_MOBILE)).thenReturn(false);

      NetworkControllerImpl networkControllerNoMobile

              = new NetworkControllerImpl(mContext, mMockCm, mMockTm, mMockWm, mMockSm,

                        mConfig, mContext.getMainLooper(), mCallbackHandler,

                        mConfig, TestableLooper.get(this).getLooper(), mCallbackHandler,

                        mock(AccessPointControllerImpl.class),

                        mock(DataUsageController.class), mMockSubDefaults,

                        mock(DeviceProvisionedController.class));

import android.support.test.runner.AndroidJUnit4;

import android.telephony.TelephonyManager;

import android.test.suitebuilder.annotation.SmallTest;

import android.testing.AndroidTestingRunner;

import android.testing.TestableLooper;

import android.testing.TestableLooper.RunWithLooper;

import com.android.settingslib.net.DataUsageController;

import org.junit.runner.RunWith;

@SmallTest

@RunWith(AndroidJUnit4.class)

@RunWith(AndroidTestingRunner.class)

@RunWithLooper

public class NetworkControllerDataTest extends NetworkControllerBaseTest {

    @Test

        setConnectivity(NetworkCapabilities.TRANSPORT_CELLULAR, false, false);

        when(mMockProvisionController.isUserSetup(anyInt())).thenReturn(false);

        mUserCallback.onUserSetupChanged();

        TestableLooper.get(this).processAllMessages();

        // Don't show the X until the device is setup.

        verifyDataIndicators(0, 0);

import android.net.NetworkCapabilities;

import android.support.test.runner.AndroidJUnit4;

import android.test.suitebuilder.annotation.SmallTest;

import android.testing.AndroidTestingRunner;

import android.testing.TestableLooper.RunWithLooper;

import com.android.systemui.statusbar.policy.NetworkController.IconState;

import static junit.framework.Assert.assertEquals;

@SmallTest

@RunWith(AndroidJUnit4.class)

@RunWith(AndroidTestingRunner.class)

@RunWithLooper

public class NetworkControllerEthernetTest extends NetworkControllerBaseTest {

    @Test

import android.telephony.SubscriptionInfo;

import android.telephony.TelephonyManager;

import android.test.suitebuilder.annotation.SmallTest;

import android.testing.AndroidTestingRunner;

import android.testing.TestableLooper.RunWithLooper;

import com.android.internal.telephony.PhoneConstants;

import com.android.internal.telephony.TelephonyIntents;

import static org.mockito.Mockito.mock;

@SmallTest

@RunWith(AndroidJUnit4.class)

@RunWith(AndroidTestingRunner.class)

@RunWithLooper

public class NetworkControllerSignalTest extends NetworkControllerBaseTest {

    @Test

import android.net.wifi.WifiManager;

import android.support.test.runner.AndroidJUnit4;

import android.test.suitebuilder.annotation.SmallTest;

import android.testing.AndroidTestingRunner;

import android.testing.TestableLooper.RunWithLooper;

import com.android.systemui.statusbar.policy.NetworkController.IconState;

import static org.mockito.Matchers.anyBoolean;

@SmallTest

@RunWith(AndroidJUnit4.class)

@RunWith(AndroidTestingRunner.class)

@RunWithLooper

public class NetworkControllerWifiTest extends NetworkControllerBaseTest {

    // These match the constants in WifiManager and need to be kept up to date.

    private static final int MIN_RSSI = -100;