def test__get_image_download_in_progress(self):
def _create_instance_path(*args):
open(master_path, 'w').close()
return True
temp_dir = tempfile.mkdtemp()
instance_path = os.path.join(temp_dir, 'instance_path')
fileutils.ensure_tree(temp_dir)
master_uuid = 'instance_uuid'
master_path = os.path.join(temp_dir, master_uuid)
lock_file = os.path.join(temp_dir, 'instance_uuid.lock')
self.mox.StubOutWithMock(pxe, '_download_in_progress')
pxe._download_in_progress(lock_file).\
WithSideEffects(_create_instance_path).\
AndReturn(True)
self.mox.ReplayAll()
pxe._get_image(None, instance_path, master_uuid, temp_dir)
self.mox.VerifyAll()
self.assertTrue(os.path.exists(instance_path))
def test__download_in_progress(self):
temp_dir = tempfile.mkdtemp()
lock_file = os.path.join(temp_dir, 'lock_file')
self.assertFalse(pxe._download_in_progress(lock_file))
self.assertTrue(os.path.exists(lock_file))
def test__download_in_progress(self):
temp_dir = tempfile.mkdtemp()
lock_file = os.path.join(temp_dir, 'lock_file')
self.assertFalse(pxe._download_in_progress(lock_file))
self.assertTrue(os.path.exists(lock_file))
