test

gravitational · Jan 22, 2025 · a1b9e2b · a1b9e2b
1 parent 435f8cd
commit a1b9e2b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/integration/kube_integration_test.go b/integration/kube_integration_test.go
@@ -1245,6 +1245,11 @@ func runKubeDisconnectTest(t *testing.T, suite *KubeSuite, tc disconnectTestCase
 		require.NoError(t, err)
 	}()
 
+	require.Eventually(t, func() bool {
+		// wait for the shell prompt
+		return strings.Contains(term.AllOutput(), "#")
+	}, 5*time.Second, 10*time.Millisecond)
+
 	// lets type something followed by "enter" and then hang the session
 	require.NoError(t, enterInput(sessionCtx, term, "echo boring platypus\r\n", ".*boring platypus.*"))
 	time.Sleep(tc.disconnectTimeout)