Azure Update VM Capabilities: Add Location

If the capabilities dictionary is empty for any reason, the hook implementation would have no way to know which region is being queried.

Author: kamalca

lisa/sut_orchestrator/azure/hooks.py

@@ -37,7 +37,7 @@ def azure_update_arm_template(
 
     @hookspec
     def azure_update_vm_capabilities(
-        self, capabilities: Dict[str, AzureCapability]
+        self, location: str, capabilities: Dict[str, AzureCapability]
     ) -> None:
         """
         Implement it to update the vm capabilities.

lisa/sut_orchestrator/azure/platform_.py

@@ -1134,7 +1134,9 @@ def get_location_info(self, location: str, log: Logger) -> AzureLocation:
                     except Exception as e:
                         log.error(f"unknown sku: {sku_obj}")
                         raise e
-            plugin_manager.hook.azure_update_vm_capabilities(capabilities=all_skus)
+            plugin_manager.hook.azure_update_vm_capabilities(
+                location=location, capabilities=all_skus
+            )
             location_data = AzureLocation(location=location, capabilities=all_skus)
             log.debug(f"{location}: saving to disk")
             with open(cached_file_name, "w") as f: