feat(generate): accept []interface{}

a631807682 · Jan 18, 2022 · bb8de83 · bb8de83
1 parent 12a358b
commit bb8de83
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/internal/check/checktest.go b/internal/check/checktest.go
@@ -27,8 +27,15 @@ func (m *InterfaceMethod) GetTestResultParamInTmpl() string {
 func testParamToString(params []parser.Param) string {
 	var res []string
 	for i, param := range params {
-		tmplString := fmt.Sprintf("tt.Input.Args[%d].(%s)", i, param.Type)
-		res = append(res, tmplString)
+		// TODO manage array and pointer
+		typ := param.Type
+		if param.IsArray {
+			typ = "[]" + typ
+		}
+		if param.IsPointer {
+			typ = "*" + typ
+		}
+		res = append(res, fmt.Sprintf("tt.Input.Args[%d].(%s)", i, typ))
 	}
 	return strings.Join(res, ",")
 }

diff --git a/internal/parser/parser.go b/internal/parser/parser.go
@@ -231,11 +231,14 @@ func (p *Param) astGetEltType(expr ast.Expr) string {
 	case *ast.StarExpr:
 		p.IsPointer = true
 		p.astGetEltType(v.X)
+	case *ast.InterfaceType:
+		p.Type = "interface{}"
 	default:
 		log.Fatalf("unknow param type: %+v", v)
 	}
 	return p.Type
 }
+
 func (p *Param) astGetPackageName(expr ast.Expr) {
 	switch v := expr.(type) {
 	case *ast.Ident: