commit 39654c33cf1a9c1125ba99c8384d6514ba8132c7
author Hansen Kurli <hkurli@google.com> Tue Sep 12 15:05:46 2023 +0800
committer Hansen Kurli <hkurli@google.com> Fri Oct 13 15:30:43 2023 +0800
tree c394ebd992a1b8ec96c9b0e7e2a18c6d5e7cccc4
parent ed972a6fd267471dc729fdef36711e76f0981cfc

Mock Vpn.setLockdown()

Call setRequireVpnForUids directly to enable lockdown instead of
calling the real Vpn method.

Bug: 230548427
Test: atest FrameworksNetTests
Change-Id: I91ec59f7542d145e9250a7e7a414593db3d99424

tests/unit/java/com/android/server/ConnectivityServiceTest.java

diff --git a/tests/unit/java/com/android/server/ConnectivityServiceTest.java b/tests/unit/java/com/android/server/ConnectivityServiceTest.java
index af0220e..bc840c2 100755
--- a/tests/unit/java/com/android/server/ConnectivityServiceTest.java
+++ b/tests/unit/java/com/android/server/ConnectivityServiceTest.java
@@ -10206,7 +10206,8 @@
         // Init lockdown state to simulate LockdownVpnTracker behavior.
         mCm.setLegacyLockdownVpnEnabled(true);
         mMockVpn.setEnableTeardown(false);
-        mMockVpn.setLockdown(true);
+        final Set<Range<Integer>> ranges = UidRange.toIntRanges(Set.of(PRIMARY_UIDRANGE));
+        mCm.setRequireVpnForUids(true /* requireVpn */, ranges);
 
         // Bring up a network.
         final LinkProperties cellLp = new LinkProperties();