fix and add tests

2019-05-02 18:02:24 +02:00
parent c1c0a8fb30
commit 08d70c9d0a
2 changed files with 22 additions and 2 deletions
--- a/client/mock/mock.go
+++ b/client/mock/mock.go
@@ -84,7 +84,7 @@ func (m *MockClient) Call(ctx context.Context, req client.Request, rsp interface
 		response := r.Response
 		if t := reflect.TypeOf(r.Response); t.Kind() == reflect.Func {
 			var reqBody []reflect.Value
-			if t.NumIn() == 0 {
+			if t.NumIn() == 1 {
 				reqBody = append(reqBody, reflect.ValueOf(req.Body()))
 			}
 			response = reflect.ValueOf(r.Response).Call(reqBody)[0].Interface()