cmd/cue/cmd/fix_test.go

// 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 cmd

import (
	"testing"

	"cuelang.org/go/cue/format"
	"cuelang.org/go/cue/parser"
)

func TestFix(t *testing.T) {
	testCases := []struct {
		name string
		in   string
		out  string
	}{{
		name: "referenced string fields",
		in: `package foo

"foo": 3
"foo-bar": 2
"baz": ` + "`foo-bar`" + `

a: {
	// qux
	"qux": 3 // qux line
	// qux-quux
	"qux-quux": qux
	"qaz": ` + "`qux-quux`" + `
	qax:  qux

	fiz: 4
	faz: ` + "`fiz`" + `
	// fuz
	fuz: ` + "`qux-quux`" + ` // fuz

	// biz
	` + "`biz`" + `: 5 // biz
	buz: ` + "`biz`" + `
	baz: ` + "`qux`" + `
}
`,
		out: `package foo

"foo":        3
X1="foo-bar": 2
"baz":        X1

a: {
	// qux
	X2="qux": 3 // qux line
	// qux-quux
	X3="qux-quux": X2
	"qaz":         X3
	qax:           X2

	fiz: 4
	faz: fiz
	// fuz
	fuz: X3 // fuz

	// biz
	biz: 5 // biz
	buz: biz
	baz: X2
}
`,
	}, {
		name: "comprehensions",
		in: `
package fix

"\(k)": v for k, v in src

"\(k)": v <-
 for k, v in src

/* foo
   bar
 */

a: 3 + /* foo */ 5
	 `,
		out: `package fix

for k, v in src {
	"\(k)": v
}

// foo
// bar
for k, v in src {
	"\(k)": v
}

a:
	// foo
	3 + 5
`,
	}, {
		name: "comments",
		in: `package foo

a: /* b */ 3 + 5
a: 3 /* c */ + 5
a: 3 + /* d */ 5
a: 3 + 5 /* e
f */
`,
		out: `package foo

// b
a: 3 + 5
a:
	// c
	3 + 5
a:
	// d
	3 + 5
// e
// f
a: 3 + 5
`,
	}, {
		name: "templates",
		in: `package foo

a: <Name>: { name: Name }
b: <X>:    { name: string }
`,
		out: `package foo

a: [Name=_]: {name: Name}
b: [string]: {name: string}
`,
		// 	}, {
		// 		name: "slice",
		// 		in: `package foo

		// // keep comment
		// l[3:4] // and this one

		// a: len(l[3:4])
		// b: len(l[a:_])
		// c: len(l[_:x])
		// d: len(l[_:_])
		// `,
		// 		out: `package foo

		// import list6c6973 "list"

		// // keep comment
		// list6c6973.Slice(l, 3, 4)// and this one

		// a: len(list6c6973.Slice(l, 3, 4))
		// b: len(list6c6973.Slice(l, a, len(l)))
		// c: len(list6c6973.Slice(l, 0, x))
		// d: len(list6c6973.Slice(l, 0, len(l)))
		// `,
		// 	}, {
		// 		name: "slice2",
		// 		in: `package foo

		// import "list"

		// a: list.Contains("foo")
		// b: len(l[_:_])
		// `,
		// 		out: `package foo

		// import (
		// 	"list"
		// 	list6c6973 "list"
		// )

		// a: list.Contains("foo")
		// b: len(list6c6973.Slice(l, 0, len(l)))
		// `,
	}}
	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			f, err := parser.ParseFile("", tc.in, parser.ParseComments)
			if err != nil {
				t.Fatal(err)
			}
			n := fix(f)
			b, err := format.Node(n)
			if err != nil {
				t.Fatal(err)
			}
			got := string(b)
			if got != tc.out {
				t.Errorf("got %v; want %v", got, tc.out)
			}
			_, err = parser.ParseFile("rewritten", got, parser.ParseComments)
			if err != nil {
				t.Fatal(err)
			}
		})
	}
}