| // Copyright 2019 CUE 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 jsonschema |
| |
| import ( |
| "bytes" |
| "flag" |
| "io/ioutil" |
| "os" |
| "path" |
| "path/filepath" |
| "strings" |
| "testing" |
| |
| "github.com/google/go-cmp/cmp" |
| "github.com/rogpeppe/go-internal/txtar" |
| "github.com/stretchr/testify/assert" |
| |
| "cuelang.org/go/cue" |
| "cuelang.org/go/cue/ast" |
| "cuelang.org/go/cue/errors" |
| "cuelang.org/go/cue/format" |
| "cuelang.org/go/cue/token" |
| "cuelang.org/go/encoding/json" |
| "cuelang.org/go/encoding/yaml" |
| _ "cuelang.org/go/pkg" |
| ) |
| |
| var update = flag.Bool("update", false, "update the test files") |
| |
| // TestDecode reads the testdata/*.txtar files, converts the contained |
| // JSON schema to CUE and compares it against the output. |
| // |
| // Use the --update flag to update test files with the corresponding output. |
| func TestDecode(t *testing.T) { |
| err := filepath.Walk("testdata", func(fullpath string, info os.FileInfo, err error) error { |
| _ = err |
| if !strings.HasSuffix(fullpath, ".txtar") { |
| return nil |
| } |
| |
| t.Run(fullpath, func(t *testing.T) { |
| a, err := txtar.ParseFile(fullpath) |
| if err != nil { |
| t.Fatal(err) |
| } |
| |
| cfg := &Config{ID: fullpath} |
| |
| if bytes.Contains(a.Comment, []byte("openapi")) { |
| cfg.Root = "#/components/schemas/" |
| cfg.Map = func(p token.Pos, a []string) ([]ast.Label, error) { |
| // Just for testing: does not validate the path. |
| return []ast.Label{ast.NewIdent("#" + a[len(a)-1])}, nil |
| } |
| } |
| |
| r := &cue.Runtime{} |
| var in *cue.Instance |
| var out, errout []byte |
| outIndex := -1 |
| errIndex := -1 |
| |
| for i, f := range a.Files { |
| switch path.Ext(f.Name) { |
| case ".json": |
| in, err = json.Decode(r, f.Name, f.Data) |
| case ".yaml": |
| in, err = yaml.Decode(r, f.Name, f.Data) |
| case ".cue": |
| out = f.Data |
| outIndex = i |
| case ".err": |
| errout = f.Data |
| errIndex = i |
| } |
| } |
| if err != nil { |
| t.Fatal(err) |
| } |
| |
| updated := false |
| |
| expr, err := Extract(in, cfg) |
| if err != nil { |
| got := []byte(errors.Details(err, nil)) |
| |
| got = bytes.TrimSpace(got) |
| errout = bytes.TrimSpace(errout) |
| |
| switch { |
| case !cmp.Equal(errout, got): |
| if *update { |
| a.Files[errIndex].Data = got |
| updated = true |
| break |
| } |
| t.Error(cmp.Diff(string(got), string(errout))) |
| } |
| } |
| |
| if expr != nil { |
| b, err := format.Node(expr, format.Simplify()) |
| if err != nil { |
| t.Fatal(errors.Details(err, nil)) |
| } |
| |
| // verify the generated CUE. |
| if !bytes.Contains(a.Comment, []byte("#noverify")) { |
| if _, err = r.Compile(fullpath, b); err != nil { |
| t.Fatal(errors.Details(err, nil)) |
| } |
| } |
| |
| b = bytes.TrimSpace(b) |
| out = bytes.TrimSpace(out) |
| |
| switch { |
| case !cmp.Equal(b, out): |
| if *update { |
| updated = true |
| a.Files[outIndex].Data = b |
| break |
| } |
| t.Error(cmp.Diff(string(out), string(b))) |
| } |
| } |
| |
| if updated { |
| b := txtar.Format(a) |
| err = ioutil.WriteFile(fullpath, b, 0644) |
| if err != nil { |
| t.Fatal(err) |
| } |
| } |
| }) |
| return nil |
| }) |
| assert.NoError(t, err) |
| } |