[1/2] KVM Test: Add a new kvm subtest multi_disk.

Commit Message

Feng Yang May 25, 2011, 9:19 a.m. UTC

From: Feng Yang <[e-mail]>

This case test multi disk suport in kvm guest os. It can work on Linux and
Windows guest.

Signed-off-by: Feng Yang <[e-mail]>
---
 client/tests/kvm/tests/multi_disk.py |  127 ++++++++++++++++++++++++++++++++++
 1 files changed, 127 insertions(+), 0 deletions(-)
 create mode 100644 client/tests/kvm/tests/multi_disk.py

Patch

diff --git a/client/tests/kvm/tests/multi_disk.py b/client/tests/kvm/tests/multi_disk.py
new file mode 100644
index 0000000..262a65d
--- /dev/null
+++ b/client/tests/kvm/tests/multi_disk.py
@@ -0,0 +1,127 @@ 
+import logging, re, random
+from autotest_lib.client.common_lib import error
+
+def run_multi_disk(test, params, env):
+    """
+    Test multi disk suport of guest, this case will:
+    1)create disks image in configuration file
+    2)start the guest with those disks.
+    3)format those disks.
+    4)cope file into / out of those disks.
+    5)compare the original file and the copyed file by md5 or fc comand.
+    6) Repeat step 3-5 if need
+
+    @param test: kvm test object
+    @param params: Dictionary with the test parameters
+    @param env: Dictionary with test environment.
+    """
+    vm = env.get_vm(params["main_vm"])
+    vm.verify_alive()
+    session = vm.wait_for_login(timeout=int(params.get("login_timeout", 360)))
+
+    images = params.get("images").split()
+    n_repeat = int(params.get("n_repeat", "1"))
+    image_num = len(images)
+    disk_num = 0
+    file_system = params.get("file_system").split()
+    fs_num = len(file_system)
+    cmd_timeout = float(params.get("cmd_timeout", 360))
+    re_str = params.get("re_str")
+    block_list = params.get("block_list").split()
+    try:
+        if params.get("clean_cmd"):
+            cmd = params.get("clean_cmd")
+            session.cmd_status_output(cmd)
+        if params.get("pre_cmd"):
+            cmd = params.get("pre_cmd")
+            (s,output) = session.cmd_status_output(cmd, timeout=cmd_timeout)
+            if s != 0:
+                raise error.TestFail("Create partition on disk failed.\n"
+                                     "Output is:%s \n" % output)
+        cmd = params.get("list_volume_command")
+        (s,output) = session.cmd_status_output(cmd, timeout=cmd_timeout)
+        if s != 0:
+            raise error.TestFail("List volume command failed.\n"
+                                 "Output is:%s\n" % output)
+        disks = re.findall(re_str, output)
+        disks.sort()
+        logging.debug("Volume list that meet regular expressions: %s" % disks)
+        if len(disks) < image_num:
+            raise error.TestFail("Fail to list all the volume!")
+
+        tmp_list = []
+        for disk in disks:
+            if disk.strip() in block_list:
+                tmp_list.append(disk)
+        for disk in tmp_list:
+            logging.info("No need check volume %s" % disk)
+            disks.remove(disk)
+
+        for i in range(n_repeat):
+            logging.info("iterations: %s" %(i + 1))
+            for disk in disks:
+                disk = disk.strip()
+
+                logging.info("Format disk: %s..." % disk)
+                index = random.randint(0,fs_num -1)
+
+                # Random select one file system from file_system
+                fs = file_system[index].strip()
+                cmd = params.get("format_command") % (fs, disk)
+                (s, output) = session.cmd_status_output(cmd,
+                                                        timeout=cmd_timeout)
+                if s != 0:
+                    raise error.TestFail("Format disk failed with output: %s" %
+                                                                        output)
+                if params.get("mount_command"):
+                    cmd = params.get("mount_command") % (disk, disk, disk)
+                    (s, output) = session.cmd_status_output(cmd)
+                    if s != 0:
+                        raise error.TestFail("Mount disk failed. Output: %s" %
+                                                                     output)
+
+            for disk in disks:
+                disk = disk.strip()
+
+                logging.info("Performing I/O on disk: %s..." % disk)
+                cmd_list = params.get("cmd_list").split()
+                for cmd_l in cmd_list:
+                    if params.get(cmd_l):
+                        cmd = params.get(cmd_l) % disk
+                        (s, output) = session.cmd_status_output(cmd,
+                                                               timeout=cmd_timeout)
+                        if s != 0:
+                            raise error.TestFail("Failed command: %. Output:%s" %
+                                                                      (cmd,output))
+                cmd = params.get("compare_command")
+                (s, output) = session.cmd_status_output(cmd)
+                if s != 0:
+                    raise error.TestFail("Fail to compare two files. Output:%s" %
+                                                                      output)
+                key_word = params.get("check_result_key_word")
+                if key_word and key_word in output:
+                    logging.debug("Guest's virtual disk %s works fine" % disk)
+                elif key_word:
+                    raise error.TestFail("Two files are not the same!")
+                else:
+                    raise error.TestError("check_result_key_word is not specified!")
+
+            if params.get("umount_command"):
+                cmd = params.get("show_mount_cmd")
+                (s,output) = session.cmd_status_output(cmd)
+                disks = re.findall(re_str, output)
+                disks.sort()
+                for disk in disks:
+                    disk = disk.strip()
+
+                    cmd = params.get("umount_command") % (disk, disk)
+                    (s,output) = session.cmd_status_output(cmd)
+                    if s != 0:
+                        raise error.TestFail("Fail to unmount disk. Output is:%s" %
+                                                                            output)
+    finally:
+        if params.get("post_cmd"):
+            cmd = params.get("post_cmd")
+            session.cmd_status_output(cmd)
+        session.close()
+