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()
