tests: add tests on generateImports

Author: ldez

pkg/commands/internal/imports_test.go

@@ -0,0 +1,36 @@
+package internal
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func Test_generateImports(t *testing.T) {
+	cfg := &Configuration{
+		Version: "v1.57.0",
+		Plugins: []*Plugin{
+			{
+				Module:  "example.org/foo/bar",
+				Import:  "example.org/foo/bar/test",
+				Version: "v1.2.3",
+			},
+			{
+				Module: "example.com/foo/bar",
+				Import: "example.com/foo/bar/test",
+				Path:   "/my/path",
+			},
+		},
+	}
+
+	data, err := generateImports(cfg)
+	require.NoError(t, err)
+
+	expected, err := os.ReadFile(filepath.Join("testdata", "imports.go"))
+	require.NoError(t, err)
+
+	assert.Equal(t, expected, data)
+}

pkg/commands/internal/testdata/imports.go

@@ -0,0 +1,6 @@
+package main
+
+import (
+	_ "example.com/foo/bar/test"
+	_ "example.org/foo/bar/test"
+)