mirror of
https://github.com/prowler-cloud/prowler.git
synced 2026-10-31 07:44:21 +00:00
e3c4368d32
Co-authored-by: Hugo P.Brito <hugopbrit@gmail.com> Co-authored-by: Hugo Pereira Brito <101209179+HugoPBrito@users.noreply.github.com>
57 lines
2.0 KiB
Python
57 lines
2.0 KiB
Python
from azure.identity import AzureAuthorityHosts
|
|
|
|
from prowler.providers.azure.lib.regions.regions import (
|
|
AZURE_CHINA_CLOUD,
|
|
AZURE_CHINA_GRAPH_HOST,
|
|
AZURE_CHINA_GRAPH_SCOPE,
|
|
AZURE_CHINA_LOGS_ENDPOINT,
|
|
AZURE_GENERIC_CLOUD,
|
|
AZURE_GENERIC_GRAPH_HOST,
|
|
AZURE_GENERIC_GRAPH_SCOPE,
|
|
AZURE_GENERIC_LOGS_ENDPOINT,
|
|
AZURE_US_GOV_CLOUD,
|
|
AZURE_US_GOV_GRAPH_HOST,
|
|
AZURE_US_GOV_GRAPH_SCOPE,
|
|
AZURE_US_GOV_LOGS_ENDPOINT,
|
|
get_regions_config,
|
|
)
|
|
|
|
|
|
class Test_azure_regions:
|
|
def test_get_regions_config(self):
|
|
allowed_regions = [
|
|
"AzureCloud",
|
|
"AzureChinaCloud",
|
|
"AzureUSGovernment",
|
|
]
|
|
expected_output = {
|
|
"AzureCloud": {
|
|
"authority": None,
|
|
"base_url": AZURE_GENERIC_CLOUD,
|
|
"credential_scopes": [AZURE_GENERIC_CLOUD + "/.default"],
|
|
"graph_host": AZURE_GENERIC_GRAPH_HOST,
|
|
"graph_scope": AZURE_GENERIC_GRAPH_SCOPE,
|
|
"logs_endpoint": AZURE_GENERIC_LOGS_ENDPOINT,
|
|
},
|
|
"AzureChinaCloud": {
|
|
"authority": AzureAuthorityHosts.AZURE_CHINA,
|
|
"base_url": AZURE_CHINA_CLOUD,
|
|
"credential_scopes": [AZURE_CHINA_CLOUD + "/.default"],
|
|
"graph_host": AZURE_CHINA_GRAPH_HOST,
|
|
"graph_scope": AZURE_CHINA_GRAPH_SCOPE,
|
|
"logs_endpoint": AZURE_CHINA_LOGS_ENDPOINT,
|
|
},
|
|
"AzureUSGovernment": {
|
|
"authority": AzureAuthorityHosts.AZURE_GOVERNMENT,
|
|
"base_url": AZURE_US_GOV_CLOUD,
|
|
"credential_scopes": [AZURE_US_GOV_CLOUD + "/.default"],
|
|
"graph_host": AZURE_US_GOV_GRAPH_HOST,
|
|
"graph_scope": AZURE_US_GOV_GRAPH_SCOPE,
|
|
"logs_endpoint": AZURE_US_GOV_LOGS_ENDPOINT,
|
|
},
|
|
}
|
|
|
|
for region in allowed_regions:
|
|
region_config = get_regions_config(region)
|
|
assert region_config == expected_output[region]
|