Skip to content

Commit 1c3ffe2

Browse files
xiangyan99allenkim0129
authored andcommitted
Add test for get_arm_endpoints (Azure#38196)
1 parent 6603ca3 commit 1c3ffe2

File tree

2 files changed

+46
-1
lines changed

2 files changed

+46
-1
lines changed

sdk/core/azure-mgmt-core/azure/mgmt/core/tools.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@
4747
"resource_id",
4848
"is_valid_resource_id",
4949
"is_valid_resource_name",
50+
"get_arm_endpoints",
5051
]
5152

5253

@@ -230,7 +231,7 @@ def get_arm_endpoints(cloud_setting: AzureClouds) -> Dict[str, Any]:
230231
"""
231232
if cloud_setting == AzureClouds.AZURE_CHINA_CLOUD:
232233
return {
233-
"resource_manager": "https://management.chinacloudapi.cn",
234+
"resource_manager": "https://management.chinacloudapi.cn/",
234235
"credential_scopes": ["https://management.chinacloudapi.cn/.default"],
235236
}
236237
if cloud_setting == AzureClouds.AZURE_US_GOVERNMENT:
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
1+
# --------------------------------------------------------------------------
2+
#
3+
# Copyright (c) Microsoft Corporation. All rights reserved.
4+
#
5+
# The MIT License (MIT)
6+
#
7+
# Permission is hereby granted, free of charge, to any person obtaining a copy
8+
# of this software and associated documentation files (the ""Software""), to deal
9+
# in the Software without restriction, including without limitation the rights
10+
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11+
# copies of the Software, and to permit persons to whom the Software is
12+
# furnished to do so, subject to the following conditions:
13+
#
14+
# The above copyright notice and this permission notice shall be included in
15+
# all copies or substantial portions of the Software.
16+
#
17+
# THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18+
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19+
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
20+
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21+
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22+
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
23+
# THE SOFTWARE.
24+
#
25+
# --------------------------------------------------------------------------
26+
from azure.core import AzureClouds
27+
from azure.mgmt.core.tools import get_arm_endpoints
28+
29+
30+
def test_arm_endpoints():
31+
cloud_setting = AzureClouds.AZURE_PUBLIC_CLOUD
32+
cloud_meta = get_arm_endpoints(cloud_setting)
33+
assert cloud_meta["resource_manager"] == "https://management.azure.com/"
34+
assert cloud_meta["credential_scopes"] == ["https://management.azure.com/.default"]
35+
36+
cloud_setting = AzureClouds.AZURE_US_GOVERNMENT
37+
cloud_meta = get_arm_endpoints(cloud_setting)
38+
assert cloud_meta["resource_manager"] == "https://management.usgovcloudapi.net/"
39+
assert cloud_meta["credential_scopes"] == ["https://management.core.usgovcloudapi.net/.default"]
40+
41+
cloud_setting = AzureClouds.AZURE_CHINA_CLOUD
42+
cloud_meta = get_arm_endpoints(cloud_setting)
43+
assert cloud_meta["resource_manager"] == "https://management.chinacloudapi.cn/"
44+
assert cloud_meta["credential_scopes"] == ["https://management.chinacloudapi.cn/.default"]

0 commit comments

Comments
 (0)