def setUp(self):
self.image_manager = MagicMock()
self.vm_manager = MagicMock()
self.image_scanner = DatastoreImageScanner(self.image_manager,
self.vm_manager,
self.DATASTORE_ID)
self.synchronizer = TestSynchronizer()
self.wait_at_the_end_of_scan = False
self.raise_exception = False
self.timeout = self.TIMEOUT
def setUp(self):
self.test_dir = os.path.join(tempfile.mkdtemp(), self.BASE_TEMP_DIR)
services.register(ServiceName.AGENT_CONFIG, MagicMock())
self.image_manager = EsxImageManager(MagicMock(), MagicMock())
self.vm_manager = MagicMock()
self.image_scanner = DatastoreImageScanner(self.image_manager,
self.vm_manager,
self.DATASTORE_ID)
self.image_scanner._task_runner = MagicMock()
self.image_scanner._task_runner.is_stopped.return_value = False
self.write_count = 0
# Create various image directories and empty vmdks
dir0 = os.path.join(self.test_dir, "images/im")
image_id_1 = str(uuid.uuid4())
image_id_2 = str(uuid.uuid4())
image_id_3 = str(uuid.uuid4())
image_id_4 = "invalid_image_id"
self.image_ids = ["", image_id_1, image_id_2, image_id_3, image_id_4]
dir1 = os.path.join(dir0, image_id_1)
os.makedirs(dir1)
dir2 = os.path.join(dir0, image_id_2)
os.makedirs(dir2)
dir3 = os.path.join(dir0, image_id_3)
os.makedirs(dir3)
dir4 = os.path.join(dir0, image_id_4)
os.makedirs(dir4)
# Create a vmdk under "im", since the image_id is
# not a valid uuid it should be skipped
open(os.path.join(dir0, "im.vmdk"), 'w').close()
# Create a good image vmdk under image_id_1, the name
# of the vmdk matches the directory that contains it
# so this is a valid image to remove
vmdk_filename = image_id_1 + ".vmdk"
open(os.path.join(dir1, vmdk_filename), 'w').close()
# Create a good image vmdk under image_id_2, also create
# an unused image marker file, image_id_2 should also be
# included in the list of images to remove
vmdk_filename = image_id_2 + ".vmdk"
open(os.path.join(dir2, vmdk_filename), 'w').close()
open(os.path.join(dir2, self.image_manager.IMAGE_MARKER_FILE_NAME),
'w').close()
# Don't create anything under directory dir3
# it should still mark the image as deletable
# Create a vmdk under an invalid image directory, since
# the image id is not valid it should not mark it
# for deletion
vmdk_filename = image_id_4 + ".vmdk"
open(os.path.join(dir4, vmdk_filename), 'w').close()
def setUp(self):
self.image_manager = MagicMock()
self.vm_manager = MagicMock()
self.image_scanner = DatastoreImageScanner(self.image_manager,
self.vm_manager,
self.DATASTORE_ID)
self.synchronizer = TestSynchronizer()
self.wait_at_the_end_of_scan = False
self.vm_manager.get_vm_images.side_effect = \
self.fake_get_vm_images
self.image_manager.mark_unused.side_effect = \
self.fake_mark_unused
self.raise_exception = False
self.timeout = self.TIMEOUT
def setUp(self, connect, update, creds):
# Create VM manager
creds.return_value = ["username", "password"]
self.vim_client = VimClient(auto_sync=False)
self.vim_client.wait_for_task = MagicMock()
self.patcher = patch("host.hypervisor.esx.vm_config.GetEnv")
self.patcher.start()
self.vm_manager = EsxVmManager(self.vim_client, MagicMock())
services.register(ServiceName.AGENT_CONFIG, MagicMock())
# Set up test files
self.base_dir = os.path.dirname(__file__)
self.test_dir = os.path.join(self.base_dir, "../../test_files")
self.image_manager = EsxImageManager(MagicMock(), MagicMock())
self.image_scanner = DatastoreImageScanner(self.image_manager,
self.vm_manager,
self.DATASTORE_ID)
self.write_count = 0
