feat: Added Azure location to facts

src/cloud_what/providers/azure.py

@@ -39,7 +39,7 @@ class AzureCloudProvider(BaseCloudProvider):
     # for very long time. It would be good to update the version from time to time,
     # because old versions (three years) are deprecated. It would be good to update
     # the API version with every minor version of RHEL
-    API_VERSION = "2021-02-01"
+    API_VERSION = "2023-07-01"
 
     BASE_CLOUD_PROVIDER_METADATA_URL = "http://169.254.169.254/metadata/instance?api-version="
 

src/rhsmlib/facts/cloud_facts.py

@@ -118,12 +118,13 @@ def get_aws_facts(self) -> Dict[str, Union[str, None]]:
 
     def get_azure_facts(self) -> Dict[str, str]:
         """
-        Try to get facts of VM running on Azure public cloud. Returned dictionary has following format:
+        Try to get facts of VM running on Azure public cloud. Returned dictionary has the following format:
             {
                 "azure_instance_id": some_instance_ID,
                 "azure_offer": some_offer,
                 "azure_sku": some_sku,
                 "azure_subscription_id": some_subscription_ID
+                "azure_location: azure region the VM is running in
             }
         :return: dictionary containing Azure facts, when the machine is able to gather metadata
             from Azure cloud provider; otherwise returns empty dictionary {}
@@ -143,6 +144,13 @@ def get_azure_facts(self) -> Dict[str, str]:
                     facts["azure_offer"] = values["compute"]["offer"]
                 if "subscriptionId" in values["compute"]:
                     facts["azure_subscription_id"] = values["compute"]["subscriptionId"]
+                if "location" in values["compute"]:
+                    facts["azure_location"] = values["compute"]["location"]
+                if "extendedLocation" in values["compute"]:
+                    if "name" in values["compute"]["extendedLocation"]:
+                        facts["azure_extended_location_name"] = values["compute"]["extendedLocation"]["name"]
+                    if "type" in values["compute"]["extendedLocation"]:
+                        facts["azure_extended_location_type"] = values["compute"]["extendedLocation"]["type"]
         return facts
 
     def get_gcp_facts(self) -> Dict[str, str]:

test/rhsmlib/facts/test_cloud_facts.py

@@ -48,6 +48,10 @@
 {
     "compute": {
         "azEnvironment": "AzurePublicCloud",
+        "extendedLocation": {
+            "type": "edgeZone",
+            "name": "microsoftlondon"
+        },
         "customData": "",
         "location": "westeurope",
         "name": "foo-bar",
@@ -169,6 +173,12 @@
 AZURE_SKU = "8.1-ci"
 AZURE_OFFER = "RHEL"
 AZURE_SUBSCRIPTION_ID = "01234567-0123-0123-0123-012345679abc"
+# There is no list of valid values of locations, extended locations and
+# types of extended locations. Value "microsoftlondon" probably does not
+# exist at all, and it was added only for testing purpose.
+AZURE_LOCATION = "westeurope"
+AZURE_EXTENDED_LOCATION_NAME = "microsoftlondon"
+AZURE_EXTENDED_LOCATION_TYPE = "edgeZone"
 
 
 def mock_prepare_request(request):
@@ -309,13 +319,19 @@ def test_get_azure_facts(self):
         # azure_instance_id should be included in the facts
         self.assertIn("azure_instance_id", facts)
         self.assertEqual(facts["azure_instance_id"], AZURE_INSTANCE_ID)
-        # some other azure facts should be included in facts too
+        # some other azure facts could be included in facts too
         self.assertIn("azure_sku", facts)
         self.assertEqual(facts["azure_sku"], AZURE_SKU)
         self.assertIn("azure_offer", facts)
         self.assertEqual(facts["azure_offer"], AZURE_OFFER)
         self.assertIn("azure_subscription_id", facts)
         self.assertEqual(facts["azure_subscription_id"], AZURE_SUBSCRIPTION_ID)
+        self.assertIn("azure_location", facts)
+        self.assertEqual(facts["azure_location"], AZURE_LOCATION)
+        self.assertIn("azure_extended_location_name", facts)
+        self.assertEqual(facts["azure_extended_location_name"], AZURE_EXTENDED_LOCATION_NAME)
+        self.assertIn("azure_extended_location_type", facts)
+        self.assertEqual(facts["azure_extended_location_type"], AZURE_EXTENDED_LOCATION_TYPE)
 
     @patch("cloud_what.providers.gcp.GCPCloudProvider._write_token_to_cache_file")
     @patch("cloud_what.providers.gcp.GCPCloudProvider._get_metadata_from_cache")