kairos-agent/pkg/config/schemas/root_schema_test.go

package config_test

import (
	"fmt"
	"reflect"
	"strings"

	. "github.com/kairos-io/kairos/pkg/config"
	. "github.com/kairos-io/kairos/pkg/config/schemas"
	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
)

func getTagName(s string) string {
	if len(s) < 1 {
		return ""
	}

	f := func(c rune) bool {
		return c == '"' || c == ','
	}
	return s[:strings.IndexFunc(s, f)]
}

func structContainsField(f, t string, str interface{}) bool {
	values := reflect.ValueOf(str)
	types := values.Type()

	for j := 0; j < values.NumField(); j++ {
		tagName := getTagName(types.Field(j).Tag.Get("json"))
		if types.Field(j).Name == f || tagName == t {
			return true
		} else {
			if types.Field(j).Type.Kind() == reflect.Struct {
				if types.Field(j).Type.Name() != "" {
					model := reflect.New(types.Field(j).Type)
					if instance, ok := model.Interface().(OneOfModel); ok {
						for _, childSchema := range instance.JSONSchemaOneOf() {
							if structContainsField(f, t, childSchema) {
								return true
							}
						}
					}
				}
			}
		}
	}

	return false
}

func structFieldsContainedInOtherStruct(left, right interface{}) {
	leftValues := reflect.ValueOf(left)
	leftTypes := leftValues.Type()

	for i := 0; i < leftValues.NumField(); i++ {
		leftTagName := getTagName(leftTypes.Field(i).Tag.Get("yaml"))
		leftFieldName := leftTypes.Field(i).Name
		if leftTypes.Field(i).IsExported() {
			It(fmt.Sprintf("Checks that the new schema contians the field %s", leftFieldName), func() {
				Expect(
					structContainsField(leftFieldName, leftTagName, right),
				).To(BeTrue())
			})
		}
	}
}

var _ = Describe("Schema", func() {
	var config *KConfig
	var err error
	var yaml string

	JustBeforeEach(func() {
		config, err = NewConfigFromYAML(yaml, DefaultHeader, RootSchema{})
	})

	Context("While the new Schema is not the single source of truth", func() {
		structFieldsContainedInOtherStruct(Config{}, RootSchema{})
	})
	Context("While the new InstallSchema is not the single source of truth", func() {
		structFieldsContainedInOtherStruct(Install{}, InstallSchema{})
	})
	Context("While the new BundleSchema is not the single source of truth", func() {
		structFieldsContainedInOtherStruct(Bundle{}, BundleSchema{})
	})

	Context("With invalid YAML syntax", func() {
		BeforeEach(func() {
			yaml = `#cloud-config
this is:
- invalid
yaml`
		})

		It("errors", func() {
			Expect(err.Error()).To(MatchRegexp("yaml: line 4: could not find expected ':'"))
		})
	})

	Context("With the wrong header", func() {
		BeforeEach(func() {
			yaml = `---
users:
- name: "kairos"
  passwd: "kairos"`
		})

		It("errors", func() {
			Expect(err.Error()).To(MatchRegexp("missing #cloud-config header"))
		})
	})

	Context("When `users` is empty", func() {
		BeforeEach(func() {
			yaml = `#cloud-config
users: []`
		})

		It("errors", func() {
			Expect(err).ToNot(HaveOccurred())
			Expect(config.IsValid()).NotTo(BeTrue())
			Expect(config.ValidationError()).To(MatchRegexp("minimum 1 items required, but found 0 items"))
		})
	})
})