| // Copyright Istio Authors |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| package codegen |
| |
| import ( |
| "fmt" |
| "strings" |
| "testing" |
| ) |
| |
| import ( |
| . "github.com/onsi/gomega" |
| ) |
| |
| import ( |
| "github.com/apache/dubbo-go-pixiu/pkg/config/schema/ast" |
| "github.com/apache/dubbo-go-pixiu/pkg/test/util/assert" |
| ) |
| |
| func TestStaticCollections(t *testing.T) { |
| cases := []struct { |
| packageName string |
| m *ast.Metadata |
| err string |
| output string |
| }{ |
| { |
| packageName: "pkg", |
| m: &ast.Metadata{ |
| Collections: []*ast.Collection{ |
| { |
| Name: "foo", |
| VariableName: "Foo", |
| Description: "describes a really cool foo thing", |
| Group: "foo.group", |
| Kind: "fookind", |
| }, |
| { |
| Name: "bar", |
| VariableName: "Bar", |
| Description: "describes a really cool bar thing", |
| Group: "bar.group", |
| Kind: "barkind", |
| }, |
| }, |
| Resources: []*ast.Resource{ |
| { |
| Group: "foo.group", |
| Version: "v1", |
| Kind: "fookind", |
| Plural: "fookinds", |
| ClusterScoped: true, |
| Proto: "google.protobuf.Struct", |
| ProtoPackage: "github.com/gogo/protobuf/types", |
| Validate: "EmptyValidate", |
| }, |
| { |
| Group: "bar.group", |
| Version: "v1", |
| Kind: "barkind", |
| Plural: "barkinds", |
| ClusterScoped: false, |
| Proto: "google.protobuf.Struct", |
| ProtoPackage: "github.com/gogo/protobuf/types", |
| Validate: "EmptyValidate", |
| }, |
| }, |
| }, |
| output: ` |
| // GENERATED FILE -- DO NOT EDIT |
| // |
| |
| package pkg |
| |
| import ( |
| "github.com/apache/dubbo-go-pixiu/pkg/config/schema/collection" |
| "github.com/apache/dubbo-go-pixiu/pkg/config/schema/resource" |
| "github.com/apache/dubbo-go-pixiu/pkg/config/validation" |
| "reflect" |
| githubcomgogoprotobuftypes "github.com/gogo/protobuf/types" |
| ) |
| |
| var ( |
| |
| // Bar describes a really cool bar thing |
| Bar = collection.Builder { |
| Name: "bar", |
| VariableName: "Bar", |
| Resource: resource.Builder { |
| Group: "bar.group", |
| Kind: "barkind", |
| Plural: "barkinds", |
| Version: "v1", |
| Proto: "google.protobuf.Struct", |
| ReflectType: reflect.TypeOf(&githubcomgogoprotobuftypes.Struct{}).Elem(), |
| ProtoPackage: "github.com/gogo/protobuf/types", |
| ClusterScoped: false, |
| ValidateProto: validation.EmptyValidate, |
| }.MustBuild(), |
| }.MustBuild() |
| |
| // Foo describes a really cool foo thing |
| Foo = collection.Builder { |
| Name: "foo", |
| VariableName: "Foo", |
| Resource: resource.Builder { |
| Group: "foo.group", |
| Kind: "fookind", |
| Plural: "fookinds", |
| Version: "v1", |
| Proto: "google.protobuf.Struct", |
| ReflectType: reflect.TypeOf(&githubcomgogoprotobuftypes.Struct{}).Elem(), |
| ProtoPackage: "github.com/gogo/protobuf/types", |
| ClusterScoped: true, |
| ValidateProto: validation.EmptyValidate, |
| }.MustBuild(), |
| }.MustBuild() |
| |
| |
| // All contains all collections in the system. |
| All = collection.NewSchemasBuilder(). |
| MustAdd(Bar). |
| MustAdd(Foo). |
| Build() |
| |
| // Istio contains only Istio collections. |
| Istio = collection.NewSchemasBuilder(). |
| Build() |
| |
| // Kube contains only kubernetes collections. |
| Kube = collection.NewSchemasBuilder(). |
| Build() |
| |
| // Builtin contains only native Kubernetes collections. This differs from Kube, which has |
| // Kubernetes controlled CRDs |
| Builtin = collection.NewSchemasBuilder(). |
| Build() |
| |
| // Pilot contains only collections used by Pilot. |
| Pilot = collection.NewSchemasBuilder(). |
| Build() |
| |
| // PilotGatewayAPI contains only collections used by Pilot, including experimental Service Api. |
| PilotGatewayAPI = collection.NewSchemasBuilder(). |
| Build() |
| |
| // Deprecated contains only collections used by that will soon be used by nothing. |
| Deprecated = collection.NewSchemasBuilder(). |
| Build() |
| ) |
| `, |
| }, |
| } |
| |
| for _, c := range cases { |
| t.Run("", func(t *testing.T) { |
| g := NewWithT(t) |
| |
| s, err := StaticCollections(c.packageName, c.m, func(name string) bool { |
| return true |
| }, "") |
| if c.err != "" { |
| g.Expect(err).NotTo(BeNil()) |
| g.Expect(err.Error()).To(Equal(s)) |
| } else { |
| g.Expect(err).To(BeNil()) |
| fmt.Println(strings.TrimSpace(c.output)) |
| assert.Equal(t, strings.TrimSpace(s), strings.TrimSpace(c.output)) |
| } |
| }) |
| } |
| } |