Function TestReapInterrupt

agent/reaper/reaper_test.go:171–214

TestReapInterrupt verifies that ForkReap forwards caught signals to the child process. The test sends SIGINT to its own process and checks that the child receives it. Running in a subprocess ensures SIGINT cannot kill the parent test binary.

(t *testing.T)

// and checks that the child receives it. Running in a subprocess
// ensures SIGINT cannot kill the parent test binary.
func TestReapInterrupt(t *testing.T) {
	t.Parallel()
	if testutil.InCI() {
		t.Skip("Detected CI, skipping reaper tests")
	}
	if !runSubprocess(t) {
		return
	}

	errC := make(chan error, 1)
	pids := make(reap.PidCh, 1)

	// Use signals to notify when the child process is ready for the
	//  next step of our test.
	usrSig := make(chan os.Signal, 1)
	signal.Notify(usrSig, syscall.SIGUSR1, syscall.SIGUSR2)
	defer signal.Stop(usrSig)

	go func() {
		opts := append([]reaper.Option{
			reaper.WithPIDCallback(pids),
			reaper.WithCatchSignals(os.Interrupt),
			// Signal propagation does not extend to children of children, so
			// we create a little bash script to ensure sleep is interrupted.
			reaper.WithExecArgs("/bin/sh", "-c", fmt.Sprintf(
				"pid=0; trap 'kill -USR2 %d; kill -TERM $pid' INT; sleep 10 &\npid=$!; kill -USR1 %d; wait",
				os.Getpid(), os.Getpid(),
			)),
		}, withDone(t)...)
		exitCode, err := reaper.ForkReap(opts...)
		// The child exits with 128 + SIGTERM (15) = 143, but the trap catches
		// SIGINT and sends SIGTERM to the sleep process, so exit code varies.
		_ = exitCode
		errC <- err
	}()

	require.Equal(t, syscall.SIGUSR1, <-usrSig)

	err := syscall.Kill(os.Getpid(), syscall.SIGINT)
	require.NoError(t, err)

	require.Equal(t, syscall.SIGUSR2, <-usrSig)
	require.NoError(t, <-errC)
}