def test_kv_patch(self, _1):
with patch.dict(os.environ,
{
constants.ENV_TENANT_ID: 'tenant',
constants.ENV_SUB_ID: DEFAULT_SUBSCRIPTION_ID,
constants.ENV_KEYVAULT_CLIENT_ID: 'kv_client',
constants.ENV_KEYVAULT_SECRET_ID: 'kv_secret'
}, clear=True):
auth_params = Session().get_credentials().auth_params
self.assertEqual(auth_params.get('tenant_id'), 'tenant')
self.assertEqual(auth_params.get('subscription_id'), DEFAULT_SUBSCRIPTION_ID)
self.assertEqual(auth_params.get('keyvault_client_id'), 'kv_client')
self.assertEqual(auth_params.get('keyvault_secret_id'), 'kv_secret')
self.assertEqual(auth_params.get('client_id'), 'client')
self.assertEqual(auth_params.get('client_secret'), 'secret')
def test_kv_patch(self, _1):
reload(sys.modules['c7n_azure.session'])
with patch.dict(os.environ,
{
constants.ENV_TENANT_ID: 'tenant',
constants.ENV_SUB_ID: DEFAULT_SUBSCRIPTION_ID,
constants.ENV_KEYVAULT_CLIENT_ID: 'kv_client',
constants.ENV_KEYVAULT_SECRET_ID: 'kv_secret'
}, clear=True):
with patch('azure.common.credentials.ServicePrincipalCredentials.__init__',
autospec=True, return_value=None):
auth_params = Session().auth_params
self.assertEqual(auth_params.get('tenant_id'), 'tenant')
self.assertEqual(auth_params.get('subscription_id'), DEFAULT_SUBSCRIPTION_ID)
self.assertEqual(auth_params.get('keyvault_client_id'), 'kv_client')
self.assertEqual(auth_params.get('keyvault_secret_id'), 'kv_secret')
self.assertEqual(auth_params.get('client_id'), 'client')
self.assertEqual(auth_params.get('client_secret'), 'secret')
