[Date Prev][Date Next]   [Thread Prev][Thread Next]   [Thread Index] [Date Index] [Author Index]

[libvirt] [test-API][PATCH v5] Add test case update_devflag.py for update device flag



* repos/domain/update_devflag.py: this case is for cdrom & floppy
  update testing.
---
 repos/domain/update_devflag.py |  317 ++++++++++++++++++++++++++++++++++++++++
 1 files changed, 317 insertions(+), 0 deletions(-)
 create mode 100644 repos/domain/update_devflag.py

diff --git a/repos/domain/update_devflag.py b/repos/domain/update_devflag.py
new file mode 100644
index 0000000..1be8448
--- /dev/null
+++ b/repos/domain/update_devflag.py
@@ -0,0 +1,317 @@
+#!/usr/bin/env python
+"""Update virtual device to guest from an XML file
+   domain:update_devflag
+       guestname
+           xxx
+       devtype
+           cdrom|floppy
+       username
+           xxx
+       password
+           xxx
+"""
+
+__author__ = 'Nan Zhang: nzhang redhat com'
+__date__ = 'Fri Sep 2, 2011'
+__version__ = '0.1.0'
+__credits__ = 'Copyright (C) 2011 Red Hat, Inc.'
+__all__ = ['check_updated_device', 'update_devflag']
+
+import os
+import re
+import sys
+import time
+from xml.dom import minidom
+
+
+def append_path(path):
+    """Append root path of package"""
+    if path in sys.path:
+        pass
+    else:
+        sys.path.append(path)
+
+pwd = os.getcwd()
+result = re.search('(.*)libvirt-test-API', pwd)
+append_path(result.group(0))
+
+from lib import connectAPI
+from lib import domainAPI
+from utils.Python import utils
+from utils.Python import xmlbuilder
+from exception import LibvirtAPI
+
+def check_params(params):
+    """Verify inputing parameter dictionary"""
+    logger = params['logger']
+    keys = ['guestname', 'devtype', 'username', 'password']
+    for key in keys:
+        if key not in params:
+            logger.error("%s is required" %key)
+            return 1
+    return 0
+
+def create_image(params, util, img_name):
+    """Create an image file"""
+    logger = params['logger']
+
+    if params['devtype'] == 'cdrom':
+        cmd1 = "uuidgen|cut -d- -f1"       
+        ret1, out1 = util.exec_cmd(cmd1, shell=True)
+        if ret1:
+            logger.error("random code generated fail.")
+            return False
+
+        cmd2 = "mkdir /tmp/%s && touch /tmp/%s/$(uuidgen|cut -d- -f1)" \
+                                                   % (out1, out1)
+        ret2, out2 = util.exec_cmd(cmd2, shell=True)
+        if ret2:
+            logger.error("fail to create files for iso image: \n %s" % out1)
+            return False
+        else:
+            logger.info("prepare files for iso image creation.")
+
+        cmd3 = "genisoimage -o %s /tmp/%s" % (img_name, out1)
+        ret3, out3 = util.exec_cmd(cmd3, shell=True)
+        if ret3:
+            logger.error("iso file creation fail: \n %s" % out1)
+            return False
+        else:
+            logger.info("create an image: %s" % img_name)
+
+    elif params['devtype'] == 'floppy':
+        cmd1 = "dd if=/dev/zero of=%s bs=1440k count=1" % img_name
+        ret1, out1 = util.exec_cmd(cmd1, shell=True)
+        if ret1:
+            logger.error("floppy image creation fail: \n %s" % out1)
+            return False
+        else:
+            logger.info("create an image for floppy use: %s" % img_name)
+
+        cmd2 = "mkfs.msdos -s 1 %s" % img_name
+        ret2, out2 = util.exec_cmd(cmd2, shell=True)
+        if ret2:
+            logger.error("fail to format image: \n %s" % out2)
+            return False
+        else:
+            logger.info("succeed to format image: %s" % img_name)
+
+        cmd3 = "mount -o loop %s /mnt && touch /mnt/$(uuidgen|cut -d- -f1) \
+                                      && umount -l /mnt" % img_name
+        ret3, out3 = util.exec_cmd(cmd3, shell=True)
+        if ret3:
+            logger.error("fail to write file to floppy image: \n %s" % out3)
+            return False
+        else:
+            logger.info("succeed to write file to floopy image: %s" % img_name)
+
+    else:
+        logger.error("wrong device type was specified.")
+        return False
+
+    return True
+
+def check_device_in_guest(params, util, guestip):
+    """Check updated device in guest"""
+    logger = params['logger']
+
+    if params['devtype'] == 'cdrom':
+        cmd = "mount -o loop /dev/cdrom /media"
+    elif params['devtype'] == 'floppy':
+        cmd = "mount /dev/fd0 /media"
+    else:
+        logger.error("it's not a cdrom or floppy device.")
+        return False, None
+
+    ret, output = util.remote_exec_pexpect(guestip, params['username'], \
+                                               params['password'], cmd)
+    logger.debug(output)
+    if ret:
+        logger.error("failed to mount %s device." % params['devtype'])
+        return False, output
+
+    time.sleep(5)
+
+    ret, output = util.remote_exec_pexpect(guestip, params['username'], \
+                                           params['password'], "umount /media")
+    logger.debug(output)
+    if ret:
+        logger.error("failed to unmount %s device." % params['devtype'])
+        return False, output
+
+    time.sleep(5)
+
+    ret, output = util.remote_exec_pexpect(guestip, params['username'], \
+                                           params['password'], "ls /media")
+    logger.debug(output)
+    if ret:
+        logger.error("failed to list contents of %s device." \
+                                           % params['devtype'])
+        return False, output
+
+    return True, output
+
+def check_updated_device(params, output, util, guestip, domobj, srcfile):
+    """Check if the device is updated"""
+    logger = params['logger']
+    xmlobj = domobj.get_xml_desc(params['guestname'])
+    domxml = minidom.parseString(xmlobj)
+
+    for diskTag in domxml.getElementsByTagName("source"):
+        if diskTag.parentNode.getAttribute("device") == 'cdrom':
+            upfile = diskTag.getAttribute("file")
+        elif diskTag.parentNode.getAttribute('device') == 'floppy':
+            upfile = diskTag.getAttribute("file")
+
+    res = check_device_in_guest(params, util, guestip)
+    if res[0] and cmp(res[1], output):
+        if upfile == srcfile:
+            logger.debug("checking fail.")
+            return False, upfile
+        else:
+            logger.debug("checking successful.")
+            return True, upfile
+    else:
+        return False, upfile
+
+def update_devflag(params):
+    """Update virtual device to a domain from xml"""
+    util = utils.Utils()
+
+    # Initiate and check parameters
+    params_check_result = check_params(params)
+    if params_check_result:
+        return 1
+
+    logger = params['logger']
+    guestname = params['guestname']
+    devtype = params['devtype']
+
+    if devtype == 'cdrom':
+        xmlargs = {}
+        xmlargs['guestname'] = guestname
+        xmlargs['guesttype'] = 'kvm'
+        xmlargs['hdmodel'] = 'ide'
+        xmlargs['bootcd'] = '/var/lib/libvirt/boot/cdrom.img'
+        srcfile = xmlargs['bootcd']
+        if not create_image(params, util, srcfile):
+            return 1
+    elif devtype == 'floppy':
+        xmlargs = {}
+        xmlargs['guestname'] = guestname
+        xmlargs['floppysource'] = '/var/lib/libvirt/boot/floppy.img'
+        srcfile = xmlargs['floppysource']
+        if not create_image(params, util, srcfile):
+            return 1
+    else:
+        srcfile = None
+        logger.error("Wrong device type was specified.")
+        return 1
+
+    if not params.has_key('flag'):
+        flag = domainAPI.VIR_DOMAIN_AFFECT_CONFIG
+
+    # Connect to local hypervisor connection URI
+    uri = util.get_uri('127.0.0.1')
+    mac = util.get_dom_mac_addr(guestname)
+    guestip = util.mac_to_ip(mac, 180)
+    logger.debug("ip address: %s" % guestip)
+
+    conn = connectAPI.ConnectAPI()
+    virconn = conn.open(uri)
+    caps = conn.get_caps()
+    logger.debug(caps)
+    domobj = domainAPI.DomainAPI(virconn)
+
+    try:
+        if guestname not in domobj.get_defined_list():
+            logger.error("%s doesn't exist or in running state." % guestname)
+            return 1
+    except LibvirtAPI, e:
+        logger.error("API error message: %s, error code is %s" %
+                     (e.response()['message'], e.response()['code']))
+        return 1
+
+    guestxml = domobj.get_xml_desc(guestname)
+    guestobj = minidom.parseString(guestxml)
+
+    # Generat device XML for original use
+    origxmlobj = xmlbuilder.XmlBuilder()
+
+    if devtype == 'cdrom':
+        origxmlobj.add_cdrom(xmlargs, guestobj)
+        guestxml = origxmlobj.build_domain(guestobj)
+    elif devtype == 'floppy':
+        origxmlobj.add_floppy(xmlargs, guestobj)
+        guestxml = origxmlobj.build_domain(guestobj)
+
+    try:
+        domobj.undefine(guestname)
+        domobj.define(guestxml)
+        domobj.start(guestname)
+    except LibvirtAPI, e:
+        logger.error("API error message: %s, error code is %s" %
+                     (e.response()['message'], e.response()['code']))
+        return 1
+
+    time.sleep(60)
+    ret, output = check_device_in_guest(params, util, guestip)
+    logger.debug(output)
+    if not ret:
+        return 1
+
+    # Generate device XML for updating
+    newxmlobj = xmlbuilder.XmlBuilder()
+
+    if devtype == 'cdrom':
+        xmlargs['bootcd'] = '/var/lib/libvirt/boot/cdrom-new.img'
+        upfile = xmlargs['bootcd']
+        if not create_image(params, util, upfile):
+            logger.info("fail to create new image.")
+            return 1
+        newdevxml = newxmlobj.build_cdrom(xmlargs)
+    elif devtype == 'floppy':
+        xmlargs['floppysource'] = '/var/lib/libvirt/boot/floppy-new.img'
+        upfile = xmlargs['floppysource']
+        if not create_image(params, util, upfile):
+            logger.info("fail to create new image.")
+            return 1
+        newdevxml = newxmlobj.build_floppy(xmlargs)
+
+    logger.debug("block device xml desc for update:\n%s" % newdevxml)
+
+    logger.debug("domain xml before updating:\n%s" \
+                                   % domobj.get_xml_desc(guestname))
+
+    try:
+        domobj.update_device_flag(guestname, newdevxml, flag)
+        logger.debug("domain xml after updating:\n%s" \
+                                   % domobj.get_xml_desc(guestname))
+    except LibvirtAPI, e:
+        logger.error("API error message: %s, error code is %s" %
+                     (e.response()['message'], e.response()['code']))
+        return 1
+
+    result = check_updated_device(params, output, util, \
+                                       guestip, domobj, srcfile)
+    if result[0]:
+        logger.error("fail to update '%s' device: %s\n" % (devtype, result[1]))
+        conn.close()
+        return 1
+
+    logger.info("success to update '%s' device: %s\n" % (devtype, result[1]))
+    conn.close()
+    return 0
+
+def update_devflag_clean(params):
+    """Clean testing environment"""
+    logger = params['logger']
+
+    if params['devtype'] == 'cdrom':
+        os.unlink('/var/lib/libvirt/boot/cdrom.img')
+        os.unlink('/var/lib/libvirt/boot/cdrom-new.img')
+    elif params['devtype'] == 'floppy':
+        os.unlink('/var/lib/libvirt/boot/floppy.img')
+        os.unlink('/var/lib/libvirt/boot/floppy-new.img')
+    else:
+        logger.debug("image file does not exist or has been already removed.")
-- 
1.7.4.4


[Date Prev][Date Next]   [Thread Prev][Thread Next]   [Thread Index] [Date Index] [Author Index]