def test_fetch_vhd_image_cleans_up_vdi_on_fail(self): self._apply_stubouts() self._common_params_setup(True) self.mox.StubOutWithMock(self.session, 'call_xenapi') vm_utils._add_bittorrent_params(self.image_id, self.params) vm_utils._fetch_using_dom0_plugin_with_retry(self.context, self.session, self.image_id, "bittorrent", self.params, callback=None).AndReturn(self.vdis) vm_utils.safe_find_sr(self.session).AndReturn("sr") vm_utils._scan_sr(self.session, "sr") vm_utils._check_vdi_size(self.context, self.session, self.instance, "vdi").AndRaise(exception.InstanceTypeDiskTooSmall) self.session.call_xenapi("VDI.get_by_uuid", "vdi").AndReturn("ref") vm_utils.destroy_vdi(self.session, "ref") self.mox.ReplayAll() self.assertRaises(exception.InstanceTypeDiskTooSmall, vm_utils._fetch_vhd_image, self.context, self.session, self.instance, self.image_id) self.mox.VerifyAll()
def test_fetch_vhd_image_cleans_up_vdi_on_fail(self): self._apply_stubouts() self._common_params_setup(True) self.mox.StubOutWithMock(self.session, 'call_xenapi') vm_utils._add_bittorrent_params(self.params) vm_utils._fetch_using_dom0_plugin_with_retry(self.context, self.session, self.image_id, "bittorrent", self.params, callback=None).AndReturn(self.vdis) vm_utils.safe_find_sr(self.session).AndReturn("sr") vm_utils._scan_sr(self.session, "sr") vm_utils._check_vdi_size(self.context, self.session, self.instance, "vdi").AndRaise(exception.InstanceTypeDiskTooSmall) self.session.call_xenapi("VDI.get_by_uuid", "vdi").AndReturn("ref") vm_utils.destroy_vdi(self.session, "ref") self.mox.ReplayAll() self.assertRaises(exception.InstanceTypeDiskTooSmall, vm_utils._fetch_vhd_image, self.context, self.session, self.instance, self.image_id) self.mox.VerifyAll()
def test_fetch_vhd_image_works_with_bittorrent(self): self._apply_stubouts() self._common_params_setup(True) vm_utils._add_bittorrent_params(self.params) vm_utils._fetch_using_dom0_plugin_with_retry(self.context, self.session, self.image_id, "bittorrent", self.params, callback=None).AndReturn( self.vdis) vm_utils.safe_find_sr(self.session).AndReturn("sr") vm_utils._scan_sr(self.session, "sr") vm_utils._check_vdi_size(self.context, self.session, self.instance, "vdi") self.mox.ReplayAll() self.assertEqual( "vdi", vm_utils._fetch_vhd_image(self.context, self.session, self.instance, self.image_id)['root']['uuid']) self.mox.VerifyAll()
def test_fetch_vhd_image_works_with_glance(self): self._apply_stubouts() self._common_params_setup(False) vm_utils._generate_glance_callback(self.context).AndReturn("dummy") vm_utils._fetch_using_dom0_plugin_with_retry( self.context, self.session, self.image_id, "glance", self.params, callback="dummy").AndReturn(self.vdis) vm_utils.safe_find_sr(self.session).AndReturn("sr") vm_utils._scan_sr(self.session, "sr") vm_utils._check_vdi_size(self.context, self.session, self.instance, "vdi") self.mox.ReplayAll() self.assertEqual( "vdi", vm_utils._fetch_vhd_image(self.context, self.session, self.instance, self.image_id)['root']['uuid']) self.mox.VerifyAll()
def test_fetch_vhd_image_works_with_bittorrent(self): self._apply_stubouts() self._common_params_setup(True) vm_utils._add_bittorrent_params(self.image_id, self.params) vm_utils._fetch_using_dom0_plugin_with_retry(self.context, self.session, self.image_id, "bittorrent", self.params, callback=None).AndReturn(self.vdis) vm_utils.safe_find_sr(self.session).AndReturn("sr") vm_utils._scan_sr(self.session, "sr") vm_utils._check_vdi_size(self.context, self.session, self.instance, "vdi") self.mox.ReplayAll() self.assertEqual("vdi", vm_utils._fetch_vhd_image(self.context, self.session, self.instance, self.image_id)['root']['uuid']) self.mox.VerifyAll()
def test_fetch_vhd_image_works_with_glance(self): self._apply_stubouts() self._common_params_setup(False) vm_utils._generate_glance_callback(self.context).AndReturn("dummy") vm_utils._fetch_using_dom0_plugin_with_retry(self.context, self.session, self.image_id, "glance", self.params, callback="dummy").AndReturn(self.vdis) vm_utils.safe_find_sr(self.session).AndReturn("sr") vm_utils._scan_sr(self.session, "sr") vm_utils._check_vdi_size(self.context, self.session, self.instance, "vdi") self.mox.ReplayAll() self.assertEqual("vdi", vm_utils._fetch_vhd_image(self.context, self.session, self.instance, self.image_id)['root']['uuid']) self.mox.VerifyAll()