test: fix connection event race condition in present proof test (#620)

hyperledger · Aug 8, 2023 · faee7e4 · faee7e4
1 parent e79f503
commit faee7e4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/e2e-tests/src/test/kotlin/features/connection/ConnectionSteps.kt b/tests/e2e-tests/src/test/kotlin/features/connection/ConnectionSteps.kt
@@ -101,8 +101,8 @@ class ConnectionSteps {
     fun inviterReceivesTheConnectionRequest(inviter: Actor) {
         wait(
             {
-                val lastEvent = ListenToEvents.`as`(inviter).connectionEvents.last()
-                lastEvent.data.thid == inviter.recall<String>("thid") &&
+                val lastEvent = ListenToEvents.`as`(inviter).connectionEvents.lastOrNull()
+                lastEvent != null && lastEvent.data.thid == inviter.recall<String>("thid") &&
                         lastEvent.data.state == ConnectionState.CONNECTION_RESPONSE_SENT
             },
             "Inviter connection didn't reach ${ConnectionState.CONNECTION_RESPONSE_SENT} state",
@@ -114,8 +114,8 @@ class ConnectionSteps {
         // Bob (Holder) receives final connection response
         wait(
             {
-                val lastEvent = ListenToEvents.`as`(invitee).connectionEvents.last()
-                lastEvent.data.thid == invitee.recall<String>("thid") &&
+                val lastEvent = ListenToEvents.`as`(invitee).connectionEvents.lastOrNull()
+                lastEvent != null && lastEvent.data.thid == invitee.recall<String>("thid") &&
                         lastEvent.data.state == ConnectionState.CONNECTION_RESPONSE_RECEIVED
             },
             "Invitee connection didn't reach ${ConnectionState.CONNECTION_RESPONSE_RECEIVED} state.",