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