def test_bad_user(uris):
amqp_uri = uris["bad_user"]
with pytest.raises(IOError) as exc_info:
verify_amqp_uri(amqp_uri)
message = str(exc_info.value)
assert "Error connecting to broker" in message
assert "invalid credentials" in message
def register_provider(self, provider):
self.provider = provider
amqp_uri = provider.container.config[AMQP_URI_CONFIG_KEY]
verify_amqp_uri(amqp_uri)
self.connection = Connection(amqp_uri)
self.queue = provider.queue
self._setup_consumer()
def test_bad_vhost(uris):
amqp_uri = uris["bad_vhost"]
with pytest.raises(IOError) as exc_info:
verify_amqp_uri(amqp_uri)
message = str(exc_info.value)
assert "Error connecting to broker" in message
assert "invalid or unauthorized vhost" in message
def test_ssl_to_wrong_port(rabbit_config, rabbit_ssl_config):
amqp_ssl_uri = rabbit_config['AMQP_URI'] # not ssl uri
with pytest.raises(IOError) as exc_info:
verify_amqp_uri(amqp_ssl_uri, ssl=rabbit_ssl_config['AMQP_SSL'])
message = str(exc_info.value)
assert 'unknown protocol' in message
def test_bad_vhost(rabbit_config):
scheme, auth, host, port, path, _, _ = parse_url(rabbit_config['AMQP_URI'])
amqp_uri = Url(scheme, auth, host, port, '/unknown').url
with pytest.raises(IOError) as exc_info:
verify_amqp_uri(amqp_uri)
message = str(exc_info.value)
assert 'Error connecting to broker' in message
assert 'invalid or unauthorized vhost' in message
def test_bad_user(rabbit_config):
scheme, auth, host, port, path, _, _ = parse_url(rabbit_config['AMQP_URI'])
amqp_uri = Url(scheme, 'invalid:invalid', host, port, path).url
with pytest.raises(IOError) as exc_info:
verify_amqp_uri(amqp_uri)
message = str(exc_info.value)
assert 'Error connecting to broker' in message
assert 'invalid credentials' in message
def register_provider(self, provider):
self.provider = provider
amqp_uri = provider.container.config[AMQP_URI_CONFIG_KEY]
verify_amqp_uri(amqp_uri)
self.connection = Connection(amqp_uri)
self.queue = provider.queue
self._setup_queue()
message_iterator = self._poll_messages()
message_iterator.send(None) # start generator
self.get_message = message_iterator.send
def test_ssl_missing_option(rabbit_ssl_config):
amqp_ssl_uri = rabbit_ssl_config['AMQP_URI']
amqp_ssl_config = rabbit_ssl_config['AMQP_SSL']
del amqp_ssl_config['keyfile']
with pytest.raises(IOError) as exc_info:
verify_amqp_uri(amqp_ssl_uri, ssl=amqp_ssl_config)
message = str(exc_info.value)
# without the private key we'll get a PEM lib error trying to connect
assert 'PEM lib' in message
def register_provider(self, provider):
self.provider = provider
self.serializer = provider.container.config.get(SERIALIZER_CONFIG_KEY, DEFAULT_SERIALIZER)
self.accept = [self.serializer]
amqp_uri = provider.container.config[AMQP_URI_CONFIG_KEY]
verify_amqp_uri(amqp_uri)
self.connection = Connection(amqp_uri)
self.queue = provider.queue
self._setup_consumer()
def register_provider(self, provider):
self.provider = provider
self.serializer, self.accept = serialization.setup(
provider.container.config)
amqp_uri = provider.container.config[AMQP_URI_CONFIG_KEY]
ssl = provider.container.config.get(AMQP_SSL_CONFIG_KEY)
verify_amqp_uri(amqp_uri, ssl=ssl)
self.connection = Connection(amqp_uri, ssl=ssl)
self.queue = provider.queue
self._setup_consumer()
self.stopped = False
def setup(self):
self.amqp_uri = self.container.config[AMQP_URI_CONFIG_KEY]
exchange = self.exchange
queue = self.queue
verify_amqp_uri(self.amqp_uri)
with self.get_connection() as conn:
if queue is not None:
maybe_declare(queue, conn)
elif exchange is not None:
maybe_declare(exchange, conn)
def setup(self):
self.amqp_uri = self.container.config[AMQP_URI_CONFIG_KEY]
self.serializer = self.container.config.get(SERIALIZER_CONFIG_KEY,
DEFAULT_SERIALIZER)
exchange = self.exchange
queue = self.queue
verify_amqp_uri(self.amqp_uri)
with self.get_connection() as conn:
if queue is not None:
maybe_declare(queue, conn)
elif exchange is not None:
maybe_declare(exchange, conn)
def setup(self):
self.amqp_uri = self.container.config[AMQP_URI_CONFIG_KEY]
self.serializer = self.container.config.get(
SERIALIZER_CONFIG_KEY, DEFAULT_SERIALIZER
)
exchange = self.exchange
queue = self.queue
verify_amqp_uri(self.amqp_uri)
with self.get_connection() as conn:
if queue is not None:
maybe_declare(queue, conn)
elif exchange is not None:
maybe_declare(exchange, conn)
def test_ssl_non_default_option(rabbit_ssl_config):
amqp_ssl_uri = rabbit_ssl_config['AMQP_URI']
amqp_ssl_config = rabbit_ssl_config['AMQP_SSL']
amqp_ssl_config['ssl_version'] = ssl.PROTOCOL_TLSv1_2
verify_amqp_uri(amqp_ssl_uri, ssl=amqp_ssl_config)
def test_ssl_no_options(rabbit_ssl_config):
amqp_ssl_uri = rabbit_ssl_config['AMQP_URI']
verify_amqp_uri(amqp_ssl_uri, ssl=True)
def test_ssl_default_options(rabbit_ssl_config):
amqp_ssl_uri = rabbit_ssl_config['AMQP_URI']
verify_amqp_uri(amqp_ssl_uri, ssl=rabbit_ssl_config['AMQP_SSL'])
def test_other_error(rabbit_config):
scheme, auth, host, port, path, _, _ = parse_url(rabbit_config['AMQP_URI'])
amqp_uri = Url(scheme, auth, host, port + 1, path).url # closed port
with pytest.raises(socket.error):
verify_amqp_uri(amqp_uri)
def test_good(rabbit_config):
amqp_uri = rabbit_config['AMQP_URI']
verify_amqp_uri(amqp_uri)
def test_other_error(uris):
# other errors bubble
amqp_uri = uris["bad_port"]
with pytest.raises(socket.error):
verify_amqp_uri(amqp_uri)
def test_ssl_no_options(rabbit_ssl_config):
amqp_ssl_uri = rabbit_ssl_config['AMQP_URI']
verify_amqp_uri(amqp_ssl_uri, ssl=True)
def test_other_error(uris):
# other errors bubble
amqp_uri = uris['bad_port']
with pytest.raises(socket.error):
verify_amqp_uri(amqp_uri)
def test_good(uris):
amqp_uri = uris["good"]
verify_amqp_uri(amqp_uri)
def setup(self):
self.amqp_uri = self.container.config[AMQP_URI_CONFIG_KEY]
self.accept = self.container.accept
self.prefetch_count = self.container.max_workers
verify_amqp_uri(self.amqp_uri)
def test_other_error(rabbit_config):
scheme, auth, host, port, path, _, _ = parse_url(rabbit_config['AMQP_URI'])
amqp_uri = Url(scheme, auth, host, port + 1, path).url # closed port
with pytest.raises(socket.error):
verify_amqp_uri(amqp_uri)
def test_ssl_non_default_option(rabbit_ssl_config):
amqp_ssl_uri = rabbit_ssl_config['AMQP_URI']
amqp_ssl_config = rabbit_ssl_config['AMQP_SSL']
amqp_ssl_config['ssl_version'] = ssl.PROTOCOL_TLSv1_2
verify_amqp_uri(amqp_ssl_uri, ssl=amqp_ssl_config)
def test_good(uris):
amqp_uri = uris['good']
verify_amqp_uri(amqp_uri)
def test_ssl_default_options(rabbit_ssl_config):
amqp_ssl_uri = rabbit_ssl_config['AMQP_URI']
verify_amqp_uri(amqp_ssl_uri, ssl=rabbit_ssl_config['AMQP_SSL'])
def test_good(rabbit_config):
amqp_uri = rabbit_config['AMQP_URI']
verify_amqp_uri(amqp_uri)
def setup(self):
self.amqp_uri = self.container.config[AMQP_URI_CONFIG_KEY]
self.accept = self.container.accept
self.prefetch_count = self.container.max_workers
verify_amqp_uri(self.amqp_uri)
def setup(self):
verify_amqp_uri(self.amqp_uri)
def test_ssl_good(uris, rabbit_ssl_config):
ssl_uri = uris['ssl']
verify_amqp_uri(ssl_uri, ssl=rabbit_ssl_config['AMQP_SSL'])
