def _auth_callback(server, resource, scope):
if TestMode.is_playback(self.test_mode):
return ('Bearer', 'fake-token')
credentials = self.settings.get_credentials()
credentials.resource = resource
credentials.set_token()
return credentials.scheme, credentials.__dict__['token']['access_token']
def setUp(self):
self.working_folder = os.path.dirname(__file__)
super(AzureMgmtTestCase, self).setUp()
self.fake_settings = fake_settings
if TestMode.is_playback(self.test_mode):
self.settings = self.fake_settings
else:
import tests.mgmt_settings_real as real_settings
self.settings = real_settings
self.resource_client = self.create_mgmt_client(azure.mgmt.resource.ResourceManagementClient)
# Every test uses a different resource group name calculated from its
# qualified test name.
#
# When running all tests serially, this allows us to delete
# the resource group in teardown without waiting for the delete to
# complete. The next test in line will use a different resource group,
# so it won't have any trouble creating its resource group even if the
# previous test resource group hasn't finished deleting.
#
# When running tests individually, if you try to run the same test
# multiple times in a row, it's possible that the delete in the previous
# teardown hasn't completed yet (because we don't wait), and that
# would make resource group creation fail.
# To avoid that, we also delete the resource group in the
# setup, and we wait for that delete to complete.
self.group_name = self.get_resource_name(self.qualified_test_name.replace(".", "_"))
self.region = "westus"
if not self.is_playback():
self.delete_resource_group(wait_timeout=600)
def _auth_callback(server, resource, scope):
if TestMode.is_playback(self.test_mode):
return ('Bearer', 'fake-token')
credentials = self.settings.get_credentials()
credentials.resource = resource
credentials.set_token()
return credentials.scheme, credentials.__dict__['token']['access_token']
def setUp(self):
self.working_folder = os.path.dirname(__file__)
super(LegacyMgmtTestCase, self).setUp()
self.fake_settings = fake_settings
if TestMode.is_playback(self.test_mode):
self.settings = self.fake_settings
else:
import tests.legacy_mgmt_settings_real as real_settings
self.settings = real_settings
def setUp(self):
self.working_folder = os.path.dirname(__file__)
super(LegacyMgmtTestCase, self).setUp()
self.fake_settings = fake_settings
if TestMode.is_playback(self.test_mode):
self.settings = self.fake_settings
else:
import tests.legacy_mgmt_settings_real as real_settings
self.settings = real_settings
def setUp(self):
self.working_folder = os.path.dirname(__file__)
super(ServiceBusTestCase, self).setUp()
self.fake_settings = fake_settings
if TestMode.is_playback(self.test_mode):
self.settings = self.fake_settings
else:
import tests.servicebus_settings_real as real_settings
self.settings = real_settings
def setUp(self):
self.working_folder = os.path.dirname(__file__)
super(ServiceBusTestCase, self).setUp()
self.fake_settings = fake_settings
if TestMode.is_playback(self.test_mode):
self.settings = self.fake_settings
else:
import tests.servicebus_settings_real as real_settings # pylint: disable=import-error,no-name-in-module
self.settings = real_settings
def setUp(self):
self.working_folder = os.path.dirname(__file__)
super(ServiceBusTestCase, self).setUp()
self.fake_settings = fake_settings
if TestMode.is_playback(self.test_mode):
self.settings = self.fake_settings
else:
import tests.servicebus_settings_real as real_settings # pylint: disable=import-error,no-name-in-module
self.settings = real_settings
def setUp(self):
self.working_folder = os.path.dirname(__file__)
super(ServiceBusTestCase, self).setUp()
self.fake_settings = fake_settings
if TestMode.is_playback(self.test_mode):
self.settings = self.fake_settings
else:
import tests.servicebus_settings_real as real_settings
self.settings = real_settings
def setUpClass(cls):
LOG.warning('Starting Batch tests')
LOG.debug("Setting up Batch tests:")
cls.working_folder = os.path.dirname(__file__)
try:
cls.test_mode = init_test_mode(cls.working_folder)
cls.fake_settings = fake_settings
if TestMode.is_playback(cls.test_mode):
LOG.debug(" running in playback mode")
cls.live = False
cls.settings = cls.fake_settings
else:
LOG.debug(" running in live mode")
import tests.mgmt_settings_real as real_settings
cls.settings = real_settings
cls.live = True
LOG.debug(' creating resource client')
cls.resource_client = create_mgmt_client(cls.settings,
azure.mgmt.resource.resources.ResourceManagementClientConfiguration,
azure.mgmt.resource.resources.ResourceManagementClient
)
LOG.debug(' creating storage client')
cls.storage_client = create_mgmt_client(cls.settings,
azure.mgmt.storage.StorageManagementClientConfiguration,
azure.mgmt.storage.StorageManagementClient
)
LOG.debug(' creating batch client')
cls.batch_mgmt_client = create_mgmt_client(cls.settings,
azure.mgmt.batch.BatchManagementClientConfiguration,
azure.mgmt.batch.BatchManagementClient
)
if cls.live:
LOG.debug(' creating resource group')
create_resource_group(cls.resource_client)
LOG.debug(' creating storage')
create_storage_account(cls.storage_client)
LOG.debug(' creating batch account')
cls.batch_client = create_batch_account(cls.batch_mgmt_client, cls.settings, cls.live)
except Exception as err:
cls.tearDownClass()
raise AssertionError("Failed to setup Batch Account: {}".format(err))
LOG.debug(" finished setup")
return super(BatchMgmtTestCase, cls).setUpClass()
def setUp(self):
self.working_folder = os.path.dirname(__file__)
super(AzureKeyVaultTestCase, self).setUp()
def mock_key_vault_auth_base(self, request):
challenge = HttpBearerChallenge(request.url, 'Bearer authorization=fake-url,resource=https://vault.azure.net')
self.set_authorization_header(request, challenge)
return request
self.fake_settings = fake_settings
if TestMode.is_playback(self.test_mode):
self.settings = self.fake_settings
KeyVaultAuthBase.__call__ = mock_key_vault_auth_base
else:
import tests.mgmt_settings_real as real_settings
self.settings = real_settings
self.client = self.create_keyvault_client()
def setUp(self):
self.working_folder = os.path.dirname(__file__)
super(AzureKeyVaultTestCase, self).setUp()
def mock_key_vault_auth_base(self, request):
challenge = HttpBearerChallenge(request.url, 'Bearer authorization=fake-url,resource=https://vault.azure.net')
self.set_authorization_header(request, challenge)
return request
self.fake_settings = fake_settings
if TestMode.is_playback(self.test_mode):
self.settings = self.fake_settings
azure.keyvault.key_vault_authentication.KeyVaultAuthBase.__call__ = mock_key_vault_auth_base
else:
import tests.mgmt_settings_real as real_settings
self.settings = real_settings
self.client = self.create_keyvault_client()
def setUp(self):
self.working_folder = os.path.dirname(__file__)
super(AzureMgmtTestCase, self).setUp()
self.fake_settings = fake_settings
if TestMode.is_playback(self.test_mode):
self.settings = self.fake_settings
else:
import tests.mgmt_settings_real as real_settings
self.settings = real_settings
self.resource_client = self.create_mgmt_client(
azure.mgmt.resource.resources.ResourceManagementClientConfiguration,
azure.mgmt.resource.resources.ResourceManagementClient
)
# Every test uses a different resource group name calculated from its
# qualified test name.
#
# When running all tests serially, this allows us to delete
# the resource group in teardown without waiting for the delete to
# complete. The next test in line will use a different resource group,
# so it won't have any trouble creating its resource group even if the
# previous test resource group hasn't finished deleting.
#
# When running tests individually, if you try to run the same test
# multiple times in a row, it's possible that the delete in the previous
# teardown hasn't completed yet (because we don't wait), and that
# would make resource group creation fail.
# To avoid that, we also delete the resource group in the
# setup, and we wait for that delete to complete.
self.group_name = self.get_resource_name(
self.qualified_test_name.replace('.', '_')
)
self.region = 'westus'
if not self.is_playback():
self.delete_resource_group(wait_timeout=600)
