def test_init_rg_client_usgov(cmclient, spcred):
client = MagicMock()
cmclient.return_value = client
cred = MagicMock()
spcred.return_value = cred
init_resource_graph_client(SECRETS_USGOV)
spcred.assert_called_with(client_id=SECRETS_USGOV.get("client_id"),
secret=SECRETS_USGOV.get('client_secret'),
tenant=SECRETS_USGOV.get('tenant_id'),
cloud_environment=__get_cloud_env_by_name(
SECRETS_USGOV.get('azure_cloud')))
cmclient.assert_called_with(
credentials=ANY,
base_url=__get_cloud_env_by_name(
SECRETS_USGOV.get("azure_cloud")).endpoints.resource_manager)
def test_init_client_public(cmclient, spcred):
client = MagicMock()
cmclient.return_value = client
cred = MagicMock()
spcred.return_value = cred
init_client(SECRETS_PUBLIC, CONFIG)
spcred.assert_called_with(client_id=SECRETS_PUBLIC.get("client_id"),
secret=SECRETS_PUBLIC.get('client_secret'),
tenant=SECRETS_PUBLIC.get('tenant_id'),
cloud_environment=__get_cloud_env_by_name(
SECRETS_PUBLIC.get('azure_cloud')))
cmclient.assert_called_with(
credentials=ANY,
subscription_id=CONFIG['azure']['subscription_id'],
base_url=__get_cloud_env_by_name(
SECRETS_PUBLIC.get("azure_cloud")).endpoints.resource_manager)
def test_credential_bad(spcred):
cred = MagicMock()
spcred.return_value = cred
init_client(SECRETS_BAD_CLOUD, CONFIG)
spcred.assert_called_with(client_id=SECRETS_BAD_CLOUD.get("client_id"),
secret=SECRETS_BAD_CLOUD.get('client_secret'),
tenant=SECRETS_BAD_CLOUD.get('tenant_id'),
cloud_environment=__get_cloud_env_by_name(
SECRETS_BAD_CLOUD.get('azure_cloud')))
def test_init_client_public_token_credentials(cmclient, aadcred):
client = MagicMock()
cmclient.return_value = client
cred = MagicMock()
aadcred.return_value = cred
init_client(SECRETS_PUBLIC_TOKEN, CONFIG)
aadcred.assert_called_with(
dict(accessToken=SECRETS_PUBLIC_TOKEN.get('access_token')),
client_id=SECRETS_PUBLIC_TOKEN.get('client_id'))
cmclient.assert_called_with(
credentials=ANY,
subscription_id=CONFIG['azure']['subscription_id'],
base_url=__get_cloud_env_by_name(
SECRETS_PUBLIC.get("azure_cloud")).endpoints.resource_manager)
def test_get_env_by_name_bad3():
parsed_base_url = __get_cloud_env_by_name(5).endpoints.resource_manager
assert parsed_base_url == AZURE_PUBLIC_CLOUD.endpoints.resource_manager
def test_get_env_by_name_bad():
parsed_base_url = __get_cloud_env_by_name(
SECRETS_BAD_CLOUD.get("azure_cloud")).endpoints.resource_manager
assert parsed_base_url == AZURE_PUBLIC_CLOUD.endpoints.resource_manager
def test_get_env_by_name_us():
parsed_base_url = __get_cloud_env_by_name(
SECRETS_USGOV.get("azure_cloud")).endpoints.resource_manager
assert parsed_base_url == AZURE_US_GOV_CLOUD.endpoints.resource_manager
def test_get_env_by_name_germany():
parsed_base_url = __get_cloud_env_by_name(
SECRETS_GERMANY.get("azure_cloud")).endpoints.resource_manager
assert parsed_base_url == AZURE_GERMAN_CLOUD.endpoints.resource_manager
def test_get_env_by_name_china():
parsed_base_url = __get_cloud_env_by_name(
SECRETS_CHINA.get("azure_cloud")).endpoints.resource_manager
assert parsed_base_url == AZURE_CHINA_CLOUD.endpoints.resource_manager
