{HDInsight}Use getting default sku api to set workernode and headnode size if customer does not provide

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

@@ -2,11 +2,11 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
-
 from knack.log import get_logger
 from knack.prompting import prompt_pass, NoTTYException
 from knack.util import CLIError
 from azure.cli.core.util import sdk_no_wait, user_confirmation
+from azure.cli.core.azclierror import RequiredArgumentMissingError
 
 logger = get_logger(__name__)
 
@@ -15,7 +15,7 @@
 def create_cluster(cmd, client, cluster_name, resource_group_name, cluster_type,
                    location=None, tags=None, no_wait=False, cluster_version='default', cluster_tier=None,
                    cluster_configurations=None, component_version=None,
-                   headnode_size='large', workernode_size='large', zookeepernode_size=None, edgenode_size=None,
+                   headnode_size=None, workernode_size=None, zookeepernode_size=None, edgenode_size=None,
                    kafka_management_node_size=None, kafka_management_node_count=2,
                    kafka_client_group_id=None, kafka_client_group_name=None,
                    workernode_count=3, workernode_data_disks_per_node=None,
@@ -39,7 +39,7 @@ def create_cluster(cmd, client, cluster_name, resource_group_name, cluster_type,
                    resource_provider_connection=None, enable_private_link=None, enable_compute_isolation=None,
                    host_sku=None, no_validation_timeout=False):
     from .util import build_identities_info, build_virtual_network_profile, parse_domain_name, \
-        get_storage_account_endpoint, validate_esp_cluster_create_params
+        get_storage_account_endpoint, validate_esp_cluster_create_params, set_vm_size
     from azure.mgmt.hdinsight.models import ClusterCreateParametersExtended, ClusterCreateProperties, OSType, \
         ClusterDefinition, ComputeProfile, HardwareProfile, Role, OsProfile, LinuxOperatingSystemProfile, \
         StorageProfile, StorageAccount, DataDisksGroups, SecurityProfile, \
@@ -197,6 +197,16 @@ def create_cluster(cmd, client, cluster_name, resource_group_name, cluster_type,
         )
     ]
 
+    # call get default vm size api to set vm size if customer does not provide the value
+    if not (workernode_size and headnode_size):
+        headnode_size, workernode_size = set_vm_size(cmd.cli_ctx, location, cluster_type, headnode_size,
+                                                     workernode_size)
+
+    if not headnode_size:
+        raise RequiredArgumentMissingError('Please specify --headnode-size explicitly.')
+    if not workernode_size:
+        raise RequiredArgumentMissingError('Please specify --workernode-size explicitly.')
+
     os_profile = OsProfile(
         linux_operating_system_profile=LinuxOperatingSystemProfile(
             username=ssh_username,

src/azure-cli/azure/cli/command_modules/hdinsight/tests/latest/recordings/test_hdinsight_usage.yaml

@@ -13,24 +13,24 @@ interactions:
       ParameterSetName:
       - -l
       User-Agent:
-      - python/3.7.2 (Windows-10-10.0.17763-SP0) msrest/0.6.7 msrest_azure/0.6.1 azure-mgmt-hdinsight/0.2.1
-        Azure-SDK-For-Python AZURECLI/2.0.69
+      - python/3.7.2 (Windows-10-10.0.19041-SP0) msrest/0.6.21 msrest_azure/0.6.3
+        azure-mgmt-hdinsight/2.2.0 Azure-SDK-For-Python AZURECLI/2.22.1
       accept-language:
       - en-US
     method: GET
-    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/providers/Microsoft.HDInsight/locations/eastus2/usages?api-version=2018-06-01-preview
+    uri: https://management.azure.com/subscriptions/00000000-0000-0000-0000-000000000000/providers/Microsoft.HDInsight/locations/southcentralus/usages?api-version=2018-06-01-preview
   response:
     body:
-      string: '{"value":[{"unit":"Count","currentValue":0,"limit":60,"name":{"value":"cores","localizedValue":"Cores"}}]}'
+      string: '{"value":[{"unit":"Count","currentValue":40,"limit":3000,"name":{"value":"cores","localizedValue":"Cores"}}]}'
     headers:
       cache-control:
       - no-cache
       content-length:
-      - '106'
+      - '109'
       content-type:
       - application/json; charset=utf-8
       date:
-      - Fri, 26 Jul 2019 04:16:01 GMT
+      - Sun, 25 Apr 2021 13:22:04 GMT
       expires:
       - '-1'
       pragma:
@@ -44,7 +44,7 @@ interactions:
       x-content-type-options:
       - nosniff
       x-ms-hdi-served-by:
-      - global
+      - southcentralus
     status:
       code: 200
       message: OK