github.com/asdine/cueimports@v0.3.2/cueimports_test.go

package cueimports

import (
	"fmt"
	"os"
	"testing"

	"cuelang.org/go/cue/parser"
	"github.com/stretchr/testify/require"
)

func TestImport(t *testing.T) {
	err := os.Chdir("testdata")
	require.NoError(t, err)

	tests := []struct {
		filename    string
		passContent bool
		want        string
	}{
		{"file.cue", false, "file.cue.golden"},
		{"file.cue", true, "file.cue.golden"},
		{"default.cue", false, "default.cue.golden"},
		{"default.cue", true, "default.cue.golden"},
		{"unchanged.cue", false, "unchanged.cue.golden"},
		{"unchanged.cue", true, "unchanged.cue.golden"},
		{"unused_import.cue", false, "unused_import.cue.golden"},
		{"unused_import.cue", true, "unused_import.cue.golden"},
		{"top_level_name_clash.cue", false, "top_level_name_clash.cue.golden"},
		{"top_level_name_clash.cue", true, "top_level_name_clash.cue.golden"},
		{"import_colon.cue", true, "import_colon.cue.golden"},
		{"import_colon_unchanged.cue", true, "import_colon_unchanged.cue.golden"},
	}

	for _, tt := range tests {
		t.Run(fmt.Sprintf("%s/%v", tt.filename, tt.passContent), func(t *testing.T) {
			var content []byte
			if tt.passContent {
				content, err = os.ReadFile(tt.filename)
				require.NoError(t, err)
			}
			res, err := Import(tt.filename, content)
			require.NoError(t, err)

			expected, err := os.ReadFile(tt.want)
			require.NoError(t, err)

			require.Equal(t, string(expected), string(res))
		})
	}
}

func TestImportInsertImports(t *testing.T) {
	tests := []struct {
		name     string
		input    string
		packages map[string]string
		want     string
	}{
		{
			name:  "empty file",
			input: "",
			want:  "\n",
		},
		{
			name:  "no package",
			input: "b: math.Round(1.5)",
			packages: map[string]string{
				"math": "math",
			},
			want: `import "math"

b: math.Round(1.5)
`,
		},
		{
			name: "with package",
			input: `package test

			b: math.Round(1.5)
`,
			packages: map[string]string{
				"math": "math",
			},
			want: `package test

import "math"

b: math.Round(1.5)
`,
		},
		{
			name: "with comment",
			input: `package test
			// some comment

			b: math.Round(1.5)
`,
			packages: map[string]string{
				"math": "math",
			},
			want: `package test

// some comment
import "math"

b: math.Round(1.5)
`,
		},
		{
			name: "with comment at the top",
			input: `// some comment
			package test

			b: math.Round(1.5)
`,
			packages: map[string]string{
				"math": "math",
			},
			want: `// some comment
package test

import "math"

b: math.Round(1.5)
`,
		},
		{
			name: "with existing import",
			input: `import "math"

b: math.Round(1.5)
`,
			packages: map[string]string{
				"math": "math",
			},
			want: `import "math"

b: math.Round(1.5)
`,
		},
		{
			name: "with existing different import",
			input: `import "encoding/json"

b: math.Round(1.5)
c: json.Marshal(1)
`,
			packages: map[string]string{
				"math": "math",
				"json": "encoding/json",
			},
			want: `import (
	"encoding/json"
	"math"
)

b: math.Round(1.5)
c: json.Marshal(1)
`,
		},
		{
			name: "with custom package",
			input: `
a: foo.#Bar
b: math.Round(1.5)
c: json.Marshal(1)
`,
			packages: map[string]string{
				"math": "math",
				"json": "encoding/json",
				"foo":  "test.com/foo",
			},
			want: `import (
	"encoding/json"
	"math"
	
	"test.com/foo"
)

a: foo.#Bar
b: math.Round(1.5)
c: json.Marshal(1)
`,
		},
		{
			name: "with std package clash",
			input: `import "encoding/json"
a: tool.#Foo
b: math.Round(1.5)
c: json.Marshal(1)
`,
			packages: map[string]string{
				"math": "math",
				"json": "encoding/json",
				"tool": "foo.com/tool",
			},
			want: `import (
	"encoding/json"
	"math"
	
	"foo.com/tool"
)

a: tool.#Foo
b: math.Round(1.5)
c: json.Marshal(1)
`,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			f, err := parser.ParseFile("", tt.input,
				parser.ParseComments,
				parser.AllowPartial,
			)
			require.NoError(t, err)
			res, err := insertImports(f, tt.packages)
			require.NoError(t, err)
			require.Equal(t, tt.want, string(res))
		})
	}
}