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()
