show zone information on (#5014)

src/command_modules/azure-cli-vm/HISTORY.rst

@@ -5,6 +5,7 @@ Release History
 
 2.0.19
 ++++++
+* show zone information on `az vm list-skus -otable`
 * Update the storage multiapi package reference
 
 2.0.18

src/command_modules/azure-cli-vm/azure/cli/command_modules/vm/commands.py

@@ -95,6 +95,8 @@ def transform_sku_for_table_output(skus):
         order_dict['resourceType'] = k['resourceType']
         order_dict['locations'] = str(k['locations']) if len(k['locations']) > 1 else k['locations'][0]
         order_dict['name'] = k['name']
+        if k.get('locationInfo'):
+            order_dict['zones'] = ','.join(sorted(k['locationInfo'][0].get('zones', [])))
         order_dict['size'] = k['size']
         order_dict['tier'] = k['tier']
         if k['capabilities']:

src/command_modules/azure-cli-vm/azure/cli/command_modules/vm/tests/test_vm_commands.py

@@ -2234,6 +2234,16 @@ def test_disk_create_zones(self, resource_group, resource_group_location):
         table_output = set(result.output.splitlines()[2].split())
         self.assertTrue(set([resource_group, resource_group_location, disk_name, zones]).issubset(table_output))
 
+    def test_list_skus_contains_zone_info(self):
+        from azure_devtools.scenario_tests import LargeResponseBodyProcessor
+        large_resp_body = next((r for r in self.recording_processors if isinstance(r, LargeResponseBodyProcessor)), None)
+        if large_resp_body:
+            large_resp_body._max_response_body = 2048
+        # we pick eastus2 as it is one of 3 regions so far with zone support
+        location = 'eastus2'
+        result = self.cmd('vm list-skus -otable -l {} -otable'.format(location))
+        self.assertTrue(next(l for l in result.output.splitlines() if '1,2,3' in l).split()[-1] == '1,2,3')
+
 
 class VMRunCommandScenarioTest(ScenarioTest):
     @ResourceGroupPreparer()