unistack-org/micro
3
0
Fork 3
Code Issues 14 Pull Requests 2 Actions 1 Packages Projects Releases 242 Wiki Activity

send requestBody to mock function if it can handle it

Browse Source
This commit is contained in:
Hunyadvári Péter 2019-05-02 17:14:41 +02:00
parent e4704a8f41
commit c1c0a8fb30
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
client/mock/mock.go
View File

@ -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))