def test_twisted_connection_timeout(self):
with self.assertRaises(exceptions.AMQPConnectionError) as err_ctx:
with mock.patch(
'pika.TwistedConnection._create_tcp_connection_socket',
return_value=mock.Mock(
spec_set=socket.socket,
connect=mock.Mock(
side_effect=mock_timeout))) as create_sock_mock:
params = pika.ConnectionParameters(socket_timeout=2.0)
conn = twisted_connection.TwistedConnection(params)
conn._on_connect_timer()
create_sock_mock.return_value.settimeout.assert_called_with(2.0)
self.assertIn('timeout', str(err_ctx.exception))
def test_twisted_connection_timeout(self, connect, settimeout):
connect.side_effect = mock_timeout
with self.assertRaises(exceptions.AMQPConnectionError):
params = pika.ConnectionParameters(socket_timeout=2.0)
twisted_connection.TwistedConnection(params)
settimeout.assert_called_with(2.0)
routing_key='burger.irc.out')
@defer.inlineCallbacks
def start_outbound_queue(self, callback):
qo = yield self.channel.basic_consume(
consumer_callback=callback, queue=self.outbound_queue.method.queue)
l = task.LoopingCall(callback, qo)
l.start(0.01)
def on_error(error, er):
print 'error!', error, er
def on_close():
print 'close!'
if __name__ == "__main__":
amqp = AMQP()
factory = IRCFactory(CONFIG, amqp)
params = pika.ConnectionParameters(CONFIG.amqp_server, CONFIG.amqp_port)
conn = twisted_connection.TwistedConnection(
parameters=params,
on_open_callback=amqp.setup,
on_open_error_callback=on_error,
on_close_callback=on_close)
reactor.connectTCP(CONFIG.irc_server, CONFIG.irc_port, factory)
reactor.run()
