Add func args support.

mocktail.go

@@ -223,6 +223,9 @@ func getTypeImports(t types.Type) []string {
 	case *types.Interface:
 		return []string{""}
 
+	case *types.Signature:
+		return getTupleImports(v.Params(), v.Results())
+
 	default:
 		panic(fmt.Sprintf("OOPS %[1]T %[1]s", t))
 	}

syrup.go

@@ -271,7 +271,12 @@ func (s Syrup) methodOn(writer io.Writer) error {
 		name := getParamName(param, i)
 
 		w.Print(name)
-		argNames = append(argNames, name)
+
+		if _, ok := param.Type().(*types.Signature); ok {
+			argNames = append(argNames, "mock.Anything")
+		} else {
+			argNames = append(argNames, name)
+		}
 
 		w.Print(" " + s.getTypeName(param.Type(), i == params.Len()-1))
 
@@ -312,7 +317,12 @@ func (s Syrup) methodOnRaw(writer io.Writer) error {
 		name := getParamName(param, i)
 
 		w.Print(name)
-		argNames = append(argNames, name)
+
+		if _, ok := param.Type().(*types.Signature); ok {
+			argNames = append(argNames, "mock.Anything")
+		} else {
+			argNames = append(argNames, name)
+		}
 
 		w.Print(" interface{}")
 
@@ -616,11 +626,31 @@ func (s Syrup) getTypeName(t types.Type, last bool) string {
 	case *types.Interface:
 		return v.String()
 
+	case *types.Signature:
+		fn := "func(" + strings.Join(s.getTupleTypes(v.Params()), ",") + ")"
+
+		if v.Results().Len() > 0 {
+			fn += " (" + strings.Join(s.getTupleTypes(v.Results()), ",") + ")"
+		}
+
+		return fn
+
 	default:
 		panic(fmt.Sprintf("OOPS %[1]T %[1]s", t))
 	}
 }
 
+func (s Syrup) getTupleTypes(t *types.Tuple) []string {
+	var tupleTypes []string
+	for i := 0; i < t.Len(); i++ {
+		param := t.At(i)
+
+		tupleTypes = append(tupleTypes, s.getTypeName(param.Type(), false))
+	}
+
+	return tupleTypes
+}
+
 func writeImports(writer io.Writer, descPkg PackageDesc) error {
 	base := template.New("templateImports")
 

testdata/src/a/foo.go

@@ -28,6 +28,7 @@ type Coconut interface {
 	Voo(src *module.Version) time.Duration
 	Yoo(st string) interface{}
 	Zoo(st interface{}) string
+	Moo(fn func(st, stban Strawberry) Pineapple) string
 }
 
 type Water struct{}