1func ({{if .DBMethod}}db *driverDB{{else}}c *driverClient{{end}}) {{ .Name }}({{ .DriverArgs }}) {{ .ReturnArgs }} {
2 expected := &Expected{{ .Name }}{
3{{ .InputVariables }}
4 }
5 if err := {{if .DBMethod}}db.client{{else}}c{{end}}.nextExpectation(expected); err != nil {
6 return {{ .ZeroReturns }}
7 }
8{{- if eq .Name "DB" }}
9expected.ret0.mu.Lock()
10expected.ret0.name = arg0
11expected.ret0.mu.Unlock()
12{{- end }}
13 if expected.callback != nil {
14 return expected.callback({{.CallbackArgs}})
15 }
16 return {{ .ExpectedReturns }}
17}
View as plain text