LineageOS/android_packages_modules_Connectivity
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge "[MS44.2] Add unit test for API to query tagged UID summary"

Browse Source
This commit is contained in:
Junyu Lai
2022-01-13 11:25:15 +00:00
committed by Gerrit Code Review
parent b209f64b3c 7ef0a17544
commit 5db1ea12b5
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/unit/java/android/app/usage/NetworkStatsManagerTest.java
View File

@@ -220,6 +220,26 @@ public class NetworkStatsManagerTest {
TEST_SUBSCRIBER_ID)); TEST_SUBSCRIBER_ID));
} }
@Test
public void testQueryTaggedSummary() throws Exception {
final long startTime = 1;
final long endTime = 100;
reset(mStatsSession);
when(mService.openSessionForUsageStats(anyInt(), anyString())).thenReturn(mStatsSession);
when(mStatsSession.getTaggedSummaryForAllUid(any(NetworkTemplate.class),
anyLong(), anyLong()))
.thenReturn(new android.net.NetworkStats(0, 0));
final NetworkTemplate template = new NetworkTemplate.Builder(NetworkTemplate.MATCH_MOBILE)
.setMeteredness(NetworkStats.Bucket.METERED_YES).build();
NetworkStats stats = mManager.queryTaggedSummary(template, startTime, endTime);
verify(mStatsSession, times(1)).getTaggedSummaryForAllUid(
eq(template), eq(startTime), eq(endTime));
assertFalse(stats.hasNextBucket());
}
private void assertBucketMatches(Entry expected, NetworkStats.Bucket actual) { private void assertBucketMatches(Entry expected, NetworkStats.Bucket actual) {
assertEquals(expected.uid, actual.getUid()); assertEquals(expected.uid, actual.getUid());
assertEquals(expected.rxBytes, actual.getRxBytes()); assertEquals(expected.rxBytes, actual.getRxBytes());