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