commit 6f3df74e07b9f5394f4fb81f3b852b71d7e0dcff
author Cody Kesting <ckesting@google.com> Tue Nov 17 21:42:40 2020 +0000
committer Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> Tue Nov 17 21:42:40 2020 +0000
tree 4e7ddf70bb3c1ba69ea678b4cca11789a2372ded
parent eb59b1abb9014b3dd78dbbaf10035a097decac95
parent 596a918c0e128231ac288ea9d8a0e99e1afe38ea

Merge "Verify that IkeNetworkDiedException is for the right Network." am: 596a918c0e

Original change: https://android-review.googlesource.com/c/platform/packages/modules/IPsec/+/1496913

Change-Id: I9af8a64699738bfa4905f10b8ea2c18d0dbf1fe5

tests/iketests/src/java/com/android/internal/net/ipsec/ike/IkeSessionStateMachineTest.java

diff --git a/tests/iketests/src/java/com/android/internal/net/ipsec/ike/IkeSessionStateMachineTest.java b/tests/iketests/src/java/com/android/internal/net/ipsec/ike/IkeSessionStateMachineTest.java
index c853cf8..b9c31f3 100644
--- a/tests/iketests/src/java/com/android/internal/net/ipsec/ike/IkeSessionStateMachineTest.java
+++ b/tests/iketests/src/java/com/android/internal/net/ipsec/ike/IkeSessionStateMachineTest.java
@@ -5349,7 +5349,10 @@
         IkeDefaultNetworkCallback callback = verifyMobikeEnabled(true /* doesPeerSupportMobike */);
         callback.onLost(mMockDefaultNetwork);
 
-        verify(mMockIkeSessionCallback).onError(any(IkeNetworkDiedException.class));
+        ArgumentCaptor<IkeException> exceptionCaptor = ArgumentCaptor.forClass(IkeException.class);
+        verify(mMockIkeSessionCallback).onError(exceptionCaptor.capture());
+        IkeNetworkDiedException cause = (IkeNetworkDiedException) exceptionCaptor.getValue();
+        assertEquals(mMockDefaultNetwork, cause.getNetwork());
     }
 
     @Test