update tests

Azure · Aug 19, 2021 · 014067f · 014067f
1 parent 8b98977
commit 014067f
Show file tree

Hide file tree

Showing 52 changed files with 9,832 additions and 126 deletions.
diff --git a/test/scenario-test.ts b/test/scenario-test.ts
@@ -17,7 +17,7 @@ export enum TestMode {
     ExtNoSdk = 'extnosdk',
     ExtDefaultFolder = 'ext_default_folder',
     ExtNoSdkNoFlattenTrack1 = 'extnosdknoflattentrack1',
-    CoreTrack2 = 'coretrack2',
+    CoreTrack1 = 'coretrack2',
 }
 
 describe('ScenarioTest', () => {
@@ -120,13 +120,13 @@ describe('ScenarioTest', () => {
             key = CodeGenConstants.azureCliFolder;
             extraOption[key] = outputDir;
             return extraOption;
-        } else if (testMode === TestMode.CoreTrack2) {
+        } else if (testMode === TestMode.CoreTrack1) {
             let key = CodeGenConstants.targetMode;
             extraOption[key] = TargetMode.Core;
             key = CodeGenConstants.azureCliFolder;
             extraOption[key] = outputDir;
             key = CodeGenConstants.compatibleLevel;
-            extraOption[key] = CompatibleLevel.Track2;
+            extraOption[key] = CompatibleLevel.Track1;
             return extraOption;
         } else if (testMode === TestMode.ExtFlatten) {
             let key = CodeGenConstants.azureCliExtFolder;

diff --git a/...te/output/core_Incremental/src/azure-cli/azure/cli/command_modules/vm/generated/custom.py b/...te/output/core_Incremental/src/azure-cli/azure/cli/command_modules/vm/generated/custom.py
@@ -18,7 +18,7 @@ def vm_virtual_machine_assess_patch(client,
                                     vm_name,
                                     no_wait=False):
     return sdk_no_wait(no_wait,
-                       client.assess_patches,
+                       client.begin_assess_patches,
                        resource_group_name=resource_group_name,
                        vm_name=vm_name)
 
@@ -102,7 +102,7 @@ def vm_virtual_machine_scale_set_vm_extension_create(client,
     if len(extension_parameters['instance_view']) == 0:
         del extension_parameters['instance_view']
     return sdk_no_wait(no_wait,
-                       client.create_or_update,
+                       client.begin_create_or_update,
                        resource_group_name=resource_group_name,
                        vm_scale_set_name=vm_scale_set_name,
                        instance_id=instance_id,

diff --git a/...to/output/coredefault/src/azure-cli/azure/cli/command_modules/kusto/generated/commands.py b/...to/output/coredefault/src/azure-cli/azure/cli/command_modules/kusto/generated/commands.py
@@ -74,7 +74,10 @@ def load_command_table(self, _):
         g.custom_show_command('show', 'kusto_attached_database_configuration_show')
         g.custom_command('create', 'kusto_attached_database_configuration_create', supports_no_wait=True)
         g.generic_update_command(
-            'update', supports_no_wait=True, custom_func_name='kusto_attached_database_configuration_update'
+            'update',
+            supports_no_wait=True,
+            custom_func_name='kusto_attached_database_configuration_update',
+            setter_name='begin_create_or_update',
         )
         g.custom_command(
             'delete', 'kusto_attached_database_configuration_delete', supports_no_wait=True, confirmation=True
@@ -118,7 +121,10 @@ def load_command_table(self, _):
         g.custom_show_command('show', 'kusto_cluster_principal_assignment_show')
         g.custom_command('create', 'kusto_cluster_principal_assignment_create', supports_no_wait=True)
         g.generic_update_command(
-            'update', supports_no_wait=True, custom_func_name='kusto_cluster_principal_assignment_update'
+            'update',
+            supports_no_wait=True,
+            custom_func_name='kusto_cluster_principal_assignment_update',
+            setter_name='begin_create_or_update',
         )
         g.custom_command(
             'delete', 'kusto_cluster_principal_assignment_delete', supports_no_wait=True, confirmation=True
@@ -165,7 +171,10 @@ def load_command_table(self, _):
         g.custom_show_command('show', 'kusto_database_principal_assignment_show')
         g.custom_command('create', 'kusto_database_principal_assignment_create', supports_no_wait=True)
         g.generic_update_command(
-            'update', supports_no_wait=True, custom_func_name='kusto_database_principal_assignment_update'
+            'update',
+            supports_no_wait=True,
+            custom_func_name='kusto_database_principal_assignment_update',
+            setter_name='begin_create_or_update',
         )
         g.custom_command(
             'delete', 'kusto_database_principal_assignment_delete', supports_no_wait=True, confirmation=True

diff --git a/...usto/output/coredefault/src/azure-cli/azure/cli/command_modules/kusto/generated/custom.py b/...usto/output/coredefault/src/azure-cli/azure/cli/command_modules/kusto/generated/custom.py
@@ -83,7 +83,7 @@ def kusto_cluster_create(client,
     if len(parameters['identity']) == 0:
         del parameters['identity']
     return sdk_no_wait(no_wait,
-                       client.create_or_update,
+                       client.begin_create_or_update,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        parameters=parameters)
@@ -143,7 +143,7 @@ def kusto_cluster_update(client,
     if len(parameters['identity']) == 0:
         del parameters['identity']
     return sdk_no_wait(no_wait,
-                       client.update,
+                       client.begin_update,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        parameters=parameters)
@@ -154,7 +154,7 @@ def kusto_cluster_delete(client,
                          cluster_name,
                          no_wait=False):
     return sdk_no_wait(no_wait,
-                       client.delete,
+                       client.begin_delete,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name)
 
@@ -164,11 +164,14 @@ def kusto_cluster_add_language_extension(client,
                                          cluster_name,
                                          value=None,
                                          no_wait=False):
+    language_extensions_to_add = {}
+    if value is not None:
+        language_extensions_to_add['value'] = value
     return sdk_no_wait(no_wait,
-                       client.add_language_extensions,
+                       client.begin_add_language_extensions,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
-                       value=value)
+                       language_extensions_to_add=language_extensions_to_add)
 
 
 def kusto_cluster_detach_follower_database(client,
@@ -177,20 +180,22 @@ def kusto_cluster_detach_follower_database(client,
                                            cluster_resource_id,
                                            attached_database_configuration_name,
                                            no_wait=False):
+    follower_database_to_remove = {}
+    follower_database_to_remove['cluster_resource_id'] = cluster_resource_id
+    follower_database_to_remove['attached_database_configuration_name'] = attached_database_configuration_name
     return sdk_no_wait(no_wait,
-                       client.detach_follower_databases,
+                       client.begin_detach_follower_databases,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
-                       cluster_resource_id=cluster_resource_id,
-                       attached_database_configuration_name=attached_database_configuration_name)
+                       follower_database_to_remove=follower_database_to_remove)
 
 
 def kusto_cluster_diagnose_virtual_network(client,
                                            resource_group_name,
                                            cluster_name,
                                            no_wait=False):
     return sdk_no_wait(no_wait,
-                       client.diagnose_virtual_network,
+                       client.begin_diagnose_virtual_network,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name)
 
@@ -223,19 +228,22 @@ def kusto_cluster_remove_language_extension(client,
                                             cluster_name,
                                             value=None,
                                             no_wait=False):
+    language_extensions_to_remove = {}
+    if value is not None:
+        language_extensions_to_remove['value'] = value
     return sdk_no_wait(no_wait,
-                       client.remove_language_extensions,
+                       client.begin_remove_language_extensions,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
-                       value=value)
+                       language_extensions_to_remove=language_extensions_to_remove)
 
 
 def kusto_cluster_start(client,
                         resource_group_name,
                         cluster_name,
                         no_wait=False):
     return sdk_no_wait(no_wait,
-                       client.start,
+                       client.begin_start,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name)
 
@@ -245,7 +253,7 @@ def kusto_cluster_stop(client,
                        cluster_name,
                        no_wait=False):
     return sdk_no_wait(no_wait,
-                       client.stop,
+                       client.begin_stop,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name)
 
@@ -285,7 +293,7 @@ def kusto_cluster_principal_assignment_create(client,
     if principal_type is not None:
         parameters['principal_type'] = principal_type
     return sdk_no_wait(no_wait,
-                       client.create_or_update,
+                       client.begin_create_or_update,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        principal_assignment_name=principal_assignment_name,
@@ -318,7 +326,7 @@ def kusto_cluster_principal_assignment_delete(client,
                                               principal_assignment_name,
                                               no_wait=False):
     return sdk_no_wait(no_wait,
-                       client.delete,
+                       client.begin_delete,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        principal_assignment_name=principal_assignment_name)
@@ -359,7 +367,7 @@ def kusto_database_create(client,
                        'provided!')
     parameters = all_parameters[0] if len(all_parameters) == 1 else None
     return sdk_no_wait(no_wait,
-                       client.create_or_update,
+                       client.begin_create_or_update,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        database_name=database_name,
@@ -385,7 +393,7 @@ def kusto_database_update(client,
                        'provided!')
     parameters = all_parameters[0] if len(all_parameters) == 1 else None
     return sdk_no_wait(no_wait,
-                       client.update,
+                       client.begin_update,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        database_name=database_name,
@@ -398,7 +406,7 @@ def kusto_database_delete(client,
                           database_name,
                           no_wait=False):
     return sdk_no_wait(no_wait,
-                       client.delete,
+                       client.begin_delete,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        database_name=database_name)
@@ -409,10 +417,13 @@ def kusto_database_add_principal(client,
                                  cluster_name,
                                  database_name,
                                  value=None):
+    database_principals_to_add = {}
+    if value is not None:
+        database_principals_to_add['value'] = value
     return client.add_principals(resource_group_name=resource_group_name,
                                  cluster_name=cluster_name,
                                  database_name=database_name,
-                                 value=value)
+                                 database_principals_to_add=database_principals_to_add)
 
 
 def kusto_database_list_principal(client,
@@ -429,10 +440,13 @@ def kusto_database_remove_principal(client,
                                     cluster_name,
                                     database_name,
                                     value=None):
+    database_principals_to_remove = {}
+    if value is not None:
+        database_principals_to_remove['value'] = value
     return client.remove_principals(resource_group_name=resource_group_name,
                                     cluster_name=cluster_name,
                                     database_name=database_name,
-                                    value=value)
+                                    database_principals_to_remove=database_principals_to_remove)
 
 
 def kusto_database_principal_assignment_list(client,
@@ -475,7 +489,7 @@ def kusto_database_principal_assignment_create(client,
     if principal_type is not None:
         parameters['principal_type'] = principal_type
     return sdk_no_wait(no_wait,
-                       client.create_or_update,
+                       client.begin_create_or_update,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        database_name=database_name,
@@ -511,7 +525,7 @@ def kusto_database_principal_assignment_delete(client,
                                                principal_assignment_name,
                                                no_wait=False):
     return sdk_no_wait(no_wait,
-                       client.delete,
+                       client.begin_delete,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        database_name=database_name,
@@ -553,7 +567,7 @@ def kusto_attached_database_configuration_create(client,
     if default_principals_modification_kind is not None:
         parameters['default_principals_modification_kind'] = default_principals_modification_kind
     return sdk_no_wait(no_wait,
-                       client.create_or_update,
+                       client.begin_create_or_update,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        attached_database_configuration_name=attached_database_configuration_name,
@@ -586,7 +600,7 @@ def kusto_attached_database_configuration_delete(client,
                                                  attached_database_configuration_name,
                                                  no_wait=False):
     return sdk_no_wait(no_wait,
-                       client.delete,
+                       client.begin_delete,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        attached_database_configuration_name=attached_database_configuration_name)
@@ -636,7 +650,7 @@ def kusto_data_connection_create(client,
                        'event_grid_data_connection is provided!')
     parameters = all_parameters[0] if len(all_parameters) == 1 else None
     return sdk_no_wait(no_wait,
-                       client.create_or_update,
+                       client.begin_create_or_update,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        database_name=database_name,
@@ -668,7 +682,7 @@ def kusto_data_connection_update(client,
                        'event_grid_data_connection is provided!')
     parameters = all_parameters[0] if len(all_parameters) == 1 else None
     return sdk_no_wait(no_wait,
-                       client.update,
+                       client.begin_update,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        database_name=database_name,
@@ -683,7 +697,7 @@ def kusto_data_connection_delete(client,
                                  data_connection_name,
                                  no_wait=False):
     return sdk_no_wait(no_wait,
-                       client.delete,
+                       client.begin_delete,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        database_name=database_name,
@@ -710,10 +724,14 @@ def kusto_data_connection_data_connection_validation(client,
         raise CLIError('at most one of  event_hub_data_connection, iot_hub_data_connection, event_grid_data_connection '
                        'is needed for properties!')
     properties = all_properties[0] if len(all_properties) == 1 else None
+    parameters = {}
+    if data_connection_name is not None:
+        parameters['data_connection_name'] = data_connection_name
+    if properties is not None:
+        parameters['properties'] = properties
     return sdk_no_wait(no_wait,
-                       client.data_connection_validation,
+                       client.begin_data_connection_validation,
                        resource_group_name=resource_group_name,
                        cluster_name=cluster_name,
                        database_name=database_name,
-                       data_connection_name=data_connection_name,
-                       properties=properties)
+                       parameters=parameters)
diff --git a/.../doc/sphinx/azhelpgen/doc_source_map.json → .../doc/sphinx/azhelpgen/doc_source_map.json b/.../doc/sphinx/azhelpgen/doc_source_map.json → .../doc/sphinx/azhelpgen/doc_source_map.json
diff --git a/...ure/cli/command_modules/kusto/__init__.py → ...ure/cli/command_modules/kusto/__init__.py b/...ure/cli/command_modules/kusto/__init__.py → ...ure/cli/command_modules/kusto/__init__.py
diff --git a/.../azure/cli/command_modules/kusto/_help.py → .../azure/cli/command_modules/kusto/_help.py b/.../azure/cli/command_modules/kusto/_help.py → .../azure/cli/command_modules/kusto/_help.py
diff --git a/...azure/cli/command_modules/kusto/action.py → ...azure/cli/command_modules/kusto/action.py b/...azure/cli/command_modules/kusto/action.py → ...azure/cli/command_modules/kusto/action.py
diff --git a/...azure/cli/command_modules/kusto/custom.py → ...azure/cli/command_modules/kusto/custom.py b/...azure/cli/command_modules/kusto/custom.py → ...azure/cli/command_modules/kusto/custom.py
diff --git a/...mmand_modules/kusto/generated/__init__.py → ...mmand_modules/kusto/generated/__init__.py b/...mmand_modules/kusto/generated/__init__.py → ...mmand_modules/kusto/generated/__init__.py
diff --git a/...odules/kusto/generated/_client_factory.py → ...odules/kusto/generated/_client_factory.py b/...odules/kusto/generated/_client_factory.py → ...odules/kusto/generated/_client_factory.py
diff --git a/.../command_modules/kusto/generated/_help.py → .../command_modules/kusto/generated/_help.py b/.../command_modules/kusto/generated/_help.py → .../command_modules/kusto/generated/_help.py
diff --git a/...ommand_modules/kusto/generated/_params.py → ...ommand_modules/kusto/generated/_params.py b/...ommand_modules/kusto/generated/_params.py → ...ommand_modules/kusto/generated/_params.py
diff --git a/...nd_modules/kusto/generated/_validators.py → ...nd_modules/kusto/generated/_validators.py b/...nd_modules/kusto/generated/_validators.py → ...nd_modules/kusto/generated/_validators.py
diff --git a/...command_modules/kusto/generated/action.py → ...command_modules/kusto/generated/action.py b/...command_modules/kusto/generated/action.py → ...command_modules/kusto/generated/action.py
diff --git a/...mmand_modules/kusto/generated/commands.py → ...mmand_modules/kusto/generated/commands.py b/...mmand_modules/kusto/generated/commands.py → ...mmand_modules/kusto/generated/commands.py
@@ -74,10 +74,7 @@ def load_command_table(self, _):
         g.custom_show_command('show', 'kusto_attached_database_configuration_show')
         g.custom_command('create', 'kusto_attached_database_configuration_create', supports_no_wait=True)
         g.generic_update_command(
-            'update',
-            supports_no_wait=True,
-            custom_func_name='kusto_attached_database_configuration_update',
-            setter_name='begin_create_or_update',
+            'update', supports_no_wait=True, custom_func_name='kusto_attached_database_configuration_update'
         )
         g.custom_command(
             'delete', 'kusto_attached_database_configuration_delete', supports_no_wait=True, confirmation=True
@@ -121,10 +118,7 @@ def load_command_table(self, _):
         g.custom_show_command('show', 'kusto_cluster_principal_assignment_show')
         g.custom_command('create', 'kusto_cluster_principal_assignment_create', supports_no_wait=True)
         g.generic_update_command(
-            'update',
-            supports_no_wait=True,
-            custom_func_name='kusto_cluster_principal_assignment_update',
-            setter_name='begin_create_or_update',
+            'update', supports_no_wait=True, custom_func_name='kusto_cluster_principal_assignment_update'
         )
         g.custom_command(
             'delete', 'kusto_cluster_principal_assignment_delete', supports_no_wait=True, confirmation=True
@@ -171,10 +165,7 @@ def load_command_table(self, _):
         g.custom_show_command('show', 'kusto_database_principal_assignment_show')
         g.custom_command('create', 'kusto_database_principal_assignment_create', supports_no_wait=True)
         g.generic_update_command(
-            'update',
-            supports_no_wait=True,
-            custom_func_name='kusto_database_principal_assignment_update',
-            setter_name='begin_create_or_update',
+            'update', supports_no_wait=True, custom_func_name='kusto_database_principal_assignment_update'
         )
         g.custom_command(
             'delete', 'kusto_database_principal_assignment_delete', supports_no_wait=True, confirmation=True