update mock

This commit is contained in:
Asim Aslam 2018-05-10 17:39:13 +01:00
parent c3c0543733
commit b17a802675

View File

@ -49,8 +49,8 @@ func (m *MockClient) Options() client.Options {
return m.Opts return m.Opts
} }
func (m *MockClient) NewMessage(topic string, msg interface{}) client.Message { func (m *MockClient) NewMessage(topic string, msg interface{}, opts ...client.MessageOption) client.Message {
return m.Client.NewMessage(topic, msg) return m.Client.NewMessage(topic, msg, opts...)
} }
func (m *MockClient) NewRequest(service, method string, req interface{}, reqOpts ...client.RequestOption) client.Request { func (m *MockClient) NewRequest(service, method string, req interface{}, reqOpts ...client.RequestOption) client.Request {