From c1c0a8fb30e02579e7a0eaf04ad444a1232ecb33 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Hunyadv=C3=A1ri=20P=C3=A9ter?= Date: Thu, 2 May 2019 17:14:41 +0200 Subject: [PATCH] send requestBody to mock function if it can handle it --- client/mock/mock.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/client/mock/mock.go b/client/mock/mock.go index e9022c71..27317391 100644 --- a/client/mock/mock.go +++ b/client/mock/mock.go @@ -83,7 +83,11 @@ 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 { - response = reflect.ValueOf(r.Response).Call([]reflect.Value{})[0].Interface() + var reqBody []reflect.Value + if t.NumIn() == 0 { + reqBody = append(reqBody, reflect.ValueOf(req.Body())) + } + response = reflect.ValueOf(r.Response).Call(reqBody)[0].Interface() } v.Set(reflect.ValueOf(response))