def test_get_function_target_subscription(self):
with patch(
'azure.common.credentials.ServicePrincipalCredentials.__init__',
autospec=True,
return_value=None):
with patch.dict(os.environ, {
constants.ENV_TENANT_ID: 'tenant',
constants.ENV_SUB_ID: DEFAULT_SUBSCRIPTION_ID,
constants.ENV_CLIENT_ID: 'client',
constants.ENV_CLIENT_SECRET: 'secret'
},
clear=True):
s = Session()
self.assertEqual(s.get_function_target_subscription_name(),
DEFAULT_SUBSCRIPTION_ID)
self.assertEqual(s.get_function_target_subscription_ids(),
[DEFAULT_SUBSCRIPTION_ID])
with patch.dict(
os.environ, {
constants.ENV_TENANT_ID: 'tenant',
constants.ENV_SUB_ID: DEFAULT_SUBSCRIPTION_ID,
constants.ENV_CLIENT_ID: 'client',
constants.ENV_CLIENT_SECRET: 'secret',
constants.ENV_FUNCTION_SUB_ID: CUSTOM_SUBSCRIPTION_ID
},
clear=True):
s = Session()
self.assertEqual(s.get_function_target_subscription_name(),
CUSTOM_SUBSCRIPTION_ID)
self.assertEqual(s.get_function_target_subscription_ids(),
[CUSTOM_SUBSCRIPTION_ID])
with patch.dict(
os.environ, {
constants.ENV_TENANT_ID: 'tenant',
constants.ENV_SUB_ID: DEFAULT_SUBSCRIPTION_ID,
constants.ENV_CLIENT_ID: 'client',
constants.ENV_CLIENT_SECRET: 'secret',
constants.ENV_FUNCTION_MANAGEMENT_GROUP_NAME: 'test'
},
clear=True):
with patch(
'c7n_azure.utils.ManagedGroupHelper.get_subscriptions_list',
return_value=[]):
s = Session()
self.assertEqual(s.get_function_target_subscription_name(),
'test')
self.assertEqual(s.get_function_target_subscription_ids(),
[])
def test_get_function_target_subscription(self):
with patch('azure.common.credentials.ServicePrincipalCredentials.__init__',
autospec=True, return_value=None):
with patch.dict(os.environ,
{
constants.ENV_TENANT_ID: 'tenant',
constants.ENV_SUB_ID: DEFAULT_SUBSCRIPTION_ID,
constants.ENV_CLIENT_ID: 'client',
constants.ENV_CLIENT_SECRET: 'secret'
}, clear=True):
s = Session()
self.assertEqual(s.get_function_target_subscription_name(),
DEFAULT_SUBSCRIPTION_ID)
self.assertEqual(s.get_function_target_subscription_ids(),
[DEFAULT_SUBSCRIPTION_ID])
with patch.dict(os.environ,
{
constants.ENV_TENANT_ID: 'tenant',
constants.ENV_SUB_ID: DEFAULT_SUBSCRIPTION_ID,
constants.ENV_CLIENT_ID: 'client',
constants.ENV_CLIENT_SECRET: 'secret',
constants.ENV_FUNCTION_SUB_ID: CUSTOM_SUBSCRIPTION_ID
}, clear=True):
s = Session()
self.assertEqual(s.get_function_target_subscription_name(),
CUSTOM_SUBSCRIPTION_ID)
self.assertEqual(s.get_function_target_subscription_ids(),
[CUSTOM_SUBSCRIPTION_ID])
with patch.dict(os.environ,
{
constants.ENV_TENANT_ID: 'tenant',
constants.ENV_SUB_ID: DEFAULT_SUBSCRIPTION_ID,
constants.ENV_CLIENT_ID: 'client',
constants.ENV_CLIENT_SECRET: 'secret',
constants.ENV_FUNCTION_MANAGED_GROUP_NAME: 'test'
}, clear=True):
with patch('c7n_azure.utils.ManagedGroupHelper.get_subscriptions_list',
return_value=[]):
s = Session()
self.assertEqual(s.get_function_target_subscription_name(), 'test')
self.assertEqual(s.get_function_target_subscription_ids(), [])
