support imported type aliases

go.mod

@@ -1,6 +1,6 @@
 module github.com/matryer/moq
 
-go 1.22.5
+go 1.23
 
 require (
 	github.com/pmezard/go-difflib v1.0.0

internal/registry/method_scope.go

@@ -90,6 +90,18 @@ func (m MethodScope) populateImports(t types.Type, imports map[string]*Package)
 			}
 		}
 
+	case *types.Alias:
+		if pkg := t.Obj().Pkg(); pkg != nil {
+			imports[stripVendorPath(pkg.Path())] = m.registry.AddImport(pkg)
+		}
+		// The imports of a Type with a TypeList must be added to the imports list
+		// For example: Foo[otherpackage.Bar] , must have otherpackage imported
+		if targs := t.TypeArgs(); targs != nil {
+			for i := 0; i < targs.Len(); i++ {
+				m.populateImports(targs.At(i), imports)
+			}
+		}
+
 	case *types.Array:
 		m.populateImports(t.Elem(), imports)
 

pkg/moq/moq_test.go

@@ -412,6 +412,12 @@ func TestMockGolden(t *testing.T) {
 			interfaces: []string{"Magician"},
 			goldenFile: filepath.Join("testpackages/rangenum", "rangenum_moq.golden.go"),
 		},
+		{
+			name:       "TypeAlias",
+			cfg:        Config{SrcDir: "testpackages/typealias"},
+			interfaces: []string{"Example"},
+			goldenFile: filepath.Join("testpackages/typealias", "typealias_moq.golden.go"),
+		},
 	}
 	for _, tc := range cases {
 		t.Run(tc.name, func(t *testing.T) {

pkg/moq/testpackages/typealias/typealias.go

@@ -0,0 +1,9 @@
+package typealias
+
+import (
+	"github.com/matryer/moq/pkg/moq/testpackages/typealiastwo"
+)
+
+type Example interface {
+	Do(a typealiastwo.AliasType, b typealiastwo.GenericAliasType) error
+}

pkg/moq/testpackages/typealiastwo/internal/typealiasinternal/typealiasinternal.go

@@ -0,0 +1,10 @@
+package typealiasinternal
+
+// we shouldn't be able to import these types directly, you need to use the alias in the parent package
+type MyInternalType struct {
+	Foo int
+}
+
+type MyGenericType[T any] struct {
+	A T
+}

pkg/moq/testpackages/typealiastwo/typealiastwo.go

@@ -0,0 +1,7 @@
+package typealiastwo
+
+import "github.com/matryer/moq/pkg/moq/testpackages/typealiastwo/internal/typealiasinternal"
+
+type AliasType = typealiasinternal.MyInternalType
+
+type GenericAliasType = typealiasinternal.MyGenericType[int]