Fix TestConnect TypeExit/TypeClose race

ContainerSSH · Aug 12, 2024 · 69b37a7 · 69b37a7
1 parent 226cdb9
commit 69b37a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/internal/auditlogintegration/integration_test.go b/internal/auditlogintegration/integration_test.go
@@ -152,7 +152,7 @@ func checkStoredAuditMessages(t *testing.T, dir string, logger log.Logger) {
 	assert.Equal(t, message.TypeHandshakeSuccessful, messages[3].MessageType)
 	assert.Equal(t, message.TypeNewChannelSuccessful, messages[4].MessageType)
 	assert.Equal(t, message.TypeChannelRequestShell, messages[5].MessageType)
-	assert.Equal(t, message.TypeClose, messages[6].MessageType)
+	assert.True(t, messages[6].MessageType == message.TypeExit || messages[6].MessageType == message.TypeClose)
 	assert.True(t, messages[7].MessageType == message.TypeExit || messages[7].MessageType == message.TypeDisconnect)
 	assert.True(t, messages[8].MessageType == message.TypeExit || messages[8].MessageType == message.TypeDisconnect)
 }