Fixes build

This commit is contained in:
Ask Solem 2015-12-18 14:52:33 -08:00
parent 4d15ea542e
commit 85a82cc62e
1 changed files with 4 additions and 4 deletions

View File

@ -66,7 +66,7 @@ class test_Message(Case):
ack = self.channel.basic_ack = Mock() ack = self.channel.basic_ack = Mock()
self.message.ack() self.message.ack()
ack.assert_called_with(self.message.delivery_tag) ack.assert_called_with(self.message.delivery_tag, multiple=False)
def test_ack_not_no_ack(self): def test_ack_not_no_ack(self):
self.channel.no_ack_consumers = set() self.channel.no_ack_consumers = set()
@ -74,19 +74,19 @@ class test_Message(Case):
ack = self.channel.basic_ack = Mock() ack = self.channel.basic_ack = Mock()
self.message.ack() self.message.ack()
ack.assert_called_with(self.message.delivery_tag) ack.assert_called_with(self.message.delivery_tag, multiple=False)
def test_ack_log_error_when_no_error(self): def test_ack_log_error_when_no_error(self):
ack = self.message.ack = Mock() ack = self.message.ack = Mock()
self.message.ack_log_error(Mock(), KeyError) self.message.ack_log_error(Mock(), KeyError)
ack.assert_called_with() ack.assert_called_with(multiple=False)
def test_ack_log_error_when_error(self): def test_ack_log_error_when_error(self):
ack = self.message.ack = Mock() ack = self.message.ack = Mock()
ack.side_effect = KeyError('foo') ack.side_effect = KeyError('foo')
logger = Mock() logger = Mock()
self.message.ack_log_error(logger, KeyError) self.message.ack_log_error(logger, KeyError)
ack.assert_called_with() ack.assert_called_with(multiple=False)
self.assertTrue(logger.critical.called) self.assertTrue(logger.critical.called)
self.assertIn("Couldn't ack", logger.critical.call_args[0][0]) self.assertIn("Couldn't ack", logger.critical.call_args[0][0])