Function testSessionOutput

agent/agent_test.go:3956–3992

(t *testing.T, session *ssh.Session, expected, unexpected []string, expectedRe *regexp.Regexp)

}

func testSessionOutput(t *testing.T, session *ssh.Session, expected, unexpected []string, expectedRe *regexp.Regexp) {
	t.Helper()

	err := session.RequestPty("xterm", 128, 128, ssh.TerminalModes{})
	require.NoError(t, err)

	ptty := ptytest.New(t)
	var stdout bytes.Buffer
	session.Stdout = &stdout
	session.Stderr = ptty.Output()
	session.Stdin = ptty.Input()
	err = session.Shell()
	require.NoError(t, err)

	ptty.WriteLine("exit 0")

	waitErr := make(chan error, 1)
	go func() {
		waitErr <- session.Wait()
	}()
	select {
	case err = <-waitErr:
		require.NoError(t, err)
	case <-time.After(testutil.WaitLong):
		require.Fail(t, "timed out waiting for session to exit")
	}

	for _, unexpected := range unexpected {
		require.NotContains(t, stdout.String(), unexpected, "should not show output")
	}
	for _, expect := range expected {
		require.Contains(t, stdout.String(), expect, "should show output")
	}
	if expectedRe != nil {
		require.Regexp(t, expectedRe, stdout.String())
	}
}

func TestAgent_Metrics_SSH(t *testing.T) {
	t.Parallel()