feat: add support no return types on generate mock

akito0107 · May 30, 2018 · 2bf1ee2 · 2bf1ee2
1 parent 40e6f1b
commit 2bf1ee2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/internal/generator.go b/internal/generator.go
@@ -188,7 +188,10 @@ func (g *Generator) appendMockStruct(it *InterfaceType) {
 				a = append(a, fmt.Sprintf("a%d", i))
 			}
 		}
-		g.Printf("return mk.%sMock(%s)\n", f.Name, strings.Join(a, ","))
+		if len(f.ReturnTypes) > 0 {
+			g.Printf("return ")
+		}
+		g.Printf("mk.%sMock(%s)\n", f.Name, strings.Join(a, ","))
 		g.Printf("}\n")
 	}
 }

diff --git a/internal/generator_test.go b/internal/generator_test.go
@@ -394,7 +394,7 @@ func TestAppendMockStructMultipleFuncs(t *testing.T) {
 	}
 
 	func (mk *TestInterfaceMock) TestFunc1(a0 Arg1) {
-		return mk.TestFunc1Mock(a0)
+		mk.TestFunc1Mock(a0)
 	}
 	func (mk *TestInterfaceMock) TestFunc2(a0 Arg1, a1 Arg2) (Ret1, Ret2) {
 		return mk.TestFunc2Mock(a0, a1)
@@ -457,7 +457,7 @@ func TestAppendMockStructMultipleFuncWithPackages(t *testing.T) {
 	}
 
 	func (mk *TestInterfaceMock) TestFunc1(a0 pak1.Arg1) {
-		return mk.TestFunc1Mock(a0)
+		mk.TestFunc1Mock(a0)
 	}
 	func (mk *TestInterfaceMock) TestFunc2(a0 pak1.Arg1, a1 Arg2) (Ret1, pak2.Ret2) {
 		return mk.TestFunc2Mock(a0, a1)