From 7ef0a1754459b400beb4a86e43cac6dfb1921fde Mon Sep 17 00:00:00 2001 From: Junyu Lai Date: Fri, 7 Jan 2022 12:47:56 +0000 Subject: [PATCH] [MS44.2] Add unit test for API to query tagged UID summary Test: atest NetworkStatsManagerTest Bug: 204830222 Change-Id: I0b712141b7c644205bba3a98ebf39803f0d05306 --- .../app/usage/NetworkStatsManagerTest.java | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/tests/unit/java/android/app/usage/NetworkStatsManagerTest.java b/tests/unit/java/android/app/usage/NetworkStatsManagerTest.java index 08a3007fbf..6e51069ad2 100644 --- a/tests/unit/java/android/app/usage/NetworkStatsManagerTest.java +++ b/tests/unit/java/android/app/usage/NetworkStatsManagerTest.java @@ -220,6 +220,26 @@ public class NetworkStatsManagerTest { 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) { assertEquals(expected.uid, actual.getUid()); assertEquals(expected.rxBytes, actual.getRxBytes());