def cleanup_queue(servicebus_config, queue_name): from azure.servicebus.control_client import ServiceBusService client = ServiceBusService( service_namespace=servicebus_config['hostname'], shared_access_key_name=servicebus_config['key_name'], shared_access_key_value=servicebus_config['access_key']) client.delete_queue(queue_name)
def cleanup_queue(servicebus_config, queue_name): from azure.servicebus.control_client import ServiceBusService client = ServiceBusService( service_namespace=servicebus_config['hostname'], shared_access_key_name=servicebus_config['key_name'], shared_access_key_value=servicebus_config['access_key']) client.delete_queue(queue_name)
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()