Fix connect message race

ContainerSSH · Aug 15, 2024 · 5db135d · 5db135d
1 parent c557aab
commit 5db135d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/auditlogintegration/integration_test.go b/internal/auditlogintegration/integration_test.go
@@ -99,6 +99,10 @@ func TestConnectMessages(t *testing.T) {
 	if err := session.Close(); err != nil {
 		t.Fatal(err)
 	}
+	// Yield enough for the session to close, if we close the
+	// connection too soon then some of the closing steps might be
+	// bypassed making the check fail
+	time.Sleep(10 * time.Millisecond)
 	if err := connection.Close(); err != nil {
 		t.Fatal(err)
 	}