def test__on_underlying_close(self, mocklog): tp = AMQPTransport(Mock()) cb = Mock() tp.add_on_close_callback(cb) tp._on_underlying_close(200, sentinel.text) cb.assert_called_once_with(tp, 200, sentinel.text) self.assertEquals(mocklog.debug.call_count, 1) self.assertIn(sentinel.text, mocklog.debug.call_args[0])
def test__on_underlying_close(self, mocklog): tp = AMQPTransport(Mock()) cb = Mock() tp.add_on_close_callback(cb) tp._on_underlying_close(200, sentinel.text) cb.assert_called_once_with(tp, 200, sentinel.text) self.assertEquals(mocklog.debug.call_count, 1) self.assertIn(sentinel.text, mocklog.debug.call_args[0])
def test__on_underlying_close(self): client = Mock() tp = AMQPTransport(client) cb = Mock() tp.add_on_close_callback(cb) tp._on_underlying_close(200, sentinel.text) cb.assert_called_once_with(tp, 200, sentinel.text) self.assertEquals(client.callbacks.remove.call_count, 4) self.assertEquals(client.callbacks.remove.call_args_list, [call(client.channel_number, 'Basic.GetEmpty'), call(client.channel_number, 'Channel.Close'), call(client.channel_number, '_on_basic_deliver'), call(client.channel_number, '_on_basic_get')])
def test__on_underlying_close(self): client = Mock() tp = AMQPTransport(client) cb = Mock() tp.add_on_close_callback(cb) tp._on_underlying_close(200, sentinel.text) cb.assert_called_once_with(tp, 200, sentinel.text) self.assertEquals(client.callbacks.remove.call_count, 4) self.assertEquals(client.callbacks.remove.call_args_list, [ call(client.channel_number, 'Basic.GetEmpty'), call(client.channel_number, 'Channel.Close'), call(client.channel_number, '_on_basic_deliver'), call(client.channel_number, '_on_basic_get') ])
def test_add_on_close_callback(self): tp = AMQPTransport(Mock()) tp.add_on_close_callback(sentinel.one) tp.add_on_close_callback(sentinel.two) self.assertEquals(tp._close_callbacks, [sentinel.one, sentinel.two])
def test_add_on_close_callback(self): tp = AMQPTransport(Mock()) tp.add_on_close_callback(sentinel.one) tp.add_on_close_callback(sentinel.two) self.assertEquals(tp._close_callbacks, [sentinel.one, sentinel.two])