How to use _remove_disk_img method in autotest

Best Python code snippet using autotest_python

partition.py

Source:partition.py

...679        from the loopback device and removes the image file.680        """681        logging.debug('Removing virtual partition - device %s', self.device)682        self._detach_img_loop()683        self._remove_disk_img()684    def _create_disk_img(self, img_path):685        """686        Creates a disk image using dd.687        @param img_path: Path to the desired image file.688        @param size: Size of the desired image in MB.689        @returns: Path of the image created.690        """691        logging.debug('Creating disk image %s, size = %d MB',692                      img_path, self.size)693        try:694            cmd = 'truncate %s --size %dM' % (img_path, self.size)695            utils.run(cmd)696        except error.CmdError, e:697            e_msg = 'Error creating disk image %s: %s' % (img_path, e)698            raise error.AutotestError(e_msg)699        return img_path700    def _attach_img_loop(self):701        """702        Attaches a file image to a loopback device using losetup.703        @returns: Path of the loopback device associated.704        """705        logging.debug('Attaching image %s to a loop device', self.img)706        try:707            cmd = 'losetup -f'708            loop_path = utils.system_output(cmd)709            cmd = 'losetup -f %s' % self.img710            utils.run(cmd)711        except error.CmdError, e:712            e_msg = ('Error attaching image %s to a loop device: %s' %713                     (self.img, e))714            raise error.AutotestError(e_msg)715        return loop_path716    def _detach_img_loop(self):717        """718        Detaches the image file from the loopback device.719        """720        logging.debug('Detaching image %s from loop device %s', self.img,721                      self.loop)722        try:723            cmd = 'losetup -d %s' % self.loop724            utils.run(cmd)725        except error.CmdError, e:726            e_msg = ('Error detaching image %s from loop device %s: %s' %727                    (self.img, self.loop, e))728            raise error.AutotestError(e_msg)729    def _remove_disk_img(self):730        """731        Removes the disk image.732        """733        logging.debug('Removing disk image %s', self.img)734        try:735            os.remove(self.img)736        except:737            e_msg = 'Error removing image file %s' % self.img738            raise error.AutotestError(e_msg)739# import a site partition module to allow it to override functions740try:741    from autotest_lib.client.bin.site_partition import *742except ImportError:743    pass

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.