class _QueueTest(_ServiceTest):
    queue_name = "perfstress-" + str(uuid.uuid4())
    queue_client = None
    async_queue_client = None

    def __init__(self, arguments):
        super().__init__(arguments)
        connection_string = self.get_from_env(
            "AZURE_SERVICEBUS_CONNECTION_STRING")
        connection_props = parse_connection_string(connection_string)
        self.mgmt_client = ServiceBusService(
            service_namespace=connection_props['namespace'],
            shared_access_key_name=connection_props['shared_access_key_name'],
            shared_access_key_value=connection_props['shared_access_key'])

    async def global_setup(self):
        await super().global_setup()
        queue = Queue(max_size_in_megabytes=MAX_QUEUE_SIZE)
        self.mgmt_client.create_queue(self.queue_name, queue=queue)

    async def setup(self):
        await super().setup()
        # In T1, these operations check for the existance of the queue
        # so must be created during setup, rather than in the constructor.
        self.queue_client = self.service_client.get_queue(self.queue_name)
        self.async_queue_client = self.async_service_client.get_queue(
            self.queue_name)

    async def global_cleanup(self):
        self.mgmt_client.delete_queue(self.queue_name)
        await super().global_cleanup()
Example #2
0
def create_standard_queue(sb_config):
    from azure.servicebus.control_client import ServiceBusService, Queue
    queue_name = str(uuid.uuid4())
    queue_value = Queue(lock_duration='PT30S',
                        requires_duplicate_detection=False,
                        dead_lettering_on_message_expiration=True,
                        requires_session=False)
    client = ServiceBusService(service_namespace=sb_config['hostname'],
                               shared_access_key_name=sb_config['key_name'],
                               shared_access_key_value=sb_config['access_key'])
    if client.create_queue(queue_name, queue=queue_value, fail_on_exist=True):
        return queue_name
    raise ValueError("Queue creation failed.")
def create_standard_queue(sb_config):
    from azure.servicebus.control_client import ServiceBusService, Queue
    queue_name = str(uuid.uuid4())
    queue_value = Queue(
        lock_duration='PT30S',
        requires_duplicate_detection=False,
        dead_lettering_on_message_expiration=True,
        requires_session=False)
    client = ServiceBusService(
        service_namespace=sb_config['hostname'],
        shared_access_key_name=sb_config['key_name'],
        shared_access_key_value=sb_config['access_key'])
    if client.create_queue(queue_name, queue=queue_value, fail_on_exist=True):
        return queue_name
    raise ValueError("Queue creation failed.")