kjuulh/moq
1
0
Fork 0
Code Issues 1 Pull Requests Projects Releases Wiki Activity

support for variadic functions - fixes #5

Browse Source
This commit is contained in:
Mat Ryer 2016-10-06 11:44:32 +01:00
parent 92c01adec9
commit edc3a126a0
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
package/moq/moq.go
View File

@ -95,8 +95,8 @@ func (m *Mocker) Mock(w io.Writer, name ...string) error {
Name: meth.Name(), Name: meth.Name(),
} }
obj.Methods = append(obj.Methods, method) obj.Methods = append(obj.Methods, method)
method.Params = m.extractArgs(sig.Params(), "in%d") method.Params = m.extractArgs(sig, sig.Params(), "in%d")
method.Returns = m.extractArgs(sig.Results(), "out%d") method.Returns = m.extractArgs(sig, sig.Results(), "out%d")
} }
objs = append(objs, obj) objs = append(objs, obj)
} }
@ -121,15 +121,20 @@ func (m *Mocker) packageQualifier(pkg *types.Package) string {
return pkg.Name() return pkg.Name()
} }
func (m *Mocker) extractArgs(list *types.Tuple, nameFormat string) []*param { func (m *Mocker) extractArgs(sig *types.Signature, list *types.Tuple, nameFormat string) []*param {
var params []*param var params []*param
for ii := 0; ii < list.Len(); ii++ { listLen := list.Len()
for ii := 0; ii < listLen; ii++ {
p := list.At(ii) p := list.At(ii)
name := p.Name() name := p.Name()
if name == "" { if name == "" {
name = fmt.Sprintf(nameFormat, ii+1) name = fmt.Sprintf(nameFormat, ii+1)
} }
typename := types.TypeString(p.Type(), m.packageQualifier) typename := types.TypeString(p.Type(), m.packageQualifier)
// check for final variadic argument
if sig.Variadic() && ii == listLen-1 && typename[0:2] == "[]" {
typename = "..." + typename[2:]
}
param := &param{ param := &param{
Name: name, Name: name,
Type: typename, Type: typename,