18 lines
375 B
Go
18 lines
375 B
Go
|
package mock
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/micro/go-micro/client"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Response sets the response methods for a service
|
||
|
|
func Response(service string, response []MockResponse) client.Option {
|
||
|
|
return func(o *client.Options) {
|
||
|
|
r, ok := fromContext(o.Context)
|
||
|
|
if !ok {
|
||
|
|
r = make(map[string][]MockResponse)
|
||
|
|
}
|
||
|
|
r[service] = response
|
||
|
|
o.Context = newContext(o.Context, r)
|
||
|
|
}
|
||
|
|
}
|