class SimpleAmqpTestCase(unittest.TestCase): def get_init(self): self._q = SimpleAmqp(pool=['rabbit1', 'rabbit2'], pooltype='L') logging.getLogger('simpleAmqp').setLevel(logging.CRITICAL) self._q._connect = mock.MagicMock() self._q._close = mock.MagicMock() def setUp(self): self.get_init() def tearDown(self): pass def test_send(self): self._q.send('foo','bar') self.assertTrue(self._q._close.called) self.assertEqual(self._q._close.call_count,1) self._q._connect.assert_called_once_with() def test_sendMany(self): self._q.sendMany('foo',['bar','bla']) self.assertTrue(self._q._close.called) self.assertEqual(self._q._close.call_count,1) self._q._connect.assert_called_once_with()
#!/usr/bin/env python import pika import random import logging from simpleamqp import SimpleAmqp # Init logging level logging.getLogger('simpleAmqp').setLevel(logging.CRITICAL) # Add handler for debug logging.getLogger('simpleAmqp').setLevel(logging.INFO) logging.getLogger('simpleAmqp').addHandler(logging.StreamHandler()) q = SimpleAmqp(pool=['rabbit2', 'rabbit2'], pooltype='L') try: q.send('hello',"My message") except: logging.critical("Send message error") try: messages = ["Many message","Many message2"] q.sendMany('hello',messages) except: logging.critical("Send message error")
#!/usr/bin/env python import pika import logging from simpleamqp import SimpleAmqp def mycallback(ch, method, properties, body): print "mycallback %s" % body # Init logging level logging.getLogger('simpleAmqp').setLevel(logging.CRITICAL) # Add handler for debug logging.getLogger('simpleAmqp').setLevel(logging.INFO) logging.getLogger('simpleAmqp').addHandler(logging.StreamHandler()) q = SimpleAmqp(pool=['rabbit1', 'rabbit2'], pooltype='F') q._callback = mycallback try: q.recv('hello') except: logging.critical("Recv message error")
def get_init(self): self._q = SimpleAmqp(pool=['rabbit1', 'rabbit2'], pooltype='L') logging.getLogger('simpleAmqp').setLevel(logging.CRITICAL) self._q._connect = mock.MagicMock() self._q._close = mock.MagicMock()