commit
be1d7ca86c
@ -265,12 +265,7 @@ type {{.InterfaceName}}Mock struct {
|
||||
{{ end }}
|
||||
// CallsTo tracks calls to the methods.
|
||||
CallsTo struct {
|
||||
// Enabled indicates that calls will be tracked.
|
||||
//
|
||||
// // don't track calls
|
||||
// {{.InterfaceName}}Mock.CallsTo.Enabled = false
|
||||
Enabled bool
|
||||
{{ range .Methods }}
|
||||
{{- range .Methods }}
|
||||
lock{{.Name}} sync.Mutex // protects {{ .Name }}
|
||||
// {{ .Name }} holds details about calls to the {{.Name}} method.
|
||||
{{ .Name }} []struct {
|
||||
@ -288,7 +283,6 @@ func (mock *{{$obj.InterfaceName}}Mock) {{.Name}}({{.Arglist}}) {{.ReturnArglist
|
||||
if mock.{{.Name}}Func == nil {
|
||||
panic("moq: {{$obj.InterfaceName}}Mock.{{.Name}}Func is nil but was just called")
|
||||
}
|
||||
if mock.CallsTo.Enabled {
|
||||
mock.CallsTo.lock{{.Name}}.Lock()
|
||||
mock.CallsTo.{{.Name}} = append(mock.CallsTo.{{.Name}}, struct{
|
||||
{{- range .Params }}
|
||||
@ -300,7 +294,6 @@ func (mock *{{$obj.InterfaceName}}Mock) {{.Name}}({{.Arglist}}) {{.ReturnArglist
|
||||
{{- end }}
|
||||
})
|
||||
mock.CallsTo.lock{{.Name}}.Unlock()
|
||||
}
|
||||
{{- if .ReturnArglist }}
|
||||
return mock.{{.Name}}Func({{.ArgCallList}})
|
||||
{{- else }}
|
||||
|
Loading…
Reference in New Issue
Block a user