def test_create_account_without_subscriptions(self, mock_auth_context): mock_auth_context.acquire_token_with_client_credentials.return_value = self.token_entry1 mock_arm_client = mock.MagicMock() mock_arm_client.subscriptions.list.return_value = [] finder = SubscriptionFinder(lambda _, _2: mock_auth_context, None, lambda _: mock_arm_client) storage_mock = {'subscriptions': []} profile = Profile(storage_mock, use_global_creds_cache=False) profile._management_resource_uri = 'https://management.core.windows.net/' # action result = profile.find_subscriptions_on_login(False, '1234', 'my-secret', True, self.tenant_id, allow_no_subscriptions=True, subscription_finder=finder) # assert self.assertTrue(1, len(result)) self.assertEqual(result[0]['id'], self.tenant_id) self.assertEqual(result[0]['state'], 'Enabled') self.assertEqual(result[0]['tenantId'], self.tenant_id) self.assertEqual(result[0]['name'], 'N/A(tenant level account)')
def test_create_account_without_subscriptions_thru_common_tenant(self, mock_auth_context): mock_auth_context.acquire_token.return_value = self.token_entry1 mock_auth_context.acquire_token_with_username_password.return_value = self.token_entry1 tenant_object = mock.MagicMock() tenant_object.id = "foo-bar" tenant_object.tenant_id = self.tenant_id mock_arm_client = mock.MagicMock() mock_arm_client.subscriptions.list.return_value = [] mock_arm_client.tenants.list.return_value = (x for x in [tenant_object]) finder = SubscriptionFinder(lambda _, _2: mock_auth_context, None, lambda _: mock_arm_client) storage_mock = {'subscriptions': []} profile = Profile(storage_mock, use_global_creds_cache=False) profile._management_resource_uri = 'https://management.core.windows.net/' # action result = profile.find_subscriptions_on_login(False, '1234', 'my-secret', False, None, allow_no_subscriptions=True, subscription_finder=finder) # assert self.assertEqual(1, len(result)) self.assertEqual(result[0]['id'], self.tenant_id) self.assertEqual(result[0]['state'], 'Enabled') self.assertEqual(result[0]['tenantId'], self.tenant_id) self.assertEqual(result[0]['name'], 'N/A(tenant level account)')
def test_get_expanded_subscription_info_for_logged_in_service_principal(self, mock_auth_context): mock_auth_context.acquire_token_with_client_credentials.return_value = self.token_entry1 mock_arm_client = mock.MagicMock() mock_arm_client.subscriptions.list.return_value = [self.subscription1] finder = SubscriptionFinder(lambda _, _2: mock_auth_context, None, lambda _: mock_arm_client) storage_mock = {'subscriptions': []} profile = Profile(storage_mock, use_global_creds_cache=False) profile._management_resource_uri = 'https://management.core.windows.net/' profile.find_subscriptions_on_login(False, '1234', 'my-secret', True, self.tenant_id, False, finder) # action extended_info = profile.get_expanded_subscription_info() # assert self.assertEqual(self.id1.split('/')[-1], extended_info['subscriptionId']) self.assertEqual(self.display_name1, extended_info['subscriptionName']) self.assertEqual('1234', extended_info['client']) self.assertEqual('https://login.microsoftonline.com', extended_info['endpoints'].active_directory)