Test fixes

ContainerSSH · Jan 15, 2021 · 13cb69e · 13cb69e
1 parent 1890028
commit 13cb69e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/integration_test.go b/integration_test.go
@@ -150,8 +150,9 @@ 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.TypeExit, messages[6].MessageType)
-	assert.Equal(t, message.TypeDisconnect, messages[7].MessageType)
+	assert.Equal(t, message.TypeClose, messages[6].MessageType)
+	assert.Equal(t, message.TypeExit, messages[7].MessageType)
+	assert.Equal(t, message.TypeDisconnect, messages[8].MessageType)
 }
 
 func getStoredMessages(t *testing.T, dir string, logger log.Logger) ([]message.Message, []error, bool) {