Commits

Benjamin Smith committed eeaa8df

Fix tests, needed new keyword arg.

Comments (0)

Files changed (2)

tests/test_simple.py

 
     def should_get_message(self):
         self.client.get()
-        assert self.client.stomp.calls('receive_frame', nonblocking=False)
+        assert self.client.stomp.calls('receive_frame', nonblocking=False, callback=None)
 
     def should_get_message_without_blocking(self):
         self.client.get_nowait()
-        assert self.client.stomp.calls('receive_frame', nonblocking=True)
+        assert self.client.stomp.calls('receive_frame', nonblocking=True, callback=None)
 
     def should_not_get_message(self):
         self.client.stomp.receive_frame.return_value = None

tests/test_stomp.py

         assert self.frame.calls('send_frame', send_args.as_string())
 
     def should_register_callback(self):
-        callback_func = lambda message: message 
+        callback_func = lambda message, stomp: 'pass'
         this_frame = self.stomp.receive_frame(callback=callback_func)
         assert self.stomp.frame.calls('get_message')
         assert isinstance(self.stomp._callback, type(callback_func))