Skip to content

Commit a90446c

Browse files
committed
Fix mockery.templ
1 parent c7fd6ba commit a90446c

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

pkg/template/mockery.templ

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,14 +50,20 @@ func (_mock *{{$mock.MockName}}{{ $mock | TypeInstantiation }}) {{$method.Name}}
5050
($method.IsVariadic)
5151
(not (index $mock.TemplateData "unroll-variadic"))
5252
}}
53+
54+
{{- if ne (len $method.Returns) 0}}
5355
var tmpRet mock.Arguments
56+
{{- $calledString = "tmpRet" }}
57+
{{- else }}
58+
{{- $calledString = "" }}
59+
{{- end }}
60+
5461
{{- $lastParam := index $method.Params (len $method.Params | Add -1 )}}
5562
if len({{ $lastParam.Var.Name }}) > 0 {
56-
tmpRet = _mock.Called({{- if (index $mock.TemplateData "unroll-variadic") }}{{ $method.ArgCallList }}{{- else }}{{ $method.ArgCallListNoEllipsis }}{{- end }})
63+
{{- if ne (len $method.Returns) 0}}tmpRet = {{ end }}_mock.Called({{- if (index $mock.TemplateData "unroll-variadic") }}{{ $method.ArgCallList }}{{- else }}{{ $method.ArgCallListNoEllipsis }}{{- end }})
5764
} else {
58-
tmpRet = _mock.Called({{- if (index $mock.TemplateData "unroll-variadic") }}{{ $method.ArgCallListSlice 0 (len $method.Params | Add -1 )}}{{- else }}{{ $method.ArgCallListSliceNoEllipsis 0 (len $method.Params | Add -1 )}}{{- end }})
65+
{{- if ne (len $method.Returns) 0}}tmpRet = {{ end }}_mock.Called({{- if (index $mock.TemplateData "unroll-variadic") }}{{ $method.ArgCallListSlice 0 (len $method.Params | Add -1 )}}{{- else }}{{ $method.ArgCallListSliceNoEllipsis 0 (len $method.Params | Add -1 )}}{{- end }})
5966
}
60-
{{- $calledString = "tmpRet" }}
6167
{{- else }}
6268
{{- $calledString = printf "_mock.Called(%s)" $method.ArgCallList }}
6369
{{- end }}

0 commit comments

Comments
 (0)