Function Test_hijacking

x/ipc/hijack_test.go:36–81

(t *testing.T)

)

func Test_hijacking(t *testing.T) {
	socketPath := filepath.Join(os.TempDir(), "hijack.sock")
	l := newListener(t, socketPath)
	t.Cleanup(func() { l.Close() })

	httpMux := http.NewServeMux()
	ch := make(chan io.ReadWriter)
	wait := make(chan struct{})
	once := sync.OnceFunc(func() { close(wait) })
	defer once()
	httpMux.Handle(NewHijackAcceptor(testhelper.TestLogger(t), func(_ context.Context, closer io.ReadWriteCloser) {
		ch <- closer
		<-wait
	}))
	httpMux.HandleFunc("/health", func(w http.ResponseWriter, _ *http.Request) {
		w.WriteHeader(http.StatusOK)
		_, _ = w.Write([]byte("ok"))
	})
	serverErr := make(chan error, 1)
	server := &http.Server{Handler: httpMux}
	go func() {
		serverErr <- server.Serve(l)
	}()
	connClient, err := net.Dial("unix", socketPath)
	require.NoError(t, err)
	t.Cleanup(func() { connClient.Close() })
	connHijacked, err := Hijackify(connClient, hijackTimeout)
	require.NoError(t, err)
	connServer, err := testhelper.WaitForWithTimeoutV(ch)
	require.NoError(t, err)

	assert.NoError(t, writeLine(connHijacked, "ping"))
	assert.Equal(t, "ping", readLine(connServer))
	assert.NoError(t, writeLine(connServer, "pong"))
	assert.Equal(t, "pong", readLine(connHijacked))
	assert.NoError(t, connHijacked.Close())
	once()

	// The server should still be up and also be available for normal/non-hijack stuff
	health, err := requestHealthCheck(socketPath)
	assert.NoError(t, err)
	assert.Equal(t, "ok", health)

	assert.NoError(t, server.Close())
	assert.ErrorIs(t, testhelper.WaitForErrorWithTimeout(serverErr), http.ErrServerClosed)
}

func TestHijackify_hijackRequest_timeout(t *testing.T) {
	socket := testhelper.RandomShortSocketName()