def test_attach_new_datadisk_custom_on_vm(self, mock_vm_set, mock_vm_get): #pylint: disable=line-too-long faked_vhd_uri = 'https://your_stoage_account_name.blob.core.windows.net/vhds/d1.vhd' faked_vhd_uri2 = 'https://your_stoage_account_name.blob.core.windows.net/vhds/d2.vhd' #stub to get the vm which has no datadisks vhd = VirtualHardDisk(faked_vhd_uri) existing_disk = DataDisk(lun=1, vhd=vhd, name='d1', create_option=DiskCreateOptionTypes.empty) vm = FakedVM(None, [existing_disk]) mock_vm_get.return_value = vm #execute attach_new_disk('rg1', 'vm1', VirtualHardDisk(faked_vhd_uri2), None, 'd2', 512, CachingTypes.read_write) #assert self.assertTrue(mock_vm_get.called) mock_vm_set.assert_called_once_with(vm) self.assertEqual(len(vm.storage_profile.data_disks), 2) data_disk = vm.storage_profile.data_disks[1] self.assertEqual(CachingTypes.read_write, data_disk.caching) self.assertEqual(DiskCreateOptionTypes.empty, data_disk.create_option) self.assertIsNone(data_disk.image) self.assertEqual( data_disk.lun, 0 ) #the existing disk has '1', so it verifes the second one be picked as '0' self.assertEqual(data_disk.vhd.uri, faked_vhd_uri2)
def test_attach_new_datadisk_default_on_vm(self, mock_vm_set, mock_vm_get): #pylint: disable=line-too-long faked_vhd_uri = 'https://your_stoage_account_name.blob.core.windows.net/vhds/d1.vhd' #stub to get the vm which has no datadisks vm = FakedVM(None, None) mock_vm_get.return_value = vm #execute attach_new_disk('rg1', 'vm1', VirtualHardDisk(faked_vhd_uri)) #assert self.assertTrue(mock_vm_get.called) mock_vm_set.assert_called_once_with(vm) self.assertEqual(len(vm.storage_profile.data_disks), 1) data_disk = vm.storage_profile.data_disks[0] self.assertIsNone(data_disk.caching) self.assertEqual(data_disk.create_option, DiskCreateOptionTypes.empty) self.assertIsNone(data_disk.image) self.assertEqual(data_disk.lun, 0) self.assertEqual(data_disk.name, 'd1') self.assertEqual(data_disk.vhd.uri, faked_vhd_uri)
def test_attach_new_datadisk_default_on_vm(self, mock_vm_set, mock_vm_get): #pylint: disable=line-too-long faked_vhd_uri = 'https://your_stoage_account_name.blob.core.windows.net/vhds/d1.vhd' #stub to get the vm which has no datadisks vm = FakedVM(None, None) mock_vm_get.return_value = vm #execute attach_new_disk('rg1', 'vm1', VirtualHardDisk(faked_vhd_uri)) #assert self.assertTrue(mock_vm_get.called) mock_vm_set.assert_called_once_with(vm) self.assertEqual(len(vm.storage_profile.data_disks), 1) data_disk = vm.storage_profile.data_disks[0] self.assertIsNone(data_disk.caching) self.assertEqual(data_disk.create_option, DiskCreateOptionTypes.empty) self.assertIsNone(data_disk.image) self.assertEqual(data_disk.lun, 0) self.assertEqual(data_disk.name, 'd1') self.assertEqual(data_disk.vhd.uri, faked_vhd_uri)
def test_attach_new_datadisk_custom_on_vm(self, mock_vm_set, mock_vm_get): #pylint: disable=line-too-long faked_vhd_uri = 'https://your_stoage_account_name.blob.core.windows.net/vhds/d1.vhd' faked_vhd_uri2 = 'https://your_stoage_account_name.blob.core.windows.net/vhds/d2.vhd' #stub to get the vm which has no datadisks vhd = VirtualHardDisk(faked_vhd_uri) existing_disk = DataDisk(lun=1, vhd=vhd, name='d1', create_option=DiskCreateOptionTypes.empty) vm = FakedVM(None, [existing_disk]) mock_vm_get.return_value = vm #execute attach_new_disk('rg1', 'vm1', VirtualHardDisk(faked_vhd_uri2), None, 'd2', 512, CachingTypes.read_write) #assert self.assertTrue(mock_vm_get.called) mock_vm_set.assert_called_once_with(vm) self.assertEqual(len(vm.storage_profile.data_disks), 2) data_disk = vm.storage_profile.data_disks[1] self.assertEqual(CachingTypes.read_write, data_disk.caching) self.assertEqual(DiskCreateOptionTypes.empty, data_disk.create_option) self.assertIsNone(data_disk.image) self.assertEqual(data_disk.lun, 0) #the existing disk has '1', so it verifes the second one be picked as '0' self.assertEqual(data_disk.vhd.uri, faked_vhd_uri2)