Add ip service tags update hook

Author: lubaihua33

lisa/sut_orchestrator/azure/hooks.py

@@ -48,6 +48,22 @@ def azure_update_vm_capabilities(
         """
         ...
 
+    @hookspec
+    def azure_update_ip_service_tags(
+        self, subscription_id: str, ip_service_tags: Dict[str, str]
+    ) -> None:
+        """
+        Implement it to update ip service tags in-place before deployment.
+
+        Implementations should mutate the passed-in `ip_service_tags` dict (e.g.
+        add/update keys). Rebinding the parameter won't affect the caller.
+
+        Args:
+            subscription_id: current deployment subscription id.
+            ip_service_tags: ip service tags to apply on public IP.
+        """
+        ...
+
 
 class AzureHookSpecDefaultImpl:
     __error_maps: List[Tuple[str, Pattern[str], Any]] = [

lisa/sut_orchestrator/azure/platform_.py

@@ -1356,7 +1356,16 @@ def _create_deployment_parameters(
             availability_copied_fields,
         )
         if not arm_parameters.ip_service_tags:
-            if self._is_byoip_feature_registered():
+            arm_parameters.ip_service_tags = {}
+            plugin_manager.hook.azure_update_ip_service_tags(
+                subscription_id=self.subscription_id,
+                ip_service_tags=arm_parameters.ip_service_tags,
+            )
+
+            if (
+                not arm_parameters.ip_service_tags
+                and self._is_byoip_feature_registered()
+            ):
                 arm_parameters.ip_service_tags = {"FirstPartyUsage": "/SyntheticLoad"}
 
         arm_parameters.virtual_network_resource_group = (