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

Merge "Adjust NetworkStatsServiceTest to mock out the JNI."

Browse Source
This commit is contained in:
Treehugger Robot
2022-01-28 11:51:44 +00:00
committed by Gerrit Code Review
parent dd5bc74532 9fdbb7ec7d
commit c5ad7cd775
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/unit/java/com/android/server/net/NetworkStatsServiceTest.java
View File

@@ -198,6 +198,7 @@ public class NetworkStatsServiceTest extends NetworkStatsBaseTest {
@Mock @Mock
private LocationPermissionChecker mLocationPermissionChecker; private LocationPermissionChecker mLocationPermissionChecker;
private @Mock IBpfMap<U32, U8> mUidCounterSetMap; private @Mock IBpfMap<U32, U8> mUidCounterSetMap;
private @Mock NetworkStatsService.TagStatsDeleter mTagStatsDeleter;
private NetworkStatsService mService; private NetworkStatsService mService;
private INetworkStatsSession mSession; private INetworkStatsSession mSession;
@@ -358,6 +359,11 @@ public class NetworkStatsServiceTest extends NetworkStatsBaseTest {
public IBpfMap<U32, U8> getUidCounterSetMap() { public IBpfMap<U32, U8> getUidCounterSetMap() {
return mUidCounterSetMap; return mUidCounterSetMap;
} }
@Override
public NetworkStatsService.TagStatsDeleter getTagStatsDeleter() {
return mTagStatsDeleter;
}
}; };
} }
@@ -696,8 +702,10 @@ public class NetworkStatsServiceTest extends NetworkStatsBaseTest {
final Intent intent = new Intent(ACTION_UID_REMOVED); final Intent intent = new Intent(ACTION_UID_REMOVED);
intent.putExtra(EXTRA_UID, UID_BLUE); intent.putExtra(EXTRA_UID, UID_BLUE);
mServiceContext.sendBroadcast(intent); mServiceContext.sendBroadcast(intent);
verify(mTagStatsDeleter).deleteTagData(UID_BLUE);
intent.putExtra(EXTRA_UID, UID_RED); intent.putExtra(EXTRA_UID, UID_RED);
mServiceContext.sendBroadcast(intent); mServiceContext.sendBroadcast(intent);
verify(mTagStatsDeleter).deleteTagData(UID_RED);
// existing uid and total should remain unchanged; but removed UID // existing uid and total should remain unchanged; but removed UID
// should be gone completely. // should be gone completely.