Move enable increment to beginning of function

Azure · Jul 3, 2023 · 9010e01 · 9010e01
1 parent bf4cf31
commit 9010e01
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests_e2e/tests/agent_ext_workflow/extension_workflow.py b/tests_e2e/tests/agent_ext_workflow/extension_workflow.py
@@ -82,14 +82,15 @@ def __init__(self, extension: VirtualMachineExtensionClient, ssh_client: SshClie
             self.data = None
 
         def modify_ext_settings_and_enable(self, data=None):
+            self.enable_count += 1
+
             # Settings follows the following format: [ExtensionName]-[Version], Count: [Enable-count]
             setting_name = "%s-%s, %s: %s" % (self.name, self.version, self.COUNT_KEY_NAME, self.enable_count)
-            # We include data in the settings to test the special characters case. The settings format with data
-            # follows the following format: [ExtensionName]-[Version], Count: [Enable-count], data: [data]
+            # We include data in the settings to test the special characters case. The settings with data follows the
+            # following format: [ExtensionName]-[Version], Count: [Enable-count], data: [data]
             if data is not None:
                 setting_name = "{0}, {1}: {2}".format(setting_name, self.DATA_KEY_NAME, data)
 
-            self.enable_count += 1
             self.expected_message = setting_name
             settings = {self.NAME_KEY_NAME: setting_name.encode('utf-8')}