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_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_good(uris): amqp_uri = uris['good'] verify_amqp_uri(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'])