feat: support types.Struct as interface args

traefik · Feb 7, 2025 · d5507e7 · d5507e7
1 parent 77e6780
commit d5507e7
Show file tree

Hide file tree

Showing 5 changed files with 640 additions and 0 deletions.
diff --git a/mocktail.go b/mocktail.go
@@ -216,6 +216,13 @@ func getTypeImports(t types.Type) []string {
 	case *types.Array:
 		return getTypeImports(v.Elem())
 
+	case *types.Struct:
+		var imports []string
+		for i := range v.NumFields() {
+			imports = append(imports, getTypeImports(v.Field(i).Type())...)
+		}
+		return imports
+
 	case *types.Map:
 		imports := getTypeImports(v.Key())
 		imports = append(imports, getTypeImports(v.Elem())...)

diff --git a/testdata/src/a/a.go b/testdata/src/a/a.go
@@ -31,6 +31,7 @@ type Coconut interface {
 	Zoo(st interface{}) string
 	Moo(fn func(st, stban Strawberry) Pineapple) string
 	Noo(ar [][2]string) string
+	Poo(str struct{ name string }) string
 }
 
 type Water struct{}