def setUp(self): super(AzureEventSubscriptionsTest, self).setUp() account = self.setup_account() queue_name = 'cctesteventsub' StorageUtilities.create_queue_from_storage_account(account, queue_name) event_sub_destination = StorageQueueEventSubscriptionDestination( resource_id=account.id, queue_name=queue_name) AzureEventSubscription.create(event_sub_destination, self.event_sub_name)
def setUp(self): super(AzureEventSubscriptionsTest, self).setUp() account = self.setup_account() queue_name = 'cctesteventsub' StorageUtilities.create_queue_from_storage_account(account, queue_name) event_sub_destination = StorageQueueEventSubscriptionDestination( resource_id=account.id, queue_name=queue_name) AzureEventSubscription.create(event_sub_destination, self.event_sub_name)
def test_create_azure_event_subscription(self): account = self.setup_account() queue_name = 'cctestevensub' StorageUtilities.create_queue_from_storage_account(account, queue_name) sub_destination = StorageQueueEventSubscriptionDestination(resource_id=account.id, queue_name=queue_name) sub_name = 'custodiantestsubscription' event_subscription = AzureEventSubscription.create(sub_destination, sub_name) self.assertEqual(event_subscription.name, sub_name) self.assertEqual(event_subscription.destination.endpoint_type, 'StorageQueue')
def test_create_azure_event_subscription(self): account = self.setup_account() queue_name = 'cctestevensub' StorageUtilities.create_queue_from_storage_account(account, queue_name, self.session) sub_destination = StorageQueueEventSubscriptionDestination(resource_id=account.id, queue_name=queue_name) sub_name = 'custodiantestsubscription' event_subscription = AzureEventSubscription.create(sub_destination, sub_name) self.assertEqual(event_subscription.name, sub_name) self.assertEqual(event_subscription.destination.endpoint_type, 'StorageQueue')
def _create_storage_queue(self, queue_name, session): self.log.info("Creating storage queue") storage_client = session.client('azure.mgmt.storage.StorageManagementClient') storage_account = storage_client.storage_accounts.get_properties( self.function_params.storage_account['resource_group_name'], self.function_params.storage_account['name']) try: StorageUtilities.create_queue_from_storage_account(storage_account, queue_name, session) self.log.info("Storage queue creation succeeded") return storage_account except Exception as e: self.log.error('Queue creation failed with error: %s' % e) raise SystemExit
def _create_storage_queue(self, queue_name, session): self.log.info("Creating storage queue") storage_client = session.client('azure.mgmt.storage.StorageManagementClient') storage_account = storage_client.storage_accounts.get_properties( self.function_params.storage_account['resource_group_name'], self.function_params.storage_account['name']) try: StorageUtilities.create_queue_from_storage_account(storage_account, queue_name, session) self.log.info("Storage queue creation succeeded") return storage_account except Exception as e: self.log.error('Queue creation failed with error: %s' % e) raise SystemExit
def prepare_queue_storage(self, queue_resource_id, queue_name): """ Create a storage client using unusual ID/group reference as this is what we require for event subscriptions """ storage_client = self.session.client( 'azure.mgmt.storage.StorageManagementClient') account = storage_client.storage_accounts.get_properties( ResourceIdParser.get_resource_group(queue_resource_id), ResourceIdParser.get_resource_name(queue_resource_id)) Storage.create_queue_from_storage_account(account, queue_name, self.session) return account
def test_create_delete_queue_from_storage_account(self): account = self.setup_account() queue_name = 'testqueuecc' queue = \ StorageUtilities.create_queue_from_storage_account(account, queue_name, self.session) self.assertTrue(queue) result = \ StorageUtilities.delete_queue_from_storage_account(account, queue_name, self.session) self.assertTrue(result)
def prepare_queue_storage(self, queue_resource_id, queue_name): """ Create a storage client using unusual ID/group reference as this is what we require for event subscriptions """ # Use a different session object if the queue is in a different subscription queue_subscription_id = ResourceIdParser.get_subscription_id( queue_resource_id) if queue_subscription_id != self.session.subscription_id: session = Session(queue_subscription_id) else: session = self.session storage_client = session.client( 'azure.mgmt.storage.StorageManagementClient') account = storage_client.storage_accounts.get_properties( ResourceIdParser.get_resource_group(queue_resource_id), ResourceIdParser.get_resource_name(queue_resource_id)) Storage.create_queue_from_storage_account(account, queue_name, self.session) return account
def test_create_queue_from_storage_account(self): account = self.setup_account() queue_name = 'testqueuecc' queue = \ StorageUtilities.create_queue_from_storage_account(account, queue_name, self.session) self.assertTrue(queue)
def test_create_queue_from_storage_account(self): account = self.setup_account() queue_name = 'testqueuecc' queue = StorageUtilities.create_queue_from_storage_account( account, queue_name) self.assertTrue(queue)