NetworkStatsFactoryTest.java revision c86013be936f36c61fa0cdfad95141f6de3c6ef7
14414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey/* 24414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * Copyright (C) 2011 The Android Open Source Project 34414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * 44414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * Licensed under the Apache License, Version 2.0 (the "License"); 54414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * you may not use this file except in compliance with the License. 64414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * You may obtain a copy of the License at 74414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * 84414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * http://www.apache.org/licenses/LICENSE-2.0 94414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * 104414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * Unless required by applicable law or agreed to in writing, software 114414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * distributed under the License is distributed on an "AS IS" BASIS, 124414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 134414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * See the License for the specific language governing permissions and 144414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * limitations under the License. 154414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey */ 164414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 171059c3c30ad96a15695c1a92ae8896e078a6309fJeff Sharkeypackage com.android.internal.net; 184414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 19c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chenimport static android.net.NetworkStats.METERED_NO; 201f7e05eda687279ac9deb894f42ac927bd471ad2Jeff Davidsonimport static android.net.NetworkStats.ROAMING_NO; 218fc27e8b87bd1def854a03d84009143b315d4176Jeff Sharkeyimport static android.net.NetworkStats.SET_ALL; 22b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkeyimport static android.net.NetworkStats.SET_DEFAULT; 23b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkeyimport static android.net.NetworkStats.SET_FOREGROUND; 24fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkeyimport static android.net.NetworkStats.TAG_NONE; 25fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkeyimport static android.net.NetworkStats.UID_ALL; 268568db534118fc14cc28100306d51626464ff319Jesse Wilsonimport static com.android.server.NetworkManagementSocketTagger.kernelToTag; 274414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 284414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport android.content.res.Resources; 294414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport android.net.NetworkStats; 302d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkeyimport android.net.TrafficStats; 314414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport android.test.AndroidTestCase; 324414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 33c86013be936f36c61fa0cdfad95141f6de3c6ef7Lorenzo Colittiimport com.android.frameworks.tests.net.R; 344414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 354414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport java.io.File; 364414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport java.io.FileOutputStream; 37fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkeyimport java.io.FileWriter; 384414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport java.io.InputStream; 394414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport java.io.OutputStream; 404414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 414414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport libcore.io.IoUtils; 424414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkeyimport libcore.io.Streams; 434414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 444414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey/** 451059c3c30ad96a15695c1a92ae8896e078a6309fJeff Sharkey * Tests for {@link NetworkStatsFactory}. 464414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey */ 471059c3c30ad96a15695c1a92ae8896e078a6309fJeff Sharkeypublic class NetworkStatsFactoryTest extends AndroidTestCase { 484414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey private File mTestProc; 491059c3c30ad96a15695c1a92ae8896e078a6309fJeff Sharkey private NetworkStatsFactory mFactory; 504414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 514414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey @Override 524414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey public void setUp() throws Exception { 534414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey super.setUp(); 544414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 55a63ba59260cd1bb3f5c16e395ace45a61f1d4461Jeff Sharkey mTestProc = new File(getContext().getFilesDir(), "proc"); 56fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey if (mTestProc.exists()) { 57a63ba59260cd1bb3f5c16e395ace45a61f1d4461Jeff Sharkey IoUtils.deleteContents(mTestProc); 58fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey } 59fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey 601059c3c30ad96a15695c1a92ae8896e078a6309fJeff Sharkey mFactory = new NetworkStatsFactory(mTestProc); 614414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 624414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 634414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey @Override 644414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey public void tearDown() throws Exception { 651059c3c30ad96a15695c1a92ae8896e078a6309fJeff Sharkey mFactory = null; 664414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 67fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey if (mTestProc.exists()) { 68a63ba59260cd1bb3f5c16e395ace45a61f1d4461Jeff Sharkey IoUtils.deleteContents(mTestProc); 69fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey } 70fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey 714414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey super.tearDown(); 724414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 734414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 744414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey public void testNetworkStatsDetail() throws Exception { 754414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey stageFile(R.raw.xt_qtaguid_typical, new File(mTestProc, "net/xt_qtaguid/stats")); 764414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 771059c3c30ad96a15695c1a92ae8896e078a6309fJeff Sharkey final NetworkStats stats = mFactory.readNetworkStatsDetail(); 78163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertEquals(70, stats.size()); 79163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertStatsEntry(stats, "wlan0", 0, SET_DEFAULT, 0x0, 18621L, 2898L); 80163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertStatsEntry(stats, "wlan0", 10011, SET_DEFAULT, 0x0, 35777L, 5718L); 81163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertStatsEntry(stats, "wlan0", 10021, SET_DEFAULT, 0x7fffff01, 562386L, 49228L); 82163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertStatsEntry(stats, "rmnet1", 10021, SET_DEFAULT, 0x30100000, 219110L, 227423L); 83163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertStatsEntry(stats, "rmnet2", 10001, SET_DEFAULT, 0x0, 1125899906842624L, 984L); 844414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 854414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 864414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey public void testKernelTags() throws Exception { 874414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey assertEquals(0, kernelToTag("0x0000000000000000")); 884414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey assertEquals(0x32, kernelToTag("0x0000003200000000")); 894414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey assertEquals(2147483647, kernelToTag("0x7fffffff00000000")); 904414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey assertEquals(0, kernelToTag("0x0000000000000000")); 914414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey assertEquals(2147483136, kernelToTag("0x7FFFFE0000000000")); 922d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkey 932d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkey assertEquals(0, kernelToTag("0x0")); 942d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkey assertEquals(0, kernelToTag("0xf00d")); 952d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkey assertEquals(1, kernelToTag("0x100000000")); 962d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkey assertEquals(14438007, kernelToTag("0xdc4e7700000000")); 972d6c580262ba1c19fc4f20c3566889e332fbdcb6Jeff Sharkey assertEquals(TrafficStats.TAG_SYSTEM_DOWNLOAD, kernelToTag("0xffffff0100000000")); 984414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 994414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 100b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey public void testNetworkStatsWithSet() throws Exception { 101163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey stageFile(R.raw.xt_qtaguid_typical, new File(mTestProc, "net/xt_qtaguid/stats")); 102b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey 1031059c3c30ad96a15695c1a92ae8896e078a6309fJeff Sharkey final NetworkStats stats = mFactory.readNetworkStatsDetail(); 104163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertEquals(70, stats.size()); 105c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chen assertStatsEntry(stats, "rmnet1", 10021, SET_DEFAULT, 0x30100000, 219110L, 578L, 227423L, 106c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chen 676L); 107163e6443f27884a9bfcb9a48ef606dc635852c23Jeff Sharkey assertStatsEntry(stats, "rmnet1", 10021, SET_FOREGROUND, 0x30100000, 742L, 3L, 1265L, 3L); 108b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey } 109b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey 110ae2c1810839430c84e4a1172580d9c4b18f568caJeff Sharkey public void testNetworkStatsSingle() throws Exception { 111c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chen stageFile(R.raw.xt_qtaguid_iface_typical, 112c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chen new File(mTestProc, "net/xt_qtaguid/iface_stat_all")); 113ae2c1810839430c84e4a1172580d9c4b18f568caJeff Sharkey 114e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey final NetworkStats stats = mFactory.readNetworkStatsSummaryDev(); 115ae2c1810839430c84e4a1172580d9c4b18f568caJeff Sharkey assertEquals(6, stats.size()); 1168fc27e8b87bd1def854a03d84009143b315d4176Jeff Sharkey assertStatsEntry(stats, "rmnet0", UID_ALL, SET_ALL, TAG_NONE, 2112L, 24L, 700L, 10L); 1178fc27e8b87bd1def854a03d84009143b315d4176Jeff Sharkey assertStatsEntry(stats, "test1", UID_ALL, SET_ALL, TAG_NONE, 6L, 8L, 10L, 12L); 1188fc27e8b87bd1def854a03d84009143b315d4176Jeff Sharkey assertStatsEntry(stats, "test2", UID_ALL, SET_ALL, TAG_NONE, 1L, 2L, 3L, 4L); 119ae2c1810839430c84e4a1172580d9c4b18f568caJeff Sharkey } 120ae2c1810839430c84e4a1172580d9c4b18f568caJeff Sharkey 121e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey public void testNetworkStatsXt() throws Exception { 122e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey stageFile(R.raw.xt_qtaguid_iface_fmt_typical, 123e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey new File(mTestProc, "net/xt_qtaguid/iface_stat_fmt")); 124e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey 125e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey final NetworkStats stats = mFactory.readNetworkStatsSummaryXt(); 126e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey assertEquals(3, stats.size()); 127e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey assertStatsEntry(stats, "rmnet0", UID_ALL, SET_ALL, TAG_NONE, 6824L, 16L, 5692L, 10L); 128c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chen assertStatsEntry(stats, "rmnet1", UID_ALL, SET_ALL, TAG_NONE, 11153922L, 8051L, 190226L, 129c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chen 2468L); 130e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey assertStatsEntry(stats, "rmnet2", UID_ALL, SET_ALL, TAG_NONE, 4968L, 35L, 3081L, 39L); 131e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey } 132e8914c36276710de50b347c1e6aecfa45d6a56cdJeff Sharkey 1334414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey /** 1344414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * Copy a {@link Resources#openRawResource(int)} into {@link File} for 1354414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey * testing purposes. 1364414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey */ 1374414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey private void stageFile(int rawId, File file) throws Exception { 1384414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey new File(file.getParent()).mkdirs(); 1394414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey InputStream in = null; 1404414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey OutputStream out = null; 1414414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey try { 1424414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey in = getContext().getResources().openRawResource(rawId); 1434414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey out = new FileOutputStream(file); 1444414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey Streams.copy(in, out); 1454414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } finally { 1464414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey IoUtils.closeQuietly(in); 1474414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey IoUtils.closeQuietly(out); 1484414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 1494414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 1504414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 151fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey private void stageLong(long value, File file) throws Exception { 152fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey new File(file.getParent()).mkdirs(); 153fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey FileWriter out = null; 154fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey try { 155fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey out = new FileWriter(file); 156fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey out.write(Long.toString(value)); 157fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey } finally { 158fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey IoUtils.closeQuietly(out); 159fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey } 160fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey } 161fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey 162b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey private static void assertStatsEntry(NetworkStats stats, String iface, int uid, int set, 163b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey int tag, long rxBytes, long txBytes) { 164c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chen final int i = stats.findIndex(iface, uid, set, tag, METERED_NO, ROAMING_NO); 165b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey final NetworkStats.Entry entry = stats.getValues(i, null); 166b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey assertEquals("unexpected rxBytes", rxBytes, entry.rxBytes); 167b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey assertEquals("unexpected txBytes", txBytes, entry.txBytes); 168b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey } 169b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey 170b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey private static void assertStatsEntry(NetworkStats stats, String iface, int uid, int set, 171b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey int tag, long rxBytes, long rxPackets, long txBytes, long txPackets) { 172c926b7399a041fb82f15baa889bbf7ee0fa45643Stephen Chen final int i = stats.findIndex(iface, uid, set, tag, METERED_NO, ROAMING_NO); 173fd8be3e5e7420f3cca591daeec8a44487f5f65aaJeff Sharkey final NetworkStats.Entry entry = stats.getValues(i, null); 174b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey assertEquals("unexpected rxBytes", rxBytes, entry.rxBytes); 175b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey assertEquals("unexpected rxPackets", rxPackets, entry.rxPackets); 176b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey assertEquals("unexpected txBytes", txBytes, entry.txBytes); 177b5d55e302d2253e4bfb233ea705caf258cdc4cb9Jeff Sharkey assertEquals("unexpected txPackets", txPackets, entry.txPackets); 1784414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey } 1794414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey 1804414cea13908b8230640f84ef39603d68ff9c377Jeff Sharkey} 181