def setUp(self):
super(AMQPDriverTestCase, self).setUp()
self.user = "******"
self.password = "******"
self.credentials = config.Credentials(self.user, self.password)
self.host = "host"
self.conf = config.ConnectionConfig(self.host, self.credentials)
self.driver = driver.AMQPDriver(self.conf)
def test_driver_uses_pika_sync_engine(self):
"""
Tests that driver uses sync amqp engine if async_engine parameter=False
in config object
"""
conf = config.ConnectionConfig(self.host, self.credentials)
drvr = driver.AMQPDriver(conf)
self.assertEqual(drvr._engine, pika_sync)
def test_get_writer_for_sync_engine(self, engine_mock):
"""
Tests that driver uses Writer of sync engine
"""
conf = config.ConnectionConfig(self.host, self.credentials)
drvr = driver.AMQPDriver(conf)
writer = drvr.create_writer()
self.assertEqual(writer, engine_mock())
def test_get_reader_for_sync_engine(self, engine_mock):
"""
Tests that driver uses Reader of sync engine
"""
conf = config.ConnectionConfig(self.host, self.credentials)
drvr = driver.AMQPDriver(conf)
queue = "queue_name"
preprocessor = mock.Mock()
reader = drvr.create_reader(queue, preprocessor)
self.assertEqual(reader, engine_mock())
def test_publish_message_via_existing_reader_sync(self, get_mock):
"""
Tests that message is published via existing reader
"""
conf = config.ConnectionConfig(self.host, self.credentials)
drvr = driver.AMQPDriver(conf)
drvr._reader = mock.MagicMock()
exchange = "exchange_name"
routing_key = "rk"
message = mock.MagicMock()
drvr.publish_message(exchange, routing_key, message)
get_mock().publish_message.assert_called_once_with(
exchange, routing_key, message)
def _get_driver(self):
return driver.AMQPDriver(self._config)
def _get_driver(self):
if not self._config:
raise exceptions.IncorrectAMQPConfig(ampq_url=self._config)
driver = amqp_driver.AMQPDriver(self._config)
return driver
