grafana/pkg/schema/schema_test.go

package schema

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"path/filepath"
	"strings"
	"testing"

	"cuelang.org/go/cue"
	"cuelang.org/go/cue/cuecontext"
	"github.com/google/go-cmp/cmp"
	"golang.org/x/tools/txtar"
)

var CasesDir = filepath.Join("testdata", "trimapplydefaults")

type Case struct {
	Name    string
	CUE     string
	Full    string
	Trimmed string
}

func TestGenerate(t *testing.T) {
	cases, err := loadCases(CasesDir)
	if err != nil {
		t.Fatal(err)
	}
	for _, c := range cases {
		t.Run(c.Name+" apply defaults", func(t *testing.T) {
			ctx := cuecontext.New()
			scmInstance := ctx.CompileString(c.CUE, cue.Filename(c.Name+".cue"))
			if scmInstance.Err() != nil {
				t.Fatal(scmInstance.Err())
			}
			inputResource := Resource{Value: c.Trimmed}
			out, err := ApplyDefaults(inputResource, scmInstance)
			if err != nil {
				t.Fatal(err)
			}
			b := []byte(out.Value.(string))
			b, _ = JsonRemarshal(b)

			if s := cmp.Diff(c.Full, string(b)); s != "" {
				t.Fatal(s)
			}
		})
	}

	for _, c := range cases {
		t.Run(c.Name+" trim defaults", func(t *testing.T) {
			ctx := cuecontext.New()
			scmInstance := ctx.CompileString(c.CUE, cue.Filename(c.Name+".cue"))
			if scmInstance.Err() != nil {
				t.Fatal(scmInstance.Err())
			}
			inputResource := Resource{Value: c.Full}
			out, err := TrimDefaults(inputResource, scmInstance)
			if err != nil {
				t.Fatal(err)
			}
			b := []byte(out.Value.(string))
			b, _ = JsonRemarshal(b)

			if s := cmp.Diff(c.Trimmed, string(b)); s != "" {
				t.Fatal(s)
			}
		})
	}
}

func JsonRemarshal(bytes []byte) ([]byte, error) {
	var ifce interface{}
	err := json.Unmarshal(bytes, &ifce)
	if err != nil {
		return []byte{}, err
	}
	output, err := json.Marshal(ifce)
	outputstring := string(output)
	if err != nil {
		return []byte{}, err
	}
	outputstring = strings.Replace(outputstring, "\\u003c", "<", -1)
	outputstring = strings.Replace(outputstring, "\\u003e", ">", -1)
	outputstring = strings.Replace(outputstring, "\\u0026", "&", -1)
	return []byte(outputstring), nil
}

func loadCases(dir string) ([]Case, error) {
	files, err := ioutil.ReadDir(dir)
	if err != nil {
		return nil, err
	}
	var cases []Case
	for _, fi := range files {
		file := filepath.Join(dir, fi.Name())
		a, err := txtar.ParseFile(file)
		if err != nil {
			return nil, err
		}

		if len(a.Files) != 3 {
			return nil, fmt.Errorf("Malformed test case '%s': Must contain exactly three files (CUE, Full and Trimed), but has %d", file, len(a.Files))
		}

		fullBuffer := new(bytes.Buffer)
		fullJson := a.Files[1].Data
		if err := json.Compact(fullBuffer, fullJson); err != nil {
			return nil, err
		}

		trimBuffer := new(bytes.Buffer)
		trimedJson := a.Files[2].Data
		if err := json.Compact(trimBuffer, trimedJson); err != nil {
			return nil, err
		}

		cases = append(cases, Case{
			Name:    strings.TrimSuffix(fi.Name(), filepath.Ext(fi.Name())),
			CUE:     string(a.Files[0].Data),
			Full:    fullBuffer.String(),
			Trimmed: trimBuffer.String(),
		})
	}
	return cases, nil
}