add Fleet API 2025-03-01 (#8589)

* add Fleet 2025-03-01 api

* add api

* add serialization

* revert changes, remove old api version

* add next api

* add changes

* fix linters

* remove wait to remove get

* add name parameter alias

* redo test recordings

* redo test with discover

* redo recording

* fix recording

* add comment

* remove comment

* fix recording

* remove unused parameter

* add pylint disable

* fix trailing whitespace

---------

Co-authored-by: ftran <[email protected]>

Author: frantran

src/fleet/HISTORY.rst

@@ -115,3 +115,7 @@ Release History
 1.4.0
 ++++++
 * Set autoupgradeprofile commands to preview mode
+
+1.5.0
+++++++
+* Upgrade SDK version to 2025-03-01

src/fleet/azext_fleet/__init__.py

@@ -15,7 +15,7 @@ def register_fleet_resource_type():
     register_resource_type(
         "latest",
         CUSTOM_MGMT_FLEET,
-        SDKProfile("2024-05-02-preview"),
+        SDKProfile("2025-03-01"),
     )
 
 

src/fleet/azext_fleet/_client_factory.py

@@ -40,3 +40,7 @@ def get_resource_groups_client(cli_ctx, subscription_id=None):
 
 def cf_auto_upgrade_profiles(cli_ctx, *_):
     return get_container_service_client(cli_ctx).auto_upgrade_profiles
+
+
+def cf_auto_upgrade_profile_operations(cli_ctx, *_):
+    return get_container_service_client(cli_ctx).auto_upgrade_profile_operations

src/fleet/azext_fleet/_help.py

@@ -380,6 +380,14 @@
           text: az fleet autoupgradeprofile list -g MyFleetResourceGroup -f MyFleetName
 """
 
+helps['fleet autoupgradeprofile generate-update-run'] = """
+    type: command
+    short-summary: Generates an update run for a given auto upgrade profile.
+    examples:
+        - name: Generates an update run for a given auto upgrade profile.
+          text: az fleet autoupgradeprofile generate-update-run -g MyFleetResourceGroup -f MyFleetName --auto-upgrade-profile-name MyAutoUpgradeProfileName
+"""
+
 helps['fleet autoupgradeprofile wait'] = """
     type: command
     short-summary: Wait for an auto upgrade resource to reach a desired state.

src/fleet/azext_fleet/_params.py

@@ -110,3 +110,8 @@ def load_arguments(self, _):
     with self.argument_context('fleet autoupgradeprofile wait') as c:
         c.argument('auto_upgrade_profile_name', options_list=['--auto-upgrade-profile-name', '--profile-name'],
                    help='Specify name for the auto upgrade profile.')
+
+    with self.argument_context('fleet autoupgradeprofile generate-update-run') as c:
+        c.argument('resource_group_name', options_list=['--resource-group', '-g'], help='Name of the resource group.')
+        c.argument('fleet_name', options_list=['--fleet-name', '-f'], help='Name of the fleet.')
+        c.argument('auto_upgrade_profile_name', options_list=['--auto-upgrade-profile-name', '--profile-name', '--name', '-n'], help='Name of the auto upgrade profile.')

src/fleet/azext_fleet/commands.py

@@ -10,7 +10,9 @@
     cf_fleet_members,
     cf_update_runs,
     cf_fleet_update_strategies,
-    cf_auto_upgrade_profiles
+    cf_auto_upgrade_profiles,
+    cf_auto_upgrade_profile_operations
+
 )
 
 
@@ -46,6 +48,12 @@ def load_command_table(self, _):
         client_factory=cf_auto_upgrade_profiles
     )
 
+    auto_upgrade_profile_operations_sdk = CliCommandType(
+        operations_tmpl="azext_fleet.vendored_sdks.operations._auto_upgrade_profile_operations_operations#AutoUpgradeProfileOperationsOperations.{}",
+        operation_group="auto_upgrade_profile_operations",
+        client_factory=cf_auto_upgrade_profile_operations
+    )
+
     # fleets command group
     with self.command_group("fleet", fleets_sdk, client_factory=cf_fleets) as g:
         g.custom_command("create", "create_fleet", supports_no_wait=True)
@@ -87,9 +95,13 @@ def load_command_table(self, _):
         g.wait_command("wait")
 
     # auto upgrade profiles command group
-    with self.command_group("fleet autoupgradeprofile", auto_upgrade_profiles_sdk, client_factory=cf_auto_upgrade_profiles, is_preview=True) as g:
+    with self.command_group("fleet autoupgradeprofile", auto_upgrade_profiles_sdk, client_factory=cf_auto_upgrade_profiles) as g:
         g.custom_command("create", "create_auto_upgrade_profile", supports_no_wait=True)
         g.custom_show_command("show", "show_auto_upgrade_profile")
         g.custom_command("list", "list_auto_upgrade_profiles")
         g.custom_command("delete", "delete_auto_upgrade_profile", supports_no_wait=True, confirmation=True)
         g.wait_command("wait")
+
+    # auto upgrade profiles operation command group
+    with self.command_group("fleet autoupgradeprofile", auto_upgrade_profile_operations_sdk, client_factory=cf_auto_upgrade_profile_operations) as g:
+        g.custom_command("generate-update-run", "generate_update_run", supports_no_wait=True)

src/fleet/azext_fleet/custom.py

@@ -603,3 +603,18 @@ def delete_auto_upgrade_profile(cmd,  # pylint: disable=unused-argument
                                 name,
                                 no_wait=False):
     return sdk_no_wait(no_wait, client.begin_delete, resource_group_name, fleet_name, name)
+
+
+def generate_update_run(cmd,  # pylint: disable=unused-argument
+                        client,
+                        resource_group_name,
+                        fleet_name,
+                        auto_upgrade_profile_name,
+                        no_wait=False):
+    return sdk_no_wait(
+        no_wait,
+        client.begin_generate_update_run,
+        resource_group_name,
+        fleet_name,
+        auto_upgrade_profile_name
+    )

src/fleet/azext_fleet/tests/latest/recordings/test_fleet_hubful.yaml

@@ -41,7 +41,6 @@ def generate_ssh_keys(cls):
     @AllowLargeResponse(size_kb=9999)
     @ResourceGroupPreparer(name_prefix='cli-', random_name_length=8)
     def test_fleet_hubful(self):
-
         self.kwargs.update({
             'fleet_name': self.create_random_name(prefix='fl-', length=7),
             'member_name': self.create_random_name(prefix='flmc-', length=9),