github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-08-07 03:03:59 +00:00
Code Issues Projects Releases Wiki Activity

add test for apiservices.apiregistration.k8s.io discoverability in openapi/v3

Browse Source
This commit is contained in:
Filip Křepinský 2023-06-26 18:38:07 +02:00
parent 9e3b5b6583
commit cad338890d
1 changed files with 70 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
staging/src/k8s.io/kube-aggregator/pkg/controllers/openapiv3/aggregator/aggregator_test.go
View File

@ -23,15 +23,20 @@ import (
"strings"
"testing"
"github.com/emicklei/go-restful/v3"
"github.com/stretchr/testify/assert"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apiserver/pkg/endpoints/metrics"
openapinamer "k8s.io/apiserver/pkg/endpoints/openapi"
genericapiserver "k8s.io/apiserver/pkg/server"
"k8s.io/apiserver/pkg/server/mux"
"k8s.io/component-base/metrics/legacyregistry"
"k8s.io/component-base/metrics/testutil"
v1 "k8s.io/kube-aggregator/pkg/apis/apiregistration/v1"
openapicommon "k8s.io/kube-openapi/pkg/common"
"k8s.io/kube-openapi/pkg/handler3"
kubeopenapispec "k8s.io/kube-openapi/pkg/validation/spec"
)
type testV3APIService struct {
@ -170,6 +175,54 @@ func TestV3APIService(t *testing.T) {
assert.ElementsMatch(t, []string{openAPIV2Converter, apiService.Name}, apiServiceNames)
}
func TestV3RootAPIService(t *testing.T) {
ws := new(restful.WebService)
{
ws.Path("/apis/apiregistration.k8s.io/v1")
ws.Doc("API at/apis/apiregistration.k8s.io/v1 ")
ws.Consumes("*/*")
ws.Produces("application/json")
ws.ApiVersion("apiregistration.k8s.io/v1")
routeBuilder := ws.GET("apiservices").
To(func(request *restful.Request, response *restful.Response) {}).
Doc("list or watch objects of kind APIService").
Operation("listAPIService").
Produces("application/json").
Returns(http.StatusOK, "OK", v1.APIService{}).
Writes(v1.APIService{})
ws.Route(routeBuilder)
}
openapiConfig := genericapiserver.DefaultOpenAPIV3Config(getTestAPIServiceOpenAPIDefinitions, openapinamer.NewDefinitionNamer(runtime.NewScheme()))
downloader := Downloader{}
goRestfulContainer := restful.NewContainer()
goRestfulContainer.Add(ws)
pathHandler := mux.NewPathRecorderMux("aggregator_test")
var serveHandler http.Handler = pathHandler
specProxier, err := BuildAndRegisterAggregator(downloader, genericapiserver.NewEmptyDelegate(), goRestfulContainer, openapiConfig, pathHandler)
if err != nil {
t.Error(err)
}
expectedSpecJSON := []byte(`{"openapi":"3.0.0","info":{"title":"Generic API Server"},"paths":{"/apis/apiregistration.k8s.io/v1/apiservices":{"get":{"tags":["apiregistration_v1"],"description":"list or watch objects of kind APIService","operationId":"listApiregistrationV1APIService","responses":{"200":{"description":"OK","content":{"application/json":{"schema":{"$ref":"#/components/schemas/io.k8s.kube-aggregator.pkg.apis.apiregistration.v1.APIService"}}}}}}}},"components":{"schemas":{"io.k8s.kube-aggregator.pkg.apis.apiregistration.v1.APIService":{"description":"APIService represents a server for a particular GroupVersion. Name must be \"version.group\".","type":"object"}}}}`)
data := sendReq(t, serveHandler, "/openapi/v3")
groupVersionList := handler3.OpenAPIV3Discovery{}
if err := json.Unmarshal(data, &groupVersionList); err != nil {
t.Fatal(err)
}
path, ok := groupVersionList.Paths["apis/apiregistration.k8s.io/v1"]
if !ok {
t.Error("Expected apiregistration.k8s.io/v1 to be in group version list")
}
gotSpecJSON := sendReq(t, serveHandler, path.ServerRelativeURL)
if bytes.Compare(gotSpecJSON, expectedSpecJSON) != 0 {
t.Errorf("Spec mismatch, expected %s, got %s", expectedSpecJSON, gotSpecJSON)
}
apiServiceNames := specProxier.GetAPIServiceNames()
assert.ElementsMatch(t, []string{"k8s_internal_local_kube_aggregator_types", openAPIV2Converter}, apiServiceNames)
}
func TestOpenAPIRequestMetrics(t *testing.T) {
metrics.Register()
metrics.Reset()
@ -239,3 +292,20 @@ func sendReq(t *testing.T, handler http.Handler, path string) []byte {
handler.ServeHTTP(writer, req)
return writer.data
}
func getTestAPIServiceOpenAPIDefinitions(_ openapicommon.ReferenceCallback) map[string]openapicommon.OpenAPIDefinition {
return map[string]openapicommon.OpenAPIDefinition{
"k8s.io/kube-aggregator/pkg/apis/apiregistration/v1.APIService": buildTestAPIServiceOpenAPIDefinition(),
}
}
func buildTestAPIServiceOpenAPIDefinition() openapicommon.OpenAPIDefinition {
return openapicommon.OpenAPIDefinition{
Schema: kubeopenapispec.Schema{
SchemaProps: kubeopenapispec.SchemaProps{
Description: "APIService represents a server for a particular GroupVersion. Name must be \"version.group\".",
Type: []string{"object"},
},
},
}
}