Function TestIsHandledByHttpgrpcServer

httpgrpc/server/server_test.go:378–405

(t *testing.T)

}

func TestIsHandledByHttpgrpcServer(t *testing.T) {
	t.Run("false by default", func(t *testing.T) {
		require.False(t, IsHandledByHttpgrpcServer(context.Background()))
	})

	const testHeader = "X-HandledByHttpgrpcServer"

	// Handler will return value returned by IsHandledByHttpgrpcServer in test header.
	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set(testHeader, strconv.FormatBool(IsHandledByHttpgrpcServer(r.Context())))
		w.WriteHeader(200)
	})

	t.Run("handler runs outside of httpgrpc.Server", func(t *testing.T) {
		rec := httptest.NewRecorder()
		handler(rec, &http.Request{})
		require.Equal(t, "false", rec.Header().Get(testHeader))
	})

	t.Run("handler runs from httpgrpc.Server", func(t *testing.T) {
		s := NewServer(handler)
		resp, err := s.Handle(context.Background(), &httpgrpc.HTTPRequest{Method: "GET", Url: "/test"})
		require.NoError(t, err)
		require.NotNil(t, resp)

		require.Equal(t, []*httpgrpc.Header{{Key: http.CanonicalHeaderKey(testHeader), Values: []string{"true"}}}, resp.Headers)
	})
}