b754f1384f
Adds all the new config items to the INI file and adjusts some naming conventions so that these values sort together. Also adds defaults for some values where they're known.
114 lines
2.4 KiB
Python
114 lines
2.4 KiB
Python
import pytest
|
|
from unittest.mock import Mock
|
|
|
|
from atst.domain.csp.cloud import AzureCloudProvider
|
|
|
|
AZURE_CONFIG = {
|
|
"AZURE_CLIENT_ID": "MOCK",
|
|
"AZURE_SECRET_KEY": "MOCK",
|
|
"AZURE_TENANT_ID": "MOCK",
|
|
"AZURE_POLICY_LOCATION": "policies",
|
|
"AZURE_VAULT_URL": "http://vault",
|
|
"AZURE_POWERSHELL_CLIENT_ID": "MOCK",
|
|
"AZURE_ROLE_DEF_ID_OWNER": "MOCK",
|
|
"AZURE_ROLE_DEF_ID_CONTRIBUTOR": "MOCK",
|
|
"AZURE_ROLE_DEF_ID_BILLING_READER": "MOCK",
|
|
"AZURE_GRAPH_RESOURCE": "MOCK",
|
|
"AZURE_AADP_QTY": 5,
|
|
}
|
|
|
|
AUTH_CREDENTIALS = {
|
|
"client_id": AZURE_CONFIG["AZURE_CLIENT_ID"],
|
|
"secret_key": AZURE_CONFIG["AZURE_SECRET_KEY"],
|
|
"tenant_id": AZURE_CONFIG["AZURE_TENANT_ID"],
|
|
}
|
|
|
|
|
|
def mock_subscription():
|
|
from azure.mgmt import subscription
|
|
|
|
return Mock(spec=subscription)
|
|
|
|
|
|
def mock_authorization():
|
|
from azure.mgmt import authorization
|
|
|
|
return Mock(spec=authorization)
|
|
|
|
|
|
def mock_managementgroups():
|
|
from azure.mgmt import managementgroups
|
|
|
|
return Mock(spec=managementgroups)
|
|
|
|
|
|
def mock_graphrbac():
|
|
import azure.graphrbac as graphrbac
|
|
|
|
return Mock(spec=graphrbac)
|
|
|
|
|
|
def mock_credentials():
|
|
import azure.common.credentials as credentials
|
|
|
|
return Mock(spec=credentials)
|
|
|
|
|
|
def mock_identity():
|
|
import azure.identity as identity
|
|
|
|
return Mock(spec=identity)
|
|
|
|
|
|
def mock_policy():
|
|
from azure.mgmt.resource import policy
|
|
|
|
return Mock(spec=policy)
|
|
|
|
|
|
def mock_adal():
|
|
import adal
|
|
|
|
return Mock(spec=adal)
|
|
|
|
|
|
def mock_requests():
|
|
import requests
|
|
|
|
return Mock(spec=requests)
|
|
|
|
|
|
def mock_secrets():
|
|
from azure.keyvault import secrets
|
|
|
|
return Mock(spec=secrets)
|
|
|
|
|
|
def mock_cloud_details():
|
|
from msrestazure.azure_cloud import AZURE_PUBLIC_CLOUD
|
|
|
|
return AZURE_PUBLIC_CLOUD
|
|
|
|
|
|
class MockAzureSDK(object):
|
|
def __init__(self):
|
|
|
|
self.subscription = mock_subscription()
|
|
self.authorization = mock_authorization()
|
|
self.policy = mock_policy()
|
|
self.adal = mock_adal()
|
|
self.managementgroups = mock_managementgroups()
|
|
self.graphrbac = mock_graphrbac()
|
|
self.credentials = mock_credentials()
|
|
self.identity = mock_identity()
|
|
self.policy = mock_policy()
|
|
self.secrets = mock_secrets()
|
|
self.requests = mock_requests()
|
|
self.cloud = mock_cloud_details()
|
|
self.identity = mock_identity()
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def mock_azure():
|
|
return AzureCloudProvider(AZURE_CONFIG, azure_sdk_provider=MockAzureSDK())
|