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

Merge pull request #17 from matryer/remove-enabled

Browse Source 
Remove enabled
This commit is contained in:
Mat Ryer 2017-07-07 20:09:14 +01:00 committed by GitHub
commit be1d7ca86c
1 changed files with 13 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
package/moq/moq.go
View File

@ -264,13 +264,8 @@ type {{.InterfaceName}}Mock struct {
{{.Name}}Func func({{ .Arglist }}) {{.ReturnArglist}} {{.Name}}Func func({{ .Arglist }}) {{.ReturnArglist}}
{{ end }} {{ end }}
// CallsTo tracks calls to the methods. // CallsTo tracks calls to the methods.
CallsTo struct { CallsTo struct {
// Enabled indicates that calls will be tracked. {{- range .Methods }}
//
// // don't track calls
// {{.InterfaceName}}Mock.CallsTo.Enabled = false
Enabled bool
{{ range .Methods }}
lock{{.Name}} sync.Mutex // protects {{ .Name }} lock{{.Name}} sync.Mutex // protects {{ .Name }}
// {{ .Name }} holds details about calls to the {{.Name}} method. // {{ .Name }} holds details about calls to the {{.Name}} method.
{{ .Name }} []struct { {{ .Name }} []struct {
@ -288,19 +283,17 @@ func (mock *{{$obj.InterfaceName}}Mock) {{.Name}}({{.Arglist}}) {{.ReturnArglist
if mock.{{.Name}}Func == nil { if mock.{{.Name}}Func == nil {
panic("moq: {{$obj.InterfaceName}}Mock.{{.Name}}Func is nil but was just called") panic("moq: {{$obj.InterfaceName}}Mock.{{.Name}}Func is nil but was just called")
} }
if mock.CallsTo.Enabled { mock.CallsTo.lock{{.Name}}.Lock()
mock.CallsTo.lock{{.Name}}.Lock() mock.CallsTo.{{.Name}} = append(mock.CallsTo.{{.Name}}, struct{
mock.CallsTo.{{.Name}} = append(mock.CallsTo.{{.Name}}, struct{ {{- range .Params }}
{{- range .Params }} {{ .Name | Exported }} {{ .Type }}
{{ .Name | Exported }} {{ .Type }} {{- end }}
{{- end }} }{
}{ {{- range .Params }}
{{- range .Params }} {{ .Name | Exported }}: {{ .Name }},
{{ .Name | Exported }}: {{ .Name }}, {{- end }}
{{- end }} })
}) mock.CallsTo.lock{{.Name}}.Unlock()
mock.CallsTo.lock{{.Name}}.Unlock()
}
{{- if .ReturnArglist }} {{- if .ReturnArglist }}
return mock.{{.Name}}Func({{.ArgCallList}}) return mock.{{.Name}}Func({{.ArgCallList}})
{{- else }} {{- else }}