def test_on_channel_close(self):
ch = BaseChannel()
ch.on_channel_open(Mock())
ch._transport.channel_number = 1
ch.on_channel_close(ch, 0, 'hi')
self.assertIsNone(ch._transport)
def test_on_channel_close(self):
ch = BaseChannel()
ch.on_channel_open(Mock())
ch._transport.channel_number = 1
ch.on_channel_close(ch, 0, 'hi')
self.assertIsNone(ch._transport)
def test_get_channel_id(self):
ch = BaseChannel()
self.assertTrue(ch.get_channel_id() is None)
ch.on_channel_open(Mock())
self.assertEquals(ch.get_channel_id(), ch._transport.channel_number)
def test_get_channel_id(self):
ch = BaseChannel()
self.assertTrue(ch.get_channel_id() is None)
ch.on_channel_open(Mock())
self.assertEquals(ch.get_channel_id(), ch._transport.channel_number)
def test_on_channel_open(self):
ch = BaseChannel()
ac = Mock(pchannel.Channel)
ch.on_channel_open(ac)
ac.add_on_close_callback.assert_called_once_with(ch.on_channel_close)
self.assertEquals(ch._amq_chan, ac)
def test_on_channel_open(self):
ch = BaseChannel()
transport = Mock()
ch.on_channel_open(transport)
transport.add_on_close_callback.assert_called_once_with(ch.on_channel_close)
self.assertEquals(ch._transport, transport)
def test_on_channel_open(self):
ch = BaseChannel()
ac = Mock(pchannel.Channel)
ch.on_channel_open(ac)
ac.add_on_close_callback.assert_called_once_with(ch.on_channel_close)
self.assertEquals(ch._amq_chan, ac)
def test_on_channel_open(self):
ch = BaseChannel()
transport = Mock()
ch.on_channel_open(transport)
transport.add_on_close_callback.assert_called_once_with(
ch.on_channel_close)
self.assertEquals(ch._transport, transport)
def test_close(self):
# without close callback
transport = Mock()
ch = BaseChannel()
ch.on_channel_open(transport)
ch._fsm.current_state = ch.S_ACTIVE
ch.close()
transport.close.assert_called_once_with()
def test_close(self):
# without close callback
transport = Mock()
ch = BaseChannel()
ch.on_channel_open(transport)
ch._fsm.current_state = ch.S_ACTIVE
ch.close()
transport.close.assert_called_once_with()
def test_on_channel_closed_with_error_callback(self):
ch = BaseChannel()
ch.on_channel_open(Mock())
ch._transport.channel_number = 1
closemock = Mock()
ch.set_closed_error_callback(closemock)
ch.on_channel_close(ch, 1, 'hi')
closemock.assert_called_once_with(ch, 1, 'hi')
def test_on_channel_closed_with_error_callback(self):
ch = BaseChannel()
ch.on_channel_open(Mock())
ch._transport.channel_number = 1
closemock = Mock()
ch.set_closed_error_callback(closemock)
ch.on_channel_close(ch, 1, 'hi')
closemock.assert_called_once_with(ch, 1, 'hi')
def test_on_channel_close_with_error_in_error_callback(self, logmock):
ch = BaseChannel()
ch.on_channel_open(Mock())
ch._transport.channel_number = 1
closemock = Mock()
closemock.side_effect = StandardError
ch.set_closed_error_callback(closemock)
ch.on_channel_close(ch, 1, 'hi')
self.assertEquals(logmock.warn.call_count, 1)
def test_on_channel_close_with_error_in_error_callback(self, logmock):
ch = BaseChannel()
ch.on_channel_open(Mock())
ch._transport.channel_number = 1
closemock = Mock()
closemock.side_effect = StandardError
ch.set_closed_error_callback(closemock)
ch.on_channel_close(ch, 1, 'hi')
self.assertEquals(logmock.warn.call_count, 1)
def test_declare_exchange_point(self):
# make sure no xp param results in assertion
ch = BaseChannel()
self.assertRaises(AssertionError, ch._declare_exchange, None)
transport = Mock()
ch.on_channel_open(transport)
ch._declare_exchange('hello')
self.assertTrue(transport.declare_exchange_impl.called)
self.assertIn('hello', transport.declare_exchange_impl.call_args[0])
self.assertIn('exchange_type', transport.declare_exchange_impl.call_args[1])
self.assertIn('durable', transport.declare_exchange_impl.call_args[1])
self.assertIn('auto_delete', transport.declare_exchange_impl.call_args[1])
def test_declare_exchange_point(self):
# make sure no xp param results in assertion
ch = BaseChannel()
self.assertRaises(AssertionError, ch._declare_exchange, None)
transport = Mock()
ch.on_channel_open(transport)
ch._declare_exchange('hello')
self.assertTrue(transport.declare_exchange_impl.called)
self.assertIn('hello', transport.declare_exchange_impl.call_args[0])
self.assertIn('exchange_type',
transport.declare_exchange_impl.call_args[1])
self.assertIn('durable', transport.declare_exchange_impl.call_args[1])
self.assertIn('auto_delete',
transport.declare_exchange_impl.call_args[1])
