mirror of
https://github.com/rancher/steve.git
synced 2025-09-03 08:25:13 +00:00
Adding tests for schemaDefinitions
This commit is contained in:
Michael Bolot
325
pkg/schema/definitions/handler_test.go
Normal file
325
pkg/schema/definitions/handler_test.go
Normal file
@@ -0,0 +1,325 @@
|
|||||||
|
package definitions
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"testing"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
openapi_v2 "github.com/google/gnostic-models/openapiv2"
|
||||||
|
"github.com/rancher/apiserver/pkg/apierror"
|
||||||
|
"github.com/rancher/apiserver/pkg/types"
|
||||||
|
wschemas "github.com/rancher/wrangler/v2/pkg/schemas"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
||||||
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
||||||
|
"k8s.io/apimachinery/pkg/version"
|
||||||
|
"k8s.io/client-go/discovery"
|
||||||
|
"k8s.io/client-go/openapi"
|
||||||
|
restclient "k8s.io/client-go/rest"
|
||||||
|
)
|
||||||
|
|
||||||
|
var globalRoleObject = types.APIObject{
|
||||||
|
ID: "management.cattle.io.globalrole",
|
||||||
|
Type: "schemaDefinition",
|
||||||
|
Object: schemaDefinition{
|
||||||
|
DefinitionType: "io.cattle.management.v2.GlobalRole",
|
||||||
|
Definitions: map[string]definition{
|
||||||
|
"io.cattle.management.v2.GlobalRole": {
|
||||||
|
ResourceFields: map[string]definitionField{
|
||||||
|
"apiVersion": {
|
||||||
|
Type: "string",
|
||||||
|
Description: "The APIVersion of this resource",
|
||||||
|
},
|
||||||
|
"kind": {
|
||||||
|
Type: "string",
|
||||||
|
Description: "The kind",
|
||||||
|
},
|
||||||
|
"metadata": {
|
||||||
|
Type: "io.k8s.apimachinery.pkg.apis.meta.v1.ObjectMeta",
|
||||||
|
Description: "The metadata",
|
||||||
|
},
|
||||||
|
"spec": {
|
||||||
|
Type: "io.cattle.management.v2.GlobalRole.spec", Description: "The spec for the project",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Type: "io.cattle.management.v2.GlobalRole",
|
||||||
|
Description: "A Global Role V2 provides Global Permissions in Rancher",
|
||||||
|
},
|
||||||
|
"io.cattle.management.v2.GlobalRole.spec": {
|
||||||
|
ResourceFields: map[string]definitionField{
|
||||||
|
"clusterName": {
|
||||||
|
Type: "string",
|
||||||
|
Description: "The name of the cluster",
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
"displayName": {
|
||||||
|
Type: "string",
|
||||||
|
Description: "The UI readable name",
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
"newField": {
|
||||||
|
Type: "string",
|
||||||
|
Description: "A new field not present in v1",
|
||||||
|
},
|
||||||
|
"notRequired": {
|
||||||
|
Type: "boolean",
|
||||||
|
Description: "Some field that isn't required",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Type: "io.cattle.management.v2.GlobalRole.spec",
|
||||||
|
Description: "The spec for the project",
|
||||||
|
},
|
||||||
|
"io.k8s.apimachinery.pkg.apis.meta.v1.ObjectMeta": {
|
||||||
|
ResourceFields: map[string]definitionField{
|
||||||
|
"annotations": {
|
||||||
|
Type: "map",
|
||||||
|
SubType: "string",
|
||||||
|
Description: "annotations of the resource",
|
||||||
|
},
|
||||||
|
"name": {
|
||||||
|
Type: "string",
|
||||||
|
SubType: "",
|
||||||
|
Description: "name of the resource",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Type: "io.k8s.apimachinery.pkg.apis.meta.v1.ObjectMeta",
|
||||||
|
Description: "Object Metadata",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestByID(t *testing.T) {
|
||||||
|
schemas := types.EmptyAPISchemas()
|
||||||
|
addBaseSchema := func(names ...string) {
|
||||||
|
for _, name := range names {
|
||||||
|
schemas.MustAddSchema(types.APISchema{
|
||||||
|
Schema: &wschemas.Schema{
|
||||||
|
ID: name,
|
||||||
|
CollectionMethods: []string{"get"},
|
||||||
|
ResourceMethods: []string{"get"},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
intPtr := func(input int) *int {
|
||||||
|
return &input
|
||||||
|
}
|
||||||
|
|
||||||
|
addBaseSchema("management.cattle.io.globalrole", "management.cattle.io.missingfrommodel")
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
schemaName string
|
||||||
|
needsRefresh bool
|
||||||
|
openapiError error
|
||||||
|
serverGroupsResourcesErr error
|
||||||
|
useBadOpenApiDoc bool
|
||||||
|
unparseableGV bool
|
||||||
|
wantObject *types.APIObject
|
||||||
|
wantError bool
|
||||||
|
wantErrorCode *int
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "global role definition",
|
||||||
|
schemaName: "management.cattle.io.globalrole",
|
||||||
|
needsRefresh: true,
|
||||||
|
wantObject: &globalRoleObject,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "missing definition",
|
||||||
|
schemaName: "management.cattle.io.cluster",
|
||||||
|
needsRefresh: true,
|
||||||
|
wantError: true,
|
||||||
|
wantErrorCode: intPtr(404),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "not refreshed",
|
||||||
|
schemaName: "management.cattle.io.globalrole",
|
||||||
|
needsRefresh: false,
|
||||||
|
wantError: true,
|
||||||
|
wantErrorCode: intPtr(503),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "missing from model",
|
||||||
|
schemaName: "management.cattle.io.missingfrommodel",
|
||||||
|
needsRefresh: true,
|
||||||
|
wantError: true,
|
||||||
|
wantErrorCode: intPtr(503),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "refresh error - openapi doc unavailable",
|
||||||
|
schemaName: "management.cattle.io.globalrole",
|
||||||
|
needsRefresh: true,
|
||||||
|
openapiError: fmt.Errorf("server unavailable"),
|
||||||
|
wantError: true,
|
||||||
|
wantErrorCode: intPtr(500),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "refresh error - unable to parse openapi doc",
|
||||||
|
schemaName: "management.cattle.io.globalrole",
|
||||||
|
needsRefresh: true,
|
||||||
|
useBadOpenApiDoc: true,
|
||||||
|
wantError: true,
|
||||||
|
wantErrorCode: intPtr(500),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "refresh error - unable to retrieve groups and resources",
|
||||||
|
schemaName: "management.cattle.io.globalrole",
|
||||||
|
needsRefresh: true,
|
||||||
|
serverGroupsResourcesErr: fmt.Errorf("server not available"),
|
||||||
|
wantError: true,
|
||||||
|
wantErrorCode: intPtr(500),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "refresh error - unable to retrieve all groups and resources",
|
||||||
|
schemaName: "management.cattle.io.globalrole",
|
||||||
|
needsRefresh: true,
|
||||||
|
serverGroupsResourcesErr: &discovery.ErrGroupDiscoveryFailed{
|
||||||
|
Groups: map[schema.GroupVersion]error{
|
||||||
|
{
|
||||||
|
Group: "other.cattle.io",
|
||||||
|
Version: "v1",
|
||||||
|
}: fmt.Errorf("some group error"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wantError: true,
|
||||||
|
wantErrorCode: intPtr(500),
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "refresh error - unparesable gv",
|
||||||
|
schemaName: "management.cattle.io.globalrole",
|
||||||
|
needsRefresh: true,
|
||||||
|
unparseableGV: true,
|
||||||
|
wantError: true,
|
||||||
|
wantErrorCode: intPtr(500),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, test := range tests {
|
||||||
|
test := test
|
||||||
|
t.Run(test.name, func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
client, err := buildDefaultDiscovery()
|
||||||
|
client.DocumentErr = test.openapiError
|
||||||
|
client.GroupResourcesErr = test.serverGroupsResourcesErr
|
||||||
|
if test.useBadOpenApiDoc {
|
||||||
|
schema := client.Document.Definitions.AdditionalProperties[0]
|
||||||
|
schema.Value.Type = &openapi_v2.TypeItem{
|
||||||
|
Value: []string{"multiple", "entries"},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if test.unparseableGV {
|
||||||
|
client.Resources = append(client.Resources, &metav1.APIResourceList{
|
||||||
|
GroupVersion: "not/parse/able",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
require.Nil(t, err)
|
||||||
|
handler := schemaDefinitionHandler{
|
||||||
|
client: client,
|
||||||
|
}
|
||||||
|
if !test.needsRefresh {
|
||||||
|
handler.lastRefresh = time.Now()
|
||||||
|
handler.refreshStale = time.Minute * 1
|
||||||
|
}
|
||||||
|
request := types.APIRequest{
|
||||||
|
Schemas: schemas,
|
||||||
|
Name: test.schemaName,
|
||||||
|
}
|
||||||
|
response, err := handler.byIDHandler(&request)
|
||||||
|
if test.wantError {
|
||||||
|
require.Error(t, err)
|
||||||
|
if test.wantErrorCode != nil {
|
||||||
|
require.True(t, apierror.IsAPIError(err))
|
||||||
|
apiErr, _ := err.(*apierror.APIError)
|
||||||
|
require.Equal(t, *test.wantErrorCode, apiErr.Code.Status)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
require.NoError(t, err)
|
||||||
|
require.Equal(t, *test.wantObject, response)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func buildDefaultDiscovery() (*fakeDiscovery, error) {
|
||||||
|
document, err := openapi_v2.ParseDocument([]byte(openapi_raw))
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("unable to parse openapi document %w", err)
|
||||||
|
}
|
||||||
|
groups := []*metav1.APIGroup{
|
||||||
|
{
|
||||||
|
Name: "management.cattle.io",
|
||||||
|
PreferredVersion: metav1.GroupVersionForDiscovery{
|
||||||
|
Version: "v2",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
resources := []*metav1.APIResourceList{
|
||||||
|
{
|
||||||
|
GroupVersion: schema.GroupVersion{
|
||||||
|
Group: "management.cattle.io",
|
||||||
|
Version: "v2",
|
||||||
|
}.String(),
|
||||||
|
APIResources: []metav1.APIResource{
|
||||||
|
{
|
||||||
|
Group: "management.cattle.io",
|
||||||
|
Kind: "GlobalRole",
|
||||||
|
Version: "v2",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
GroupVersion: schema.GroupVersion{
|
||||||
|
Group: "management.cattle.io",
|
||||||
|
Version: "v1",
|
||||||
|
}.String(),
|
||||||
|
APIResources: []metav1.APIResource{
|
||||||
|
{
|
||||||
|
Group: "management.cattle.io",
|
||||||
|
Kind: "GlobalRole",
|
||||||
|
Version: "v2",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
nil,
|
||||||
|
}
|
||||||
|
return &fakeDiscovery{
|
||||||
|
Groups: groups,
|
||||||
|
Resources: resources,
|
||||||
|
Document: document,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type fakeDiscovery struct {
|
||||||
|
Groups []*metav1.APIGroup
|
||||||
|
Resources []*metav1.APIResourceList
|
||||||
|
Document *openapi_v2.Document
|
||||||
|
GroupResourcesErr error
|
||||||
|
DocumentErr error
|
||||||
|
}
|
||||||
|
|
||||||
|
// ServerGroupsAndResources is the only method we actually need for the test - just returns what is on the struct
|
||||||
|
func (f *fakeDiscovery) ServerGroupsAndResources() ([]*metav1.APIGroup, []*metav1.APIResourceList, error) {
|
||||||
|
return f.Groups, f.Resources, f.GroupResourcesErr
|
||||||
|
}
|
||||||
|
|
||||||
|
// The rest of these methods are just here to conform to discovery.DiscoveryInterface
|
||||||
|
func (f *fakeDiscovery) RESTClient() restclient.Interface { return nil }
|
||||||
|
func (f *fakeDiscovery) ServerGroups() (*metav1.APIGroupList, error) { return nil, nil }
|
||||||
|
func (f *fakeDiscovery) ServerResourcesForGroupVersion(groupVersion string) (*metav1.APIResourceList, error) {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
func (f *fakeDiscovery) ServerPreferredResources() ([]*metav1.APIResourceList, error) {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
func (f *fakeDiscovery) ServerPreferredNamespacedResources() ([]*metav1.APIResourceList, error) {
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
func (f *fakeDiscovery) ServerVersion() (*version.Info, error) { return nil, nil }
|
||||||
|
func (f *fakeDiscovery) OpenAPISchema() (*openapi_v2.Document, error) {
|
||||||
|
return f.Document, f.DocumentErr
|
||||||
|
}
|
||||||
|
func (f *fakeDiscovery) OpenAPIV3() openapi.Client { return nil }
|
||||||
|
func (f *fakeDiscovery) WithLegacy() discovery.DiscoveryInterface { return f }
|
||||||
99
pkg/schema/definitions/openapi_test.go
Normal file
99
pkg/schema/definitions/openapi_test.go
Normal file
@@ -0,0 +1,99 @@
|
|||||||
|
package definitions
|
||||||
|
|
||||||
|
const openapi_raw = `
|
||||||
|
swagger: "2.0"
|
||||||
|
info:
|
||||||
|
title: "Test openapi spec"
|
||||||
|
version: "v1.0.0"
|
||||||
|
paths:
|
||||||
|
/apis/management.cattle.io/v3/globalroles:
|
||||||
|
get:
|
||||||
|
description: "get a global role"
|
||||||
|
responses:
|
||||||
|
200:
|
||||||
|
description: "OK"
|
||||||
|
definitions:
|
||||||
|
io.cattle.management.v1.GlobalRole:
|
||||||
|
description: "A Global Role V1 provides Global Permissions in Rancher"
|
||||||
|
type: "object"
|
||||||
|
properties:
|
||||||
|
apiVersion:
|
||||||
|
description: "The APIVersion of this resource"
|
||||||
|
type: "string"
|
||||||
|
kind:
|
||||||
|
description: "The kind"
|
||||||
|
type: "string"
|
||||||
|
metadata:
|
||||||
|
description: "The metadata"
|
||||||
|
$ref: "#/definitions/io.k8s.apimachinery.pkg.apis.meta.v1.ObjectMeta"
|
||||||
|
spec:
|
||||||
|
description: "The spec for the project"
|
||||||
|
type: "object"
|
||||||
|
required:
|
||||||
|
- "clusterName"
|
||||||
|
- "displayName"
|
||||||
|
properties:
|
||||||
|
clusterName:
|
||||||
|
description: "The name of the cluster"
|
||||||
|
type: "string"
|
||||||
|
displayName:
|
||||||
|
description: "The UI readable name"
|
||||||
|
type: "string"
|
||||||
|
notRequired:
|
||||||
|
description: "Some field that isn't required"
|
||||||
|
type: "boolean"
|
||||||
|
x-kubernetes-group-version-kind:
|
||||||
|
- group: "management.cattle.io"
|
||||||
|
version: "v1"
|
||||||
|
kind: "GlobalRole"
|
||||||
|
io.cattle.management.v2.GlobalRole:
|
||||||
|
description: "A Global Role V2 provides Global Permissions in Rancher"
|
||||||
|
type: "object"
|
||||||
|
properties:
|
||||||
|
apiVersion:
|
||||||
|
description: "The APIVersion of this resource"
|
||||||
|
type: "string"
|
||||||
|
kind:
|
||||||
|
description: "The kind"
|
||||||
|
type: "string"
|
||||||
|
metadata:
|
||||||
|
description: "The metadata"
|
||||||
|
$ref: "#/definitions/io.k8s.apimachinery.pkg.apis.meta.v1.ObjectMeta"
|
||||||
|
spec:
|
||||||
|
description: "The spec for the project"
|
||||||
|
type: "object"
|
||||||
|
required:
|
||||||
|
- "clusterName"
|
||||||
|
- "displayName"
|
||||||
|
properties:
|
||||||
|
clusterName:
|
||||||
|
description: "The name of the cluster"
|
||||||
|
type: "string"
|
||||||
|
displayName:
|
||||||
|
description: "The UI readable name"
|
||||||
|
type: "string"
|
||||||
|
notRequired:
|
||||||
|
description: "Some field that isn't required"
|
||||||
|
type: "boolean"
|
||||||
|
newField:
|
||||||
|
description: "A new field not present in v1"
|
||||||
|
type: "string"
|
||||||
|
x-kubernetes-group-version-kind:
|
||||||
|
- group: "management.cattle.io"
|
||||||
|
version: "v2"
|
||||||
|
kind: "GlobalRole"
|
||||||
|
io.management.cattle.NotAKind:
|
||||||
|
type: "string"
|
||||||
|
description: "Some string which isn't a kind"
|
||||||
|
io.k8s.apimachinery.pkg.apis.meta.v1.ObjectMeta:
|
||||||
|
description: "Object Metadata"
|
||||||
|
properties:
|
||||||
|
annotations:
|
||||||
|
description: "annotations of the resource"
|
||||||
|
type: "object"
|
||||||
|
additionalProperties:
|
||||||
|
type: "string"
|
||||||
|
name:
|
||||||
|
description: "name of the resource"
|
||||||
|
type: "string"
|
||||||
|
`
|
||||||
259
pkg/schema/definitions/visitor_test.go
Normal file
259
pkg/schema/definitions/visitor_test.go
Normal file
@@ -0,0 +1,259 @@
|
|||||||
|
package definitions
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
"k8s.io/kube-openapi/pkg/util/proto"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
protoPrimitive = proto.Primitive{
|
||||||
|
BaseSchema: proto.BaseSchema{
|
||||||
|
Description: "primitive value",
|
||||||
|
},
|
||||||
|
Type: "string",
|
||||||
|
}
|
||||||
|
protoPrimitiveInt = proto.Primitive{
|
||||||
|
BaseSchema: proto.BaseSchema{
|
||||||
|
Description: "primitive value - int",
|
||||||
|
},
|
||||||
|
Type: "integer",
|
||||||
|
}
|
||||||
|
protoPrimitiveNumber = proto.Primitive{
|
||||||
|
BaseSchema: proto.BaseSchema{
|
||||||
|
Description: "primitive value - number",
|
||||||
|
},
|
||||||
|
Type: "number",
|
||||||
|
}
|
||||||
|
protoArray = proto.Array{
|
||||||
|
BaseSchema: proto.BaseSchema{
|
||||||
|
Description: "testArray",
|
||||||
|
},
|
||||||
|
SubType: &protoPrimitive,
|
||||||
|
}
|
||||||
|
protoMap = proto.Map{
|
||||||
|
BaseSchema: proto.BaseSchema{
|
||||||
|
Description: "testMap",
|
||||||
|
},
|
||||||
|
SubType: &protoPrimitive,
|
||||||
|
}
|
||||||
|
protoKind = proto.Kind{
|
||||||
|
BaseSchema: proto.BaseSchema{
|
||||||
|
Description: "testKind",
|
||||||
|
Path: proto.NewPath("io.cattle.test"),
|
||||||
|
},
|
||||||
|
Fields: map[string]proto.Schema{
|
||||||
|
"protoArray": &protoArray,
|
||||||
|
"protoPrimitive": &protoPrimitive,
|
||||||
|
"protoMap": &protoMap,
|
||||||
|
},
|
||||||
|
RequiredFields: []string{
|
||||||
|
"protoArray",
|
||||||
|
"protoPrimitive",
|
||||||
|
"missing",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
protoRefNoSubSchema = testRef{
|
||||||
|
BaseSchema: proto.BaseSchema{
|
||||||
|
Description: "testRef - no subSchema",
|
||||||
|
},
|
||||||
|
reference: "some-other-type",
|
||||||
|
}
|
||||||
|
protoRef = testRef{
|
||||||
|
BaseSchema: proto.BaseSchema{
|
||||||
|
Description: "testRef",
|
||||||
|
},
|
||||||
|
reference: "testKind",
|
||||||
|
subSchema: &protoKind,
|
||||||
|
}
|
||||||
|
protoArbitrary = proto.Arbitrary{
|
||||||
|
BaseSchema: proto.BaseSchema{
|
||||||
|
Description: "testArbitrary",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
// testRef implements proto.Reference to test VisitReference
|
||||||
|
type testRef struct {
|
||||||
|
proto.BaseSchema
|
||||||
|
reference string
|
||||||
|
subSchema proto.Schema
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *testRef) Reference() string {
|
||||||
|
return t.reference
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *testRef) SubSchema() proto.Schema {
|
||||||
|
return t.subSchema
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *testRef) Accept(v proto.SchemaVisitor) {
|
||||||
|
v.VisitReference(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t *testRef) GetName() string {
|
||||||
|
return fmt.Sprintf("Reference to %q", t.reference)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSchemaFieldVisitor(t *testing.T) {
|
||||||
|
protoKind.Fields["protoRef"] = &protoRef
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
inputSchema proto.Schema
|
||||||
|
wantDefinitions map[string]definition
|
||||||
|
wantField definitionField
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "array",
|
||||||
|
inputSchema: &protoArray,
|
||||||
|
wantDefinitions: map[string]definition{},
|
||||||
|
wantField: definitionField{
|
||||||
|
Type: "array",
|
||||||
|
Description: protoArray.Description,
|
||||||
|
SubType: protoPrimitive.Type,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "map",
|
||||||
|
inputSchema: &protoMap,
|
||||||
|
wantDefinitions: map[string]definition{},
|
||||||
|
wantField: definitionField{
|
||||||
|
Type: "map",
|
||||||
|
Description: protoMap.Description,
|
||||||
|
SubType: protoPrimitive.Type,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "string primitive",
|
||||||
|
inputSchema: &protoPrimitive,
|
||||||
|
wantDefinitions: map[string]definition{},
|
||||||
|
wantField: definitionField{
|
||||||
|
Type: protoPrimitive.Type,
|
||||||
|
Description: protoPrimitive.Description,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "integer primitive",
|
||||||
|
inputSchema: &protoPrimitiveInt,
|
||||||
|
wantDefinitions: map[string]definition{},
|
||||||
|
wantField: definitionField{
|
||||||
|
Type: "int",
|
||||||
|
Description: protoPrimitiveInt.Description,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "number primitive",
|
||||||
|
inputSchema: &protoPrimitiveNumber,
|
||||||
|
wantDefinitions: map[string]definition{},
|
||||||
|
wantField: definitionField{
|
||||||
|
Type: "int",
|
||||||
|
Description: protoPrimitiveNumber.Description,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "kind",
|
||||||
|
inputSchema: &protoKind,
|
||||||
|
wantDefinitions: map[string]definition{
|
||||||
|
protoKind.Path.String(): {
|
||||||
|
ResourceFields: map[string]definitionField{
|
||||||
|
"protoArray": {
|
||||||
|
Type: "array",
|
||||||
|
Description: protoArray.Description,
|
||||||
|
SubType: protoPrimitive.Type,
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
"protoMap": {
|
||||||
|
Type: "map",
|
||||||
|
Description: protoMap.Description,
|
||||||
|
SubType: protoPrimitive.Type,
|
||||||
|
},
|
||||||
|
"protoPrimitive": {
|
||||||
|
Type: protoPrimitive.Type,
|
||||||
|
Description: protoPrimitive.Description,
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
"protoRef": {
|
||||||
|
Type: protoKind.Path.String(),
|
||||||
|
Description: protoRef.Description,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Type: protoKind.Path.String(),
|
||||||
|
Description: protoKind.Description,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wantField: definitionField{
|
||||||
|
Description: protoKind.Description,
|
||||||
|
Type: protoKind.Path.String(),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "reference no subschema",
|
||||||
|
inputSchema: &protoRefNoSubSchema,
|
||||||
|
wantDefinitions: map[string]definition{},
|
||||||
|
wantField: definitionField{
|
||||||
|
Type: protoRefNoSubSchema.reference,
|
||||||
|
Description: protoRefNoSubSchema.Description,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "reference",
|
||||||
|
inputSchema: &protoRef,
|
||||||
|
wantDefinitions: map[string]definition{
|
||||||
|
protoKind.Path.String(): {
|
||||||
|
ResourceFields: map[string]definitionField{
|
||||||
|
"protoArray": {
|
||||||
|
Type: "array",
|
||||||
|
Description: protoArray.Description,
|
||||||
|
SubType: protoPrimitive.Type,
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
"protoMap": {
|
||||||
|
Type: "map",
|
||||||
|
Description: protoMap.Description,
|
||||||
|
SubType: protoPrimitive.Type,
|
||||||
|
},
|
||||||
|
"protoPrimitive": {
|
||||||
|
Type: protoPrimitive.Type,
|
||||||
|
Description: protoPrimitive.Description,
|
||||||
|
Required: true,
|
||||||
|
},
|
||||||
|
"protoRef": {
|
||||||
|
Type: protoKind.Path.String(),
|
||||||
|
Description: protoRef.Description,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Type: protoKind.Path.String(),
|
||||||
|
Description: protoKind.Description,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wantField: definitionField{
|
||||||
|
Type: protoKind.Path.String(),
|
||||||
|
Description: protoRef.Description,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "abitrary schema",
|
||||||
|
inputSchema: &protoArbitrary,
|
||||||
|
wantDefinitions: map[string]definition{},
|
||||||
|
wantField: definitionField{
|
||||||
|
Type: "string",
|
||||||
|
Description: protoArbitrary.Description,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, test := range tests {
|
||||||
|
t.Run(test.name, func(t *testing.T) {
|
||||||
|
definitions := map[string]definition{}
|
||||||
|
visitor := schemaFieldVisitor{
|
||||||
|
definitions: definitions,
|
||||||
|
}
|
||||||
|
test.inputSchema.Accept(&visitor)
|
||||||
|
require.Equal(t, test.wantField, visitor.field)
|
||||||
|
require.Equal(t, test.wantDefinitions, visitor.definitions)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user