예제 #1
0
def test_get_connection(rabbit_config):
    amqp_uri = rabbit_config['AMQP_URI']
    connection_ids = []

    with get_connection(amqp_uri) as connection:
        connection_ids.append(id(connection))
        assert isinstance(connection, Connection)

    with get_connection(amqp_uri) as connection:
        connection_ids.append(id(connection))
        assert len(set(connection_ids)) == 1
예제 #2
0
def test_get_connection(rabbit_config):
    amqp_uri = rabbit_config['AMQP_URI']
    connection_ids = []

    with get_connection(amqp_uri) as connection:
        connection_ids.append(id(connection))
        assert isinstance(connection, Connection)

    with get_connection(amqp_uri) as connection:
        connection_ids.append(id(connection))
        assert len(set(connection_ids)) == 1
예제 #3
0
    def setup(self):

        verify_amqp_uri(self.amqp_uri)

        with get_connection(self.amqp_uri) as conn:
            for entity in self.declare:
                maybe_declare(entity, conn)

        serializer = self.options.pop('serializer', self.serializer)

        self.publisher = self.publisher_cls(self.amqp_uri,
                                            serializer=serializer,
                                            exchange=self.exchange,
                                            declare=self.declare,
                                            **self.options)
예제 #4
0
    def setup(self):

        ssl = self.container.config.get(AMQP_SSL_CONFIG_KEY)

        with get_connection(self.amqp_uri, ssl) as conn:
            for entity in self.declare:
                maybe_declare(entity, conn.channel())

        serializer = self.options.pop('serializer', self.serializer)

        self.publisher = self.publisher_cls(self.amqp_uri,
                                            serializer=serializer,
                                            exchange=self.exchange,
                                            declare=self.declare,
                                            ssl=ssl,
                                            **self.options)
예제 #5
0
    def __init__(self, exchange=None, declare=None, **publisher_options):
        super(SchedulerPublisher, self).__init__(exchange=exchange,
                                                 declare=declare,
                                                 **publisher_options)
        default_ssl = config.get(AMQP_SSL_CONFIG_KEY)
        ssl = self.publisher_options.pop("ssl", default_ssl)

        with get_connection(self.amqp_uri, ssl) as conn:
            for entity in self.declare:
                maybe_declare(entity, conn.channel())

        serializer = self.publisher_options.pop("serializer", "json")
        self.publisher = self.publisher_cls(self.amqp_uri,
                                            ssl=ssl,
                                            serializer=serializer,
                                            exchange=self.exchange,
                                            declare=self.declare,
                                            **self.publisher_options)
예제 #6
0
    def setup(self):

        ssl = self.container.config.get(AMQP_SSL_CONFIG_KEY)

        with get_connection(self.amqp_uri, ssl) as conn:
            for entity in self.declare:
                maybe_declare(entity, conn.channel())

        serializer = self.options.pop('serializer', self.serializer)

        self.publisher = self.publisher_cls(
            self.amqp_uri,
            serializer=serializer,
            exchange=self.exchange,
            declare=self.declare,
            ssl=ssl,
            **self.options
        )