def test_vm_validator_retrieve_image_info_cross_subscription(self, factory_mock): ns = argparse.Namespace() cmd = mock.MagicMock() data_disk = mock.MagicMock() data_disk.storage_account_type = 'Standard_LRS' image_info = mock.MagicMock() client_mock = mock.MagicMock() image_info.storage_profile.os_disk.os_type.value = 'someOS' image_info.storage_profile.data_disks = [data_disk] client_mock.images.get.return_value = image_info factory_mock.return_value = client_mock ns.image = '/subscriptions/0b1f6471-1bf0-4dda-aec3-xxxxxxxxxxxx/resourceGroups/foo/providers/Microsoft.Compute/images/bar' ns.admin_username = '******' ns.admin_password = '******' ns.storage_sku = ['Premium_LRS'] ns.ultra_ssd_enabled = None ns.os_caching, ns.data_caching = None, None ns.os_type, ns.attach_os_disk, ns.storage_account, ns.storage_container_name, ns.use_unmanaged_disk, ns.data_disk_sizes_gb = None, None, None, None, False, None ns.size = 'Standard_DS1_v2' _validate_vm_create_storage_profile(cmd, ns, False) self.assertEqual(ns.os_type, 'someOS') self.assertTrue(0 in ns.disk_info)
def test_vm_validator_retrieve_image_info_cross_subscription( self, factory_mock): ns = argparse.Namespace() cmd = mock.MagicMock() data_disk = mock.MagicMock() data_disk.storage_account_type = 'Standard_LRS' image_info = mock.MagicMock() client_mock = mock.MagicMock() image_info.storage_profile.os_disk.os_type.value = 'someOS' image_info.storage_profile.data_disks = [data_disk] client_mock.images.get.return_value = image_info factory_mock.return_value = client_mock ns.image = '/subscriptions/0b1f6471-1bf0-4dda-aec3-xxxxxxxxxxxx/resourceGroups/foo/providers/Microsoft.Compute/images/bar' ns.admin_username = '******' ns.admin_password = '******' ns.storage_sku = ['Premium_LRS'] ns.ultra_ssd_enabled = None ns.os_caching, ns.data_caching = None, None ns.os_type, ns.attach_os_disk, ns.storage_account, ns.storage_container_name, ns.use_unmanaged_disk, ns.data_disk_sizes_gb = None, None, None, None, False, None ns.size = 'Standard_DS1_v2' _validate_vm_create_storage_profile(cmd, ns, False) self.assertEqual(ns.os_type, 'someOS') self.assertTrue(0 in ns.disk_info)
def test_vm_validator_retrieve_image_info_cross_subscription( self, factory_mock): ns = argparse.Namespace() cmd = mock.MagicMock() image_info = mock.MagicMock() client_mock = mock.MagicMock() image_info.storage_profile.os_disk.os_type.value = 'someOS' image_info.storage_profile.data_disks = ['does not matter'] client_mock.images.get.return_value = image_info factory_mock.return_value = client_mock ns.image = '/subscriptions/0b1f6471-1bf0-4dda-aec3-xxxxxxxxxxxx/resourceGroups/foo/providers/Microsoft.Compute/images/bar' ns.admin_username = '******' ns.admin_password = '******' ns.storage_sku = 'Premium_LRS' ns.os_type, ns.attach_os_disk, ns.storage_account, ns.storage_container_name, ns.use_unmanaged_disk = None, None, None, None, False _validate_vm_create_storage_profile(cmd, ns, False) self.assertEqual(ns.os_type, 'someOS') self.assertEqual(ns.image_data_disks, ['does not matter'])