From b17a8026753f876b0914846ff9ee9482013d1b48 Mon Sep 17 00:00:00 2001
From: Asim Aslam <asim@chuhnk.me>
Date: Thu, 10 May 2018 17:39:13 +0100
Subject: [PATCH] update mock

---
 client/mock/mock.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/client/mock/mock.go b/client/mock/mock.go
index c22d84d6..4a1e520e 100644
--- a/client/mock/mock.go
+++ b/client/mock/mock.go
@@ -49,8 +49,8 @@ func (m *MockClient) Options() client.Options {
 	return m.Opts
 }
 
-func (m *MockClient) NewMessage(topic string, msg interface{}) client.Message {
-	return m.Client.NewMessage(topic, msg)
+func (m *MockClient) NewMessage(topic string, msg interface{}, opts ...client.MessageOption) client.Message {
+	return m.Client.NewMessage(topic, msg, opts...)
 }
 
 func (m *MockClient) NewRequest(service, method string, req interface{}, reqOpts ...client.RequestOption) client.Request {