Function captureOutput

packaging/tools/agent-events/server_test.go:23–61

Helper function to capture stdout/stderr during a test run

(t *testing.T, f func())

// Helper function to capture stdout/stderr during a test run
func captureOutput(t *testing.T, f func()) (stdout, stderr string) {
	t.Helper() // Marks this as a helper function for testing framework

	originalStdout := os.Stdout
	originalStderr := os.Stderr
	oldLogger := slog.Default()
	rOut, wOut, _ := os.Pipe()
	rErr, wErr, _ := os.Pipe()
	os.Stdout = wOut
	os.Stderr = wErr

	t.Cleanup(func() {
		os.Stdout = originalStdout
		os.Stderr = originalStderr
		slog.SetDefault(oldLogger)
	})

	outCh := make(chan string)
	errCh := make(chan string)

	go func() {
		var buf bytes.Buffer
		_, _ = io.Copy(&buf, rOut)
		outCh <- buf.String()
	}()
	go func() {
		var buf bytes.Buffer
		_, _ = io.Copy(&buf, rErr)
		errCh <- buf.String()
	}()

	f() // Execute the function

	_ = wOut.Close()
	_ = wErr.Close()
	stdout = <-outCh
	stderr = <-errCh
	return stdout, stderr
}

// --- Test Suite ---