add msgid to mocks

2 files changed, 9 insertions, 5 deletions

diff --git a/eventmq/tests/test_client_jobs.py b/eventmq/tests/test_client_jobs.py
index c658187..c66875a 100644
--- a/eventmq/tests/test_client_jobs.py
+++ b/eventmq/tests/test_client_jobs.py
@@ -34,13 +34,13 @@ class TestCase(unittest.TestCase):
         self.assertEqual(d(), 12)
         self.assertEqual(d.delay(), '43e14eaa-2034-4c84-8fe7-5577c70b6a7c')
         defer_job_mock.assert_called_with(Sender_mock(), test_func, args=(),
-                                          kwargs={}, queue=None)
+                                          msgid=None, kwargs={}, queue=None)
 
         d = jobs.job(test_func, broker_addr=self.BROKER_ADDR, queue='mojo')
         d.delay(1, 2, three=3)
         defer_job_mock.assert_called_with(Sender_mock(), test_func,
                                           args=(1, 2), kwargs={'three': 3},
-                                          queue='mojo')
+                                          msgid=None, queue='mojo')
 
     @mock.patch('eventmq.client.messages.defer_job')
     @mock.patch('eventmq.client.jobs.Sender')
@@ -85,7 +85,8 @@ class TestCase(unittest.TestCase):
         sched_mock.assert_called_with(
             Sender_mock(), test_func, args=(), class_args=(),
             class_kwargs=None, cron=None, headers=('guarantee',),
-            interval_secs=None, kwargs=None, queue='default', unschedule=True)
+            interval_secs=None, kwargs=None, queue='default',
+            unschedule=True)
 
 
 def test_func():
diff --git a/eventmq/tests/test_client_messages.py b/eventmq/tests/test_client_messages.py
index 68b847e..8939e9b 100644
--- a/eventmq/tests/test_client_messages.py
+++ b/eventmq/tests/test_client_messages.py
@@ -86,6 +86,7 @@ class TestCase(unittest.TestCase):
                                        guarantee=False,
                                        retry_count=3,
                                        timeout=0,
+                                       msgid=None,
                                        queue='test_queue')
 
         with LogCapture() as log_checker:
@@ -259,7 +260,8 @@ class TestCase(unittest.TestCase):
             socket, 'REQUEST',
             ('mozo',
              'reply-requested,retry-count:2',
-             messages.serialize(msg)))
+             messages.serialize(msg)),
+            msgid=None)
 
     @mock.patch('eventmq.client.messages.send_emqp_message')
     def test_send_request_all_headers(self, snd_empq_msg_mock):
@@ -282,7 +284,8 @@ class TestCase(unittest.TestCase):
             socket, 'REQUEST',
             ('default',
              'reply-requested,guarantee,retry-count:2,timeout:3',
-             messages.serialize(msg)))
+             messages.serialize(msg)),
+            msgid=None)
 
     @mock.patch('eventmq.client.messages.send_emqp_message')
     def test_send_schedule_request(self, snd_empq_msg_mock):