k8s.io/kube-openapi@v0.0.0-20240826222958-65a50c78dec5/pkg/handler/handler_test.go

package handler

import (
	json "encoding/json"
	"fmt"
	"io"
	"mime"
	"net/http"
	"net/http/httptest"
	"os"
	"reflect"
	"testing"

	"k8s.io/kube-openapi/pkg/cached"
	"k8s.io/kube-openapi/pkg/validation/spec"
)

var returnedSwagger = []byte(`{
  "swagger": "2.0",
  "info": {
   "title": "Kubernetes",
   "version": "v1.11.0"
  }}`)

func TestRegisterOpenAPIVersionedService(t *testing.T) {
	var s spec.Swagger
	err := s.UnmarshalJSON(returnedSwagger)
	if err != nil {
		t.Errorf("Unexpected error in unmarshalling SwaggerJSON: %v", err)
	}

	returnedJSON := normalizeSwaggerOrDie(returnedSwagger)
	var decodedJSON map[string]interface{}
	if err := json.Unmarshal(returnedJSON, &decodedJSON); err != nil {
		t.Fatal(err)
	}
	returnedPb, err := ToProtoBinary(returnedJSON)
	if err != nil {
		t.Errorf("Unexpected error in preparing returnedPb: %v", err)
	}

	mux := http.NewServeMux()
	o := NewOpenAPIService(&s)
	o.RegisterOpenAPIVersionedService("/openapi/v2", mux)
	server := httptest.NewServer(mux)
	defer server.Close()
	client := server.Client()

	tcs := []struct {
		acceptHeader              string
		respStatus                int
		responseContentTypeHeader string
		respBody                  []byte
	}{
		{"", 200, "application/json", returnedJSON},
		{"*/*", 200, "application/json", returnedJSON},
		{"application/*", 200, "application/json", returnedJSON},
		{"application/json", 200, "application/json", returnedJSON},
		{"test/test", 406, "", []byte{}},
		{"application/test", 406, "", []byte{}},
		{"application/test, */*", 200, "application/json", returnedJSON},
		{"application/test, application/json", 200, "application/json", returnedJSON},
		{"application/com.github.proto-openapi.spec.v2.v1.0+protobuf", 200, "application/com.github.proto-openapi.spec.v2.v1.0+protobuf", returnedPb},
		{"application/json, application/com.github.proto-openapi.spec.v2.v1.0+protobuf", 200, "application/json", returnedJSON},
		{"application/com.github.proto-openapi.spec.v2.v1.0+protobuf, application/json", 200, "application/com.github.proto-openapi.spec.v2.v1.0+protobuf", returnedPb},
		{"application/com.github.proto-openapi.spec.v2.v1.0+protobuf; q=0.5, application/json", 200, "application/json", returnedJSON},
		{"application/com.github.proto-openapi.spec.v2@v1.0+protobuf", 200, "application/com.github.proto-openapi.spec.v2.v1.0+protobuf", returnedPb},
		{"application/json, application/com.github.proto-openapi.spec.v2@v1.0+protobuf", 200, "application/json", returnedJSON},
		{"application/com.github.proto-openapi.spec.v2@v1.0+protobuf, application/json", 200, "application/com.github.proto-openapi.spec.v2.v1.0+protobuf", returnedPb},
		{"application/com.github.proto-openapi.spec.v2@v1.0+protobuf; q=0.5, application/json", 200, "application/json", returnedJSON},
	}

	for _, tc := range tcs {
		req, err := http.NewRequest("GET", server.URL+"/openapi/v2", nil)
		if err != nil {
			t.Errorf("Accept: %v: Unexpected error in creating new request: %v", tc.acceptHeader, err)
		}

		req.Header.Add("Accept", tc.acceptHeader)
		resp, err := client.Do(req)
		if err != nil {
			t.Errorf("Accept: %v: Unexpected error in serving HTTP request: %v", tc.acceptHeader, err)
		}

		if resp.StatusCode != tc.respStatus {
			t.Errorf("Accept: %v: Unexpected response status code, want: %v, got: %v", tc.acceptHeader, tc.respStatus, resp.StatusCode)
		}
		if tc.respStatus != 200 {
			continue
		}

		responseContentType := resp.Header.Get("Content-Type")
		if responseContentType != tc.responseContentTypeHeader {
			t.Errorf("Accept: %v: Unexpected content type in response, want: %v, got: %v", tc.acceptHeader, tc.responseContentTypeHeader, responseContentType)
		}

		_, _, err = mime.ParseMediaType(responseContentType)
		if err != nil {
			t.Errorf("Unexpected error in parsing response content type: %v, err: %v", responseContentType, err)
		}

		defer resp.Body.Close()
		body, err := io.ReadAll(resp.Body)
		if err != nil {
			t.Errorf("Accept: %v: Unexpected error in reading response body: %v", tc.acceptHeader, err)
		}
		if !reflect.DeepEqual(body, tc.respBody) {
			t.Errorf("Accept: %v: Response body mismatches, \nwant: %s, \ngot:  %s", tc.acceptHeader, string(tc.respBody), string(body))
		}
	}
}

var updatedSwagger = []byte(`{
  "swagger": "2.0",
  "info": {
   "title": "Kubernetes",
   "version": "v1.12.0"
  }}`)

func getJSONBodyOrDie(server *httptest.Server) []byte {
	return getBodyOrDie(server, "application/json")
}

func getProtoBodyOrDie(server *httptest.Server) []byte {
	return getBodyOrDie(server, "application/com.github.proto-openapi.spec.v2.v1.0+protobuf")
}

func getBodyOrDie(server *httptest.Server, acceptHeader string) []byte {
	req, err := http.NewRequest("GET", server.URL+"/openapi/v2", nil)
	if err != nil {
		panic(fmt.Errorf("Unexpected error in creating new request: %v", err))
	}

	req.Header.Add("Accept", acceptHeader)
	resp, err := server.Client().Do(req)
	if err != nil {
		panic(fmt.Errorf("Unexpected error in serving HTTP request: %v", err))
	}

	defer resp.Body.Close()
	body, err := io.ReadAll(resp.Body)
	if err != nil {
		panic(fmt.Errorf("Unexpected error in reading response body: %v", err))
	}
	return body
}

func normalizeSwaggerOrDie(j []byte) []byte {
	var s spec.Swagger
	err := s.UnmarshalJSON(j)
	if err != nil {
		panic(fmt.Errorf("Unexpected error in unmarshalling SwaggerJSON: %v", err))
	}
	rj, err := json.Marshal(s)
	if err != nil {
		panic(fmt.Errorf("Unexpected error in preparing returnedJSON: %v", err))
	}
	return rj
}

func TestUpdateSpecLazy(t *testing.T) {
	returnedJSON := normalizeSwaggerOrDie(returnedSwagger)
	var s spec.Swagger
	err := s.UnmarshalJSON(returnedJSON)
	if err != nil {
		t.Errorf("Unexpected error in unmarshalling SwaggerJSON: %v", err)
	}

	mux := http.NewServeMux()
	o := NewOpenAPIService(&s)
	o.RegisterOpenAPIVersionedService("/openapi/v2", mux)
	server := httptest.NewServer(mux)
	defer server.Close()

	body := string(getJSONBodyOrDie(server))
	if body != string(returnedJSON) {
		t.Errorf("Unexpected swagger received, got %q, expected %q", body, string(returnedSwagger))
	}

	o.UpdateSpecLazy(cached.Func(func() (*spec.Swagger, string, error) {
		var s spec.Swagger
		err := s.UnmarshalJSON(updatedSwagger)
		if err != nil {
			t.Errorf("Unexpected error in unmarshalling SwaggerJSON: %v", err)
		}
		return &s, "SOMEHASH", nil
	}))

	updatedJSON := normalizeSwaggerOrDie(updatedSwagger)
	body = string(getJSONBodyOrDie(server))

	if body != string(updatedJSON) {
		t.Errorf("Unexpected swagger received, got %q, expected %q", body, string(updatedJSON))
	}
}

func TestToProtoBinary(t *testing.T) {
	bs, err := os.ReadFile("../../test/integration/testdata/aggregator/openapi.json")
	if err != nil {
		t.Fatal(err)
	}
	if _, err := ToProtoBinary(bs); err != nil {
		t.Fatal()
	}
	// TODO: add some kind of roundtrip test here
}

func TestConcurrentReadStaleCache(t *testing.T) {
	// Number of requests sent in parallel
	concurrency := 5

	var s spec.Swagger
	err := s.UnmarshalJSON(returnedSwagger)
	if err != nil {
		t.Errorf("Unexpected error in unmarshalling SwaggerJSON: %v", err)
	}

	mux := http.NewServeMux()
	o := NewOpenAPIService(&s)
	o.RegisterOpenAPIVersionedService("/openapi/v2", mux)
	server := httptest.NewServer(mux)
	defer server.Close()

	returnedJSON := normalizeSwaggerOrDie(returnedSwagger)
	returnedPb, err := ToProtoBinary(returnedJSON)
	if err != nil {
		t.Errorf("Unexpected error in preparing returnedPb: %v", err)
	}

	jsonResultsChan := make(chan []byte)
	protoResultsChan := make(chan []byte)
	updateSpecChan := make(chan struct{})
	for i := 0; i < concurrency; i++ {
		go func() {
			sc := s
			o.UpdateSpec(&sc)
			updateSpecChan <- struct{}{}
		}()
		go func() { jsonResultsChan <- getJSONBodyOrDie(server) }()
		go func() { protoResultsChan <- getProtoBodyOrDie(server) }()
	}
	for i := 0; i < concurrency; i++ {
		r := <-jsonResultsChan
		if !reflect.DeepEqual(r, returnedJSON) {
			t.Errorf("Returned and expected JSON do not match: got %v, want %v", string(r), string(returnedJSON))
		}
	}
	for i := 0; i < concurrency; i++ {
		r := <-protoResultsChan
		if !reflect.DeepEqual(r, returnedPb) {
			t.Errorf("Returned and expected pb do not match: got %v, want %v", r, returnedPb)
		}
	}
	for i := 0; i < concurrency; i++ {
		<-updateSpecChan
	}
}