Function Example_queryMixedRootDBPlusOpenAPI

tests/openapi_toplevel_test.go:142–244

Example_queryMixedRootDBPlusOpenAPI exercises a query with one real-table root (users) and one top-level OpenAPI remote root (audit_logs) in the same GraphQL document. The all-remote shortcut must not fire; psql skips the remote root; gstate injects a marker for it; execRemoteJoin then resolves the

()

// must not fire; psql skips the remote root; gstate injects a marker
// for it; execRemoteJoin then resolves the upstream call.
func Example_queryMixedRootDBPlusOpenAPI() {
	mux := http.NewServeMux()
	mux.HandleFunc("/audit-logs", func(w http.ResponseWriter, r *http.Request) {
		actor := r.URL.Query().Get("actorId")
		w.Header().Set("Content-Type", "application/json")
		fmt.Fprintf(w, `{"data":[{"id":"a1","action":"action by %s"}]}`, actor) //nolint:errcheck
	})
	listener, err := net.Listen("tcp", "localhost:0")
	if err != nil {
		panic(err)
	}
	port := listener.Addr().(*net.TCPAddr).Port
	server := &http.Server{Handler: mux}
	go func() { log.Fatal(server.Serve(listener)) }() //nolint:gosec
	for i := 0; i < 100; i++ {
		resp, err := http.Get(fmt.Sprintf("http://localhost:%d/audit-logs?actorId=ping", port))
		if err == nil {
			resp.Body.Close() //nolint:errcheck
			break
		}
		time.Sleep(50 * time.Millisecond)
	}

	specsDir, err := os.MkdirTemp("", "graphjin-openapi-mixed-*")
	if err != nil {
		panic(err)
	}
	defer os.RemoveAll(specsDir) //nolint:errcheck

	specYAML := fmt.Sprintf(`
openapi: 3.0.0
info: { title: Audit, version: '1.0' }
servers:
  - url: http://localhost:%d
paths:
  /audit-logs:
    get:
      operationId: listAuditLogs
      parameters:
        - { name: actorId, in: query, required: true, schema: { type: string } }
      responses:
        '200':
          description: ok
          content:
            application/json:
              schema:
                type: object
                properties:
                  data:
                    type: array
                    items:
                      type: object
                      properties:
                        id: { type: string }
                        action: { type: string }
`, port)
	if err := os.WriteFile(filepath.Join(specsDir, "audit.yaml"), []byte(specYAML), 0o644); err != nil {
		panic(err)