File tree Expand file tree Collapse file tree 1 file changed +9
-3
lines changed
Expand file tree Collapse file tree 1 file changed +9
-3
lines changed Original file line number Diff line number Diff 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 }}
You can’t perform that action at this time.
0 commit comments