commit b00f79db7ebdb5f013c6b0a9be069359d7236b71
author lucaslin <lucaslin@google.com> Mon Jul 25 15:17:47 2022 +0800
committer Android Build Coastguard Worker <android-build-coastguard-worker@google.com> Fri Aug 19 18:44:40 2022 +0000
tree d4b36a6be267bcd4deba6289c8e4a64e4162d428
parent 1106ab86764fc0a193b332cc7ce91db18b2b20d9

Test that VpnManager events will never be sent to Settings VPNs

This change tries to mock there is an IkeTimeoutException when
Settings VPN is establishing connection and see if the VpnManager
events will be sent to Settings VPN or not.

Bug: 236315805
Test: atest FrameworksNetTests:VpnTest
Change-Id: I1706ceb129ef95f44d45754665b9dbcf9b6fdfd4
Merged-In: I1706ceb129ef95f44d45754665b9dbcf9b6fdfd4
Merged-In: I0ee4a70ded4937a1573d34cddab552f762e4fc18
(cherry picked from commit dbbeb9937076fe6b6a7167a6e9e6bc1ad1ff5a76)
Merged-In: I1706ceb129ef95f44d45754665b9dbcf9b6fdfd4

tests/unit/java/com/android/server/connectivity/VpnTest.java

diff --git a/tests/unit/java/com/android/server/connectivity/VpnTest.java b/tests/unit/java/com/android/server/connectivity/VpnTest.java
index eb35469..6266d8c 100644
--- a/tests/unit/java/com/android/server/connectivity/VpnTest.java
+++ b/tests/unit/java/com/android/server/connectivity/VpnTest.java
@@ -1422,6 +1422,31 @@
         assertEquals(LegacyVpnInfo.STATE_FAILED, vpn.getLegacyVpnInfo().state);
     }
 
+    @Test
+    public void testVpnManagerEventWillNotBeSentToSettingsVpn() throws Exception {
+        startLegacyVpn(createVpn(primaryUser.id), mVpnProfile);
+        triggerOnAvailableAndGetCallback();
+
+        verifyInterfaceSetCfgWithFlags(IF_STATE_UP);
+
+        final IkeNonProtocolException exception = mock(IkeNonProtocolException.class);
+        final IkeTimeoutException ikeTimeoutException =
+                new IkeTimeoutException("IkeTimeoutException");
+        when(exception.getCause()).thenReturn(ikeTimeoutException);
+
+        final ArgumentCaptor<IkeSessionCallback> captor =
+                ArgumentCaptor.forClass(IkeSessionCallback.class);
+        verify(mIkev2SessionCreator, timeout(TEST_TIMEOUT_MS))
+                .createIkeSession(any(), any(), any(), any(), captor.capture(), any());
+        final IkeSessionCallback ikeCb = captor.getValue();
+        ikeCb.onClosedWithException(exception);
+
+        // userContext is the same Context that VPN is using, see createVpn().
+        final Context userContext =
+                mContext.createContextAsUser(UserHandle.of(primaryUser.id), 0 /* flags */);
+        verify(userContext, never()).startService(any());
+    }
+
     private void setAndVerifyAlwaysOnPackage(Vpn vpn, int uid, boolean lockdownEnabled) {
         assertTrue(vpn.setAlwaysOnPackage(TEST_VPN_PKG, lockdownEnabled, null));