Merge "Fixed NetworkStatsServiceTest and converted it to JUnit4."

import static android.text.format.DateUtils.HOUR_IN_MILLIS;

import static android.text.format.DateUtils.HOUR_IN_MILLIS;

import static android.text.format.DateUtils.MINUTE_IN_MILLIS;

import static android.text.format.DateUtils.MINUTE_IN_MILLIS;

import static android.text.format.DateUtils.WEEK_IN_MILLIS;

import static android.text.format.DateUtils.WEEK_IN_MILLIS;

import static com.android.server.net.NetworkStatsService.ACTION_NETWORK_STATS_POLL;

import static com.android.server.net.NetworkStatsService.ACTION_NETWORK_STATS_POLL;

import static org.easymock.EasyMock.anyInt;

import static org.easymock.EasyMock.anyInt;

import static org.easymock.EasyMock.anyLong;

import static org.easymock.EasyMock.anyLong;

import static org.easymock.EasyMock.anyObject;

import static org.easymock.EasyMock.anyObject;

import static org.easymock.EasyMock.eq;

import static org.easymock.EasyMock.eq;

import static org.easymock.EasyMock.expect;

import static org.easymock.EasyMock.expect;

import static org.easymock.EasyMock.expectLastCall;

import static org.easymock.EasyMock.expectLastCall;

import static org.easymock.EasyMock.isA;

import static org.junit.Assert.assertEquals;

import static org.junit.Assert.assertNull;

import static org.junit.Assert.assertTrue;

import static org.junit.Assert.fail;

import android.app.AlarmManager;

import android.app.AlarmManager;

import android.app.IAlarmListener;

import android.app.IAlarmManager;

import android.app.PendingIntent;

import android.app.usage.NetworkStatsManager;

import android.app.usage.NetworkStatsManager;

import android.content.Context;

import android.content.Context;

import android.content.Intent;

import android.content.Intent;

import android.net.INetworkManagementEventObserver;

import android.net.INetworkManagementEventObserver;

import android.net.INetworkStatsSession;

import android.net.INetworkStatsSession;

import android.net.LinkProperties;

import android.net.LinkProperties;

import android.net.NetworkCapabilities;

import android.net.NetworkInfo;

import android.net.NetworkInfo;

import android.net.NetworkInfo.DetailedState;

import android.net.NetworkInfo.DetailedState;

import android.net.NetworkState;

import android.net.NetworkState;

import android.os.MessageQueue.IdleHandler;

import android.os.MessageQueue.IdleHandler;

import android.os.Message;

import android.os.Message;

import android.os.PowerManager;

import android.os.PowerManager;

import android.os.WorkSource;

import android.support.test.InstrumentationRegistry;

import android.support.test.runner.AndroidJUnit4;

import android.telephony.TelephonyManager;

import android.telephony.TelephonyManager;

import android.test.AndroidTestCase;

import android.test.AndroidTestCase;

import android.test.suitebuilder.annotation.LargeTest;

import android.test.suitebuilder.annotation.Suppress;

import android.util.TrustedTime;

import android.util.TrustedTime;

import com.android.internal.net.VpnInfo;

import com.android.internal.net.VpnInfo;

import org.easymock.Capture;

import org.easymock.Capture;

import org.easymock.EasyMock;

import org.easymock.EasyMock;

import org.junit.After;

import org.junit.Before;

import org.junit.Test;

import org.junit.runner.RunWith;

import java.io.File;

import java.io.File;

import java.util.ArrayList;

import java.util.ArrayList;

 * TODO: This test is really brittle, largely due to overly-strict use of Easymock.

 * TODO: This test is really brittle, largely due to overly-strict use of Easymock.

 * Rewrite w/ Mockito.

 * Rewrite w/ Mockito.

 */

 */

@LargeTest

@RunWith(AndroidJUnit4.class)

public class NetworkStatsServiceTest extends AndroidTestCase {

public class NetworkStatsServiceTest {

    private static final String TAG = "NetworkStatsServiceTest";

    private static final String TAG = "NetworkStatsServiceTest";

    private static final String TEST_IFACE = "test0";

    private static final String TEST_IFACE = "test0";

    private INetworkStatsSession mSession;

    private INetworkStatsSession mSession;

    private INetworkManagementEventObserver mNetworkObserver;

    private INetworkManagementEventObserver mNetworkObserver;

    @Override

    @Before

    public void setUp() throws Exception {

    public void setUp() throws Exception {

        super.setUp();

        final Context context = InstrumentationRegistry.getContext();

        mServiceContext = new BroadcastInterceptingContext(getContext());

        mServiceContext = new BroadcastInterceptingContext(context);

        mStatsDir = getContext().getFilesDir();

        mStatsDir = context.getFilesDir();

        if (mStatsDir.exists()) {

        if (mStatsDir.exists()) {

            IoUtils.deleteContents(mStatsDir);

            IoUtils.deleteContents(mStatsDir);

        }

        }

    }

    }

    @Override

    @After

    public void tearDown() throws Exception {

    public void tearDown() throws Exception {

        IoUtils.deleteContents(mStatsDir);

        IoUtils.deleteContents(mStatsDir);

        mSession.close();

        mSession.close();

        mService = null;

        mService = null;

        super.tearDown();

    }

    }

    @Test

    public void testNetworkStatsWifi() throws Exception {

    public void testNetworkStatsWifi() throws Exception {

        // pretend that wifi network comes online; service should ask about full

        // pretend that wifi network comes online; service should ask about full

        // network state, and poll any existing interfaces before updating.

        // network state, and poll any existing interfaces before updating.

    }

    }

    @Test

    public void testStatsRebootPersist() throws Exception {

    public void testStatsRebootPersist() throws Exception {

        assertStatsFilesExist(false);

        assertStatsFilesExist(false);

    }

    }

    // TODO: simulate reboot to test bucket resize

    // TODO: simulate reboot to test bucket resize

    @Suppress

    // @Test

    public void testStatsBucketResize() throws Exception {

    public void testStatsBucketResize() throws Exception {

        NetworkStatsHistory history = null;

        NetworkStatsHistory history = null;

    }

    }

    @Test

    public void testUidStatsAcrossNetworks() throws Exception {

    public void testUidStatsAcrossNetworks() throws Exception {

        // pretend first mobile network comes online

        // pretend first mobile network comes online

        expectCurrentTime();

        expectCurrentTime();

    }

    }

    @Test

    public void testUidRemovedIsMoved() throws Exception {

    public void testUidRemovedIsMoved() throws Exception {

        // pretend that network comes online

        // pretend that network comes online

        expectCurrentTime();

        expectCurrentTime();

    }

    }

    @Test

    public void testUid3g4gCombinedByTemplate() throws Exception {

    public void testUid3g4gCombinedByTemplate() throws Exception {

        // pretend that network comes online

        // pretend that network comes online

        expectCurrentTime();

        expectCurrentTime();

        verifyAndReset();

        verifyAndReset();

    }

    }

    @Test

    public void testSummaryForAllUid() throws Exception {

    public void testSummaryForAllUid() throws Exception {

        // pretend that network comes online

        // pretend that network comes online

        expectCurrentTime();

        expectCurrentTime();

        verifyAndReset();

        verifyAndReset();

    }

    }

    @Test

    public void testForegroundBackground() throws Exception {

    public void testForegroundBackground() throws Exception {

        // pretend that network comes online

        // pretend that network comes online

        expectCurrentTime();

        expectCurrentTime();

        verifyAndReset();

        verifyAndReset();

    }

    }

    @Test

    public void testRoaming() throws Exception {

    public void testRoaming() throws Exception {

        // pretend that network comes online

        // pretend that network comes online

        expectCurrentTime();

        expectCurrentTime();

        verifyAndReset();

        verifyAndReset();

    }

    }

    @Test

    public void testTethering() throws Exception {

    public void testTethering() throws Exception {

        // pretend first mobile network comes online

        // pretend first mobile network comes online

        expectCurrentTime();

        expectCurrentTime();

    }

    }

    @Test

    public void testRegisterUsageCallback() throws Exception {

    public void testRegisterUsageCallback() throws Exception {

        // pretend that wifi network comes online; service should ask about full

        // pretend that wifi network comes online; service should ask about full

        // network state, and poll any existing interfaces before updating.

        // network state, and poll any existing interfaces before updating.

        EasyMock.verify(mockBinder);

        EasyMock.verify(mockBinder);

    }

    }

    @Test

    public void testUnregisterUsageCallback_unknown_noop() throws Exception {

    public void testUnregisterUsageCallback_unknown_noop() throws Exception {

        String callingPackage = "the.calling.package";

        String callingPackage = "the.calling.package";

        long thresholdInBytes = 10 * 1024 * 1024;  // 10 MB

        long thresholdInBytes = 10 * 1024 * 1024;  // 10 MB

        info.setDetailedState(DetailedState.CONNECTED, null, null);

        info.setDetailedState(DetailedState.CONNECTED, null, null);

        final LinkProperties prop = new LinkProperties();

        final LinkProperties prop = new LinkProperties();

        prop.setInterfaceName(TEST_IFACE);

        prop.setInterfaceName(TEST_IFACE);

        return new NetworkState(info, prop, null, null, null, TEST_SSID);

        final NetworkCapabilities capabilities = new NetworkCapabilities();

        return new NetworkState(info, prop, capabilities, null, null, TEST_SSID);

    }

    }

    private static NetworkState buildMobile3gState(String subscriberId) {

    private static NetworkState buildMobile3gState(String subscriberId) {

        info.setRoaming(isRoaming);

        info.setRoaming(isRoaming);

        final LinkProperties prop = new LinkProperties();

        final LinkProperties prop = new LinkProperties();

        prop.setInterfaceName(TEST_IFACE);

        prop.setInterfaceName(TEST_IFACE);

        return new NetworkState(info, prop, null, null, subscriberId, null);

        final NetworkCapabilities capabilities = new NetworkCapabilities();

        return new NetworkState(info, prop, capabilities, null, subscriberId, null);

    }

    }

    private static NetworkState buildMobile4gState(String iface) {

    private static NetworkState buildMobile4gState(String iface) {

        info.setDetailedState(DetailedState.CONNECTED, null, null);

        info.setDetailedState(DetailedState.CONNECTED, null, null);

        final LinkProperties prop = new LinkProperties();

        final LinkProperties prop = new LinkProperties();

        prop.setInterfaceName(iface);

        prop.setInterfaceName(iface);

        return new NetworkState(info, prop, null, null, null, null);

        final NetworkCapabilities capabilities = new NetworkCapabilities();

        return new NetworkState(info, prop, capabilities, null, null, null);

    }

    }

    private NetworkStats buildEmptyStats() {

    private NetworkStats buildEmptyStats() {