feat: generates returnsFn on function without args

traefik · Aug 29, 2022 · 3ec1844 · 3ec1844
1 parent e81c3a1
commit 3ec1844
Show file tree

Hide file tree

Showing 6 changed files with 382 additions and 4 deletions.
diff --git a/syrup.go b/syrup.go
@@ -208,7 +208,7 @@ func (s Syrup) mockedMethod(writer io.Writer) error {
 }
 
 func (s Syrup) writeReturnsFnCaller(w *Writer, argNames []string, params, results *types.Tuple) {
-	if len(argNames) > 0 && results.Len() > 0 {
+	if results.Len() > 0 {
 		w.Println()
 		w.Printf("\tif _rf, ok := _ret.Get(0).(%s); ok {\n", s.createFuncSignature(params, results))
 		w.Printf("\t\treturn _rf(%s", strings.Join(argNames, ", "))
@@ -453,9 +453,6 @@ func (s Syrup) returnsFn(writer io.Writer) error {
 	}
 
 	params := s.Signature.Params()
-	if params.Len() < 1 {
-		return nil
-	}
 
 	structBaseName := strcase.ToGoCamel(s.InterfaceName)
 

diff --git a/testdata/src/a/a.go b/testdata/src/a/a.go
@@ -13,6 +13,7 @@ type Pineapple interface {
 	World() string
 	Goo() (string, int, Water)
 	Coo(context.Context, string, Water) Water
+	Noo(context.Context) string
 }
 
 type Coconut interface {

diff --git a/testdata/src/a/mock_gen_test.go b/testdata/src/a/mock_gen_test.go