Function TestIgnoredSignals

cli/command/container/signals_unix_test.go:17–58

(t *testing.T)

)

func TestIgnoredSignals(t *testing.T) {
	ignoredSignals := []syscall.Signal{unix.SIGPIPE, unix.SIGCHLD, unix.SIGURG}

	for _, s := range ignoredSignals {
		t.Run(unix.SignalName(s), func(t *testing.T) {
			ctx, cancel := context.WithCancel(context.Background())
			defer cancel()

			var called bool
			apiClient := &fakeClient{containerKillFunc: func(ctx context.Context, container string, options client.ContainerKillOptions) (client.ContainerKillResult, error) {
				called = true
				return client.ContainerKillResult{}, nil
			}}

			sigc := make(chan os.Signal)
			defer close(sigc)

			done := make(chan struct{})
			go func() {
				ForwardAllSignals(ctx, apiClient, t.Name(), sigc)
				close(done)
			}()

			timer := time.NewTimer(30 * time.Second)
			defer timer.Stop()

			select {
			case <-timer.C:
				t.Fatal("timeout waiting to send signal")
			case sigc <- s:
			case <-done:
			}

			// cancel the context so ForwardAllSignals will exit after it has processed the signal we sent.
			// This is how we know the signal was actually processed and are not introducing a flakey test.
			cancel()
			<-done

			assert.Assert(t, !called, "kill was called")
		})
	}
}