feat: support types.Struct as interface args

traefik · Sep 6, 2024 · 3cc9013 · 3cc9013
1 parent b49c602
commit 3cc9013
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 := 0; i < v.NumFields(); i++ {
+			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{}