From 3ae29d455866ce358b42617134610415c04e0d53 Mon Sep 17 00:00:00 2001 From: Tsu Chiang Chuang Date: Tue, 6 Sep 2011 16:37:51 -0700 Subject: [PATCH] idle test app for bandwidth usage. Change-Id: Ib35b27617369b360972e653db70bf5cf9303ddf1 --- tests/DataIdleTest/Android.mk | 29 ++++++ tests/DataIdleTest/AndroidManifest.xml | 32 +++++++ .../com/android/tests/dataidle/DataIdleTest.java | 100 +++++++++++++++++++++ 3 files changed, 161 insertions(+) create mode 100644 tests/DataIdleTest/Android.mk create mode 100644 tests/DataIdleTest/AndroidManifest.xml create mode 100644 tests/DataIdleTest/src/com/android/tests/dataidle/DataIdleTest.java diff --git a/tests/DataIdleTest/Android.mk b/tests/DataIdleTest/Android.mk new file mode 100644 index 000000000000..acb46c545cc5 --- /dev/null +++ b/tests/DataIdleTest/Android.mk @@ -0,0 +1,29 @@ +# +# Copyright (C) 2011 The Android Open Source Project +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# +LOCAL_PATH := $(call my-dir) +include $(CLEAR_VARS) + +# We only want this apk build for tests. +LOCAL_MODULE_TAGS := tests + +LOCAL_PACKAGE_NAME := DataIdleTest +LOCAL_JAVA_LIBRARIES := android.test.runner +LOCAL_SRC_FILES := $(call all-java-files-under, src) + +# We need to sign it to get access to the network usage history. +LOCAL_CERTIFICATE := platform + +include $(BUILD_PACKAGE) diff --git a/tests/DataIdleTest/AndroidManifest.xml b/tests/DataIdleTest/AndroidManifest.xml new file mode 100644 index 000000000000..2792eeca2ad3 --- /dev/null +++ b/tests/DataIdleTest/AndroidManifest.xml @@ -0,0 +1,32 @@ + + + + + + + + + + + + + + diff --git a/tests/DataIdleTest/src/com/android/tests/dataidle/DataIdleTest.java b/tests/DataIdleTest/src/com/android/tests/dataidle/DataIdleTest.java new file mode 100644 index 000000000000..637f0d2841f6 --- /dev/null +++ b/tests/DataIdleTest/src/com/android/tests/dataidle/DataIdleTest.java @@ -0,0 +1,100 @@ +/* + * Copyright (C) 2011 The Android Open Source Project + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package com.android.tests.dataidle; + +import android.content.Context; +import android.net.INetworkStatsService; +import android.net.NetworkStats.Entry; +import android.net.NetworkTemplate; +import android.net.NetworkStats; +import android.os.Bundle; +import android.os.RemoteException; +import android.os.ServiceManager; +import android.telephony.TelephonyManager; +import android.test.InstrumentationTestCase; +import android.test.InstrumentationTestRunner; +import android.util.Log; + +/** + * A test that dumps data usage to instrumentation out, used for measuring data usage for idle + * devices. + */ +public class DataIdleTest extends InstrumentationTestCase { + + private TelephonyManager mTelephonyManager; + private INetworkStatsService mStatsService; + + private static final String LOG_TAG = "DataIdleTest"; + private final static int INSTRUMENTATION_IN_PROGRESS = 2; + + protected void setUp() throws Exception { + super.setUp(); + Context c = getInstrumentation().getTargetContext(); + mStatsService = INetworkStatsService.Stub.asInterface( + ServiceManager.getService(Context.NETWORK_STATS_SERVICE)); + mTelephonyManager = (TelephonyManager) c.getSystemService(Context.TELEPHONY_SERVICE); + } + + /** + * Test that dumps all the data usage metrics for wifi to instrumentation out. + */ + public void testWifiIdle() { + NetworkTemplate template = NetworkTemplate.buildTemplateWifi(); + fetchStats(template); + } + + /** + * Test that dumps all the data usage metrics for all mobile to instrumentation out. + */ + public void testMobile() { + String subscriberId = mTelephonyManager.getSubscriberId(); + NetworkTemplate template = NetworkTemplate.buildTemplateMobileAll(subscriberId); + fetchStats(template); + } + + /** + * Helper method that fetches all the network stats available and reports it + * to instrumentation out. + * @param template {link {@link NetworkTemplate} to match. + */ + private void fetchStats(NetworkTemplate template) { + try { + NetworkStats stats = mStatsService.getSummaryForAllUid(template, Long.MIN_VALUE, + Long.MAX_VALUE, false); + reportStats(stats); + } catch (RemoteException e) { + Log.w(LOG_TAG, "Failed to fetch network stats for wifi."); + } + } + + /** + * Print network data usage stats to instrumentation out + * @param stats {@link NetworkorStats} to print + */ + void reportStats(NetworkStats stats) { + for (int i = 0; i < stats.size(); ++i) { + Entry statsEntry = stats.getValues(i, null); + Bundle result = new Bundle(); + result.putInt("uid", statsEntry.uid); + result.putInt("tag", statsEntry.tag); + result.putInt("set", statsEntry.set); + result.putString("iface", statsEntry.iface); + result.putLong("rxBytes", statsEntry.rxBytes); + result.putLong("txBytes", statsEntry.txBytes); + getInstrumentation().sendStatus(INSTRUMENTATION_IN_PROGRESS, result); + } + } +} -- 2.11.0