format

Author: MoChilia

src/azure-cli-core/azure/cli/core/commands/__init__.py

@@ -507,7 +507,6 @@ class AzCliCommandInvoker(CommandInvoker):
     # pylint: disable=too-many-statements,too-many-locals,too-many-branches
     def execute(self, args):
         args_copy = args[:]
-        
         from knack.events import (EVENT_INVOKER_PRE_CMD_TBL_CREATE, EVENT_INVOKER_POST_CMD_TBL_CREATE,
                                   EVENT_INVOKER_CMD_TBL_LOADED, EVENT_INVOKER_PRE_PARSE_ARGS,
                                   EVENT_INVOKER_POST_PARSE_ARGS,
@@ -706,12 +705,12 @@ def _what_if(self, args):
                     index = args.index('--subscription')
                     if index + 1 < len(args):
                         subscription_value = args[index + 1]
-                        subscription_id = subscription_value    
+                        subscription_id = subscription_value
                 else:
                     from azure.cli.core.commands.client_factory import get_subscription_id
                     subscription_id = get_subscription_id(self.cli_ctx)
                     print(f"DEBUG: Using current login subscription ID: {subscription_id}")
-                
+
                 args = ["az"] + args if args[0] != 'az' else args
                 command = " ".join(args)
                 what_if_result = show_what_if(self.cli_ctx, command, subscription_id=subscription_id)
@@ -724,14 +723,17 @@ def _what_if(self, args):
                 # Return the formatted what-if output as the result
                 # Similar to the normal flow in execute() method
                 return CommandResultItem(
-                    what_if_result, 
+                    what_if_result,
                     table_transformer=None,
                     is_query_active=self.data.get('query_active', False),
                     exit_code=0
                 )
-            except Exception as ex:
+            except (CLIError, ValueError, KeyError) as ex:
                 # If what-if service fails, still show an informative message
-                return CommandResultItem(None, exit_code=1, error=CLIError(f'What-if preview failed: {str(ex)}\nNote: This was a preview operation. No actual changes were made.'))
+                return CommandResultItem(None, exit_code=1,
+                                         error=CLIError(f'What-if preview failed: {str(ex)}\n'
+                                                        f'Note: This was a preview operation. '
+                                                        f'No actual changes were made.'))
 
     @staticmethod
     def _extract_parameter_names(args):

src/azure-cli-core/azure/cli/core/what_if.py

@@ -166,6 +166,7 @@ def _create_resource_change(change_data):
 
     return WhatIfOperationResult(changes, potential_changes, [])
 
+
 def show_what_if(cli_ctx, azcli_script: str, subscription_id: str = None, no_pretty_print=False):
     from azure.cli.core.commands.client_factory import get_subscription_id
     from azure.cli.command_modules.resource._formatters import format_what_if_operation_result

src/azure-cli/azure/cli/command_modules/sql/custom.py

@@ -4369,7 +4369,7 @@ def server_create(
         external_admin_principal_type=None,
         external_admin_sid=None,
         external_admin_name=None,
-        what_if=None,
+        what_if=None,  # pylint: disable=unused-argument
         **kwargs):
     '''
     Creates a server.

src/azure-cli/azure/cli/command_modules/util/tests/latest/test_whatif.py

@@ -18,8 +18,8 @@ def setUp(self):
         self.test_script_path = os.path.join(TEST_DIR, 'test_whatif_script.sh')
 
     @patch('azure.cli.core.commands.client_factory.get_subscription_id')
-    @patch('azure.cli.core.what_if.get_auth_headers')
-    @patch('azure.cli.core.what_if.make_what_if_request')
+    @patch('azure.cli.core.what_if._get_auth_headers')
+    @patch('azure.cli.core.what_if._make_what_if_request')
     def test_what_if_command(self, mock_make_request, mock_get_headers, mock_get_subscription_id):
         mock_get_subscription_id.return_value = 'test-subscription-id'
         mock_get_headers.return_value = {'Authorization': 'Bearer test-token'}