{RDBMS} CLI code updates for new GA Api version for PostgreSQL using new SDK package (#24997)

* Update cli code for the new sdk package and update tests by rerecording

* Revert change to test-rdbms_flexible_commands.py file for retention days and keep it as 90

* Update Python SDK package to recently released version 10.2.0b6

* Fix byok test and add test for aad only auth

* Update help text for aad auth related commands for creating or updating PG flexible server

* remove unittest unused import from test_rdbms_flexible_commands file

* Record skipped tests, refactor for new aad test and fix small bug for failover (restart) to check for samezone ha

* Fix name and text for the new sample commands added for aad, also update the record test for backups as the corresponding RP changes are deployed

* Fix aad test for MySQL

* Fix the failing test after merge

Author: ambrahma

src/azure-cli/azure/cli/command_modules/rdbms/_helptext_pg.py

@@ -126,6 +126,20 @@
 
       az postgres flexible-server create -g testGroup -n testServer --location testLocation \\
         --key $keyIdentifier --identity testIdentity
+  - name: >
+      Create a PostgreSQL flexible server with active directory auth as well as password auth.
+    text: >
+      # create flexible server with aad auth and password auth enabled
+
+      az postgres flexible-server create -g testGroup -n testServer --location testLocation \\
+        --active-directory-auth Enabled
+  - name: >
+      Create a PostgreSQL flexible server with active directory auth only.
+    text: >
+      # create flexible server with aad only auth and password auth disabled
+
+      az postgres flexible-server create -g testGroup -n testServer --location testLocation \\
+        --active-directory-auth Enabled --password-auth Disabled
 """
 
 helps['postgres flexible-server show'] = """
@@ -158,6 +172,8 @@
     text: az postgres flexible-server update --resource-group testGroup --name testserver --tags "k1=v1" "k2=v2"
   - name: Reset password
     text: az postgres flexible-server update --resource-group testGroup --name testserver -p password123
+  - name: Update a flexible server to enable active directory auth for password auth enabled server
+    text: az postgres flexible-server update --resource-group testGroup --name testserver --active-directory-auth Enabled
   - name: Change key/identity for data encryption. Data encryption cannot be enabled post server creation, this will only update the key/identity.
     text: >
       # get key identifier of the existing key

src/azure-cli/azure/cli/command_modules/rdbms/flexible_server_custom_postgres.py

@@ -108,8 +108,8 @@ def flexible_server_create(cmd, client,
                                                                    byok_identity=byok_identity,
                                                                    byok_key=byok_key)
 
-    auth_config = postgresql_flexibleservers.models.AuthConfig(active_directory_auth_enabled=(active_directory_auth == 'Enabled'),
-                                                               password_auth_enabled=(password_auth == 'Enabled'))
+    auth_config = postgresql_flexibleservers.models.AuthConfig(active_directory_auth=active_directory_auth,
+                                                               password_auth=password_auth)
 
     # Create postgresql
     # Note : passing public_access has no effect as the accepted values are 'Enabled' and 'Disabled'. So the value ends up being ignored.
@@ -305,9 +305,9 @@ def flexible_server_update_custom_func(cmd, client, instance,
 
     auth_config = instance.auth_config
     if active_directory_auth:
-        auth_config.active_directory_auth_enabled = active_directory_auth == 'Enabled'
+        auth_config.active_directory_auth = active_directory_auth
     if password_auth:
-        auth_config.password_auth_enabled = password_auth == 'Enabled'
+        auth_config.password_auth = password_auth
 
     params = ServerForUpdate(sku=instance.sku,
                              storage=instance.storage,
@@ -336,8 +336,8 @@ def flexible_server_update_custom_func(cmd, client, instance,
 
 def flexible_server_restart(cmd, client, resource_group_name, server_name, fail_over=None):
     instance = client.get(resource_group_name, server_name)
-    if fail_over is not None and instance.high_availability.mode != "ZoneRedundant":
-        raise ArgumentUsageError("Failing over can only be triggered for zone redundant servers.")
+    if fail_over is not None and instance.high_availability.mode not in ("ZoneRedundant", "SameZone"):
+        raise ArgumentUsageError("Failing over can only be triggered for zone redundant or same zone servers.")
 
     if fail_over is not None:
         if fail_over.lower() not in ['planned', 'forced']: