def test_node_definition_from_amqp(self):
n1 = NodeDefinition(u'amqp://*****:*****@kota/psa')
self.assertEquals(n1.user, u'ala')
self.assertEquals(n1.password, u'ma')
self.assertEquals(n1.virtual_host, u'psa')
self.assertEquals(n1.host, u'kota')
n1 = NodeDefinition(u'amqp://*****:*****@kota/')
self.assertEquals(n1.virtual_host, u'/')
def test_on_fail(self):
"""Assert that on_fail doesn't fire if the cluster fails to connect"""
q = {'failed': False}
c = Cluster(NodeDefinition(os.environ.get('AMQP_HOST', '127.0.0.1'),
'xguest',
'xguest',
heartbeat=20),
on_fail=lambda: q.update(failed=True))
self.assertRaises(ConnectionDead, c.start)
c.shutdown()
self.assertFalse(q['failed'])
def wait_for_amqp(host_name, wait_timeout, login, password, virtual_host):
start_at = time.time()
wait_println_counter = 0
node_def = NodeDefinition(host=host_name, user=login, password=password, virtual_host=virtual_host)
while time.time() - start_at < wait_timeout:
time_remaining = wait_timeout - (time.time() - start_at)
try:
Cluster([node_def]).start(wait=True, timeout=time_remaining)
except ConnectionDead:
wait_println_counter += 3
if wait_println_counter == 3:
print("Waiting 30 more seconds...")
wait_println_counter = 0
time.sleep(10)
else:
sys.exit(0)
else:
print("Waiting time exceeded, aborting...")
sys.exit(1)
from __future__ import print_function, absolute_import, division
import logging
import os
import time
import unittest
from coolamqp.utils import monotonic
import six
from coolamqp.clustering import Cluster, MessageReceived, NothingMuch
from coolamqp.objects import Message, NodeDefinition, Queue, \
ReceivedMessage, Exchange
NODE = NodeDefinition(os.environ.get('AMQP_HOST', '127.0.0.1'),
'guest',
'guest',
heartbeat=20)
logging.basicConfig(level=logging.DEBUG)
class TestA(unittest.TestCase):
def setUp(self):
self.c = Cluster([NODE])
self.c.start()
def tearDown(self):
self.c.shutdown()
def test_queue_bind(self):
queue = Queue('my-queue')
exchange = Exchange('my-exchange', type='topic')
def test_amqpconnstring_port(self):
node = NodeDefinition('amqp://*****:*****@lol:4123/vhost')
self.assertEqual(node.port, 4123)
def test_node_with_kwargs(self):
node = NodeDefinition(host='127.0.0.1', user='******', password='******')
self.assertEquals(node.virtual_host, '/') # default