From bb18454d47fbcd6716ee8336f5fa4dbb0aac7772 Mon Sep 17 00:00:00 2001
From: Xu Tian <xutian@redhat.com>
Date: Thu, 23 May 2013 18:47:37 +0800
Subject: [PATCH] qemu.tests: new class for qemu-img function tests

Implement some basic operations for qemu-img function tests

Changes from V1:
    replace params.get() with params[] to avoid missing param errors.

Signed-off-by: Xu Tian <xutian@redhat.com>
---
 qemu/tests/qemu_disk_img.py | 127 ++++++++++++++++++++++++++++++++++++
 1 file changed, 127 insertions(+)
 create mode 100644 qemu/tests/qemu_disk_img.py

diff --git a/qemu/tests/qemu_disk_img.py b/qemu/tests/qemu_disk_img.py
new file mode 100644
index 00000000..acf60d2d
--- /dev/null
+++ b/qemu/tests/qemu_disk_img.py
@@ -0,0 +1,127 @@
+import os, logging
+from autotest.client.shared import error
+from autotest.client import utils
+from virttest import data_dir, env_process
+from virttest import storage, qemu_storage
+
+
+class QemuImgTest(qemu_storage.QemuImg):
+
+    data_dir = data_dir.get_data_dir()
+
+    def __init__(self, test, params, env, tag):
+        self.vm = None
+        self.test = test
+        self.params = params
+        self.env = env
+        self.tag = tag
+        self.trash = []
+        t_params = params.object_params(tag)
+        super(QemuImgTest, self).__init__(t_params, self.data_dir, tag)
+
+    @error.context_aware
+    def create_snapshot(self, t_params={}):
+        """
+        create snapshot image file
+        """
+        error.context("create snapshot image")
+        params = self.params.object_params(self.tag)
+        params.update(t_params)
+        if len(params.get("image_chain", "").split()) < 2:
+            return dict()
+        snapshot = storage.get_image_filename(params, self.data_dir)
+        if os.path.exists(snapshot):
+            utils.run("rm -f %s" % snapshot)
+        super(QemuImgTest, self).create(params)
+        self.trash.append(snapshot)
+        return params
+
+    @error.context_aware
+    def start_vm(self, t_params={}):
+        """
+        Start a vm and wait for it bootup;
+        """
+        error.context("start vm", logging.info)
+        params = self.params.object_params(self.tag)
+        params.update(t_params)
+        base_image = params.get("images", "image1").split()[0]
+        params["start_vm"] = "yes"
+        try:
+            del params["image_name_%s" % base_image]
+            del params["image_format_%s" % base_image]
+        except KeyError:
+            pass
+        vm_name = params["main_vm"]
+        env_process.preprocess_vm(self.test, params, self.env, vm_name)
+        vm = self.env.get_vm(vm_name)
+        vm.verify_alive()
+        login_timeout = int(self.params.get("login_timeout", 360))
+        vm.wait_for_login(timeout=login_timeout)
+        self.vm = vm
+        return vm
+
+    @error.context_aware
+    def __create_file(self, dst):
+        error.context("create tmp file on host")
+        if not self.vm:
+            return False
+        src = self.params["tmp_file_name"]
+        cmd = self.params["file_create_cmd"] % src
+        utils.run(cmd)
+        self.vm.copy_files_to(src, dst)
+        self.trash.append(src)
+        return True
+
+    def __md5sum(self, cmd):
+        if not self.vm:
+            return False
+        login_timeout = int(self.params.get("login_timeout", 360))
+        session = self.vm.wait_for_login(timeout=login_timeout)
+        md5bin = self.params["md5sum"]
+        cmd = "%s %s" % (md5bin, cmd)
+        s, o = session.cmd_status_output(cmd)
+        if s != 0:
+            logging.info("Execute '%s' with failures('%s') " % (cmd, o))
+            return False
+        return True
+
+    @error.context_aware
+    def save_file(self, dst):
+        error.context("save file('%s') md5sum in guest" % dst, logging.info)
+        self.__create_file(dst)
+        cmd = "%s > %s.md5 " % (dst, dst)
+        return self.__md5sum(cmd)
+
+    @error.context_aware
+    def check_file(self, dst):
+        error.context("check file('%s') md5sum in guest" % dst, logging.info)
+        cmd = "-c %s.md5 " % dst
+        return self.__md5sum(cmd)
+
+    @error.context_aware
+    def destroy_vm(self):
+        error.context("destroy vm", logging.info)
+        if self.vm:
+            self.vm.destroy()
+        self.vm = None
+
+    @error.context_aware
+    def check_image(self, t_params={}):
+        error.context("check image file ('%s')" % self.image_filename,
+            logging.info)
+        return super(QemuImgTest, self).check_image(t_params, self.data_dir)
+
+    @error.context_aware
+    def get_info(self):
+        error.context("get image file ('%s')" % self.image_filename)
+        return super(QemuImgTest, self).info()
+
+    @error.context_aware
+    def clean(self):
+        error.context("clean up useless images")
+        self.destroy_vm()
+        for _file in self.trash:
+            utils.run("rm -f %s" % _file)
+
+def run_qemu_disk_img(test, params, env):
+    pass
-- 
GitLab