fix china, usgov endpoints

Azure · Nov 17, 2021 · f4dcc41 · f4dcc41
1 parent 34864bf
commit f4dcc41
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/stormcollector/stormcollector/auth.py b/stormcollector/stormcollector/auth.py
@@ -24,14 +24,14 @@
         "MGMT": "https://management.core.cloudapi.de",
     },
     "CHINA": {
-        "AD": identity.KnownAuthorities.AZURE_PUBLIC_CLOUD,
+        "AD": identity.KnownAuthorities.AZURE_CHINA,
         "AAD": "https://graph.chinacloudapi.cn",
         "ARM": "https://management.chinacloudapi.cn",
         "GRAPH": "https://microsoftgraph.chinacloudapi.cn",
         "MGMT": "https://management.core.chinacloudapi.cn",
     },
     "USGOV": {
-        "AD": identity.KnownAuthorities.AZURE_PUBLIC_CLOUD,
+        "AD": identity.KnownAuthorities.AZURE_GOVERNMENT,
         "AAD": "https://graph.windows.net",
         "ARM": "https://management.usgovcloudapi.net",
         "GRAPH": "https://graph.microsoft.us/",
@@ -42,7 +42,10 @@
 
 class Context:
     def __init__(
-        self, args: Namespace, cloud: dict, authenticatedCreds: Tuple[Any],
+        self,
+        args: Namespace,
+        cloud: dict,
+        authenticatedCreds: Tuple[Any],
     ):
         self.args = args
         self.cloud = cloud