commit dbbeb9937076fe6b6a7167a6e9e6bc1ad1ff5a76
author lucaslin <lucaslin@google.com> Mon Jul 25 15:17:47 2022 +0800
committer Lucas Lin <lucaslin@google.com> Wed Jul 27 01:58:37 2022 +0000
tree 1ce8fefc58398588eb763524395580706c33f65b
parent 6745ba4245a47bd0f056e97034d203edbf72b792

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

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));