[Libvirt-cim] [PATCH 1 of 2] [TEST] 32_modify_cdrom_media.py: new test for VirtualSystemManagementService

Eduardo Lima (Etrunko) eblima at linux.vnet.ibm.com
Fri Sep 30 18:59:41 UTC 2011


 suites/libvirt-cim/cimtest/VirtualSystemManagementService/32_modify_cdrom_media.py |  221 ++++++++++
 1 files changed, 221 insertions(+), 0 deletions(-)


# HG changeset patch
# User Eduardo Lima (Etrunko) <eblima at br.ibm.com>
# Date 1317409066 10800
# Node ID 85c624ef88f4e1487f95a5e50674a626a4ebf8f4
# Parent  f4bcd9833525c6914f61e29e9d2d8adbac240682
[TEST] 32_modify_cdrom_media.py: new test for VirtualSystemManagementService

This test case covers a requirement to change the media in the CDROM using
libvirt-cim, which is done via a ModifyResourceSettings call.

Signed-off-by: Eduardo Lima (Etrunko) <eblima at br.ibm.com>

diff --git a/suites/libvirt-cim/cimtest/VirtualSystemManagementService/32_modify_cdrom_media.py b/suites/libvirt-cim/cimtest/VirtualSystemManagementService/32_modify_cdrom_media.py
new file mode 100755
--- /dev/null
+++ b/suites/libvirt-cim/cimtest/VirtualSystemManagementService/32_modify_cdrom_media.py
@@ -0,0 +1,221 @@
+#!/usr/bin/env python
+
+#
+# Copyright 2011 IBM Corp.
+#
+# Authors:
+#   Eduardo Lima (Etrunko) <eblima at br.ibm.com>
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA
+#
+
+#
+# Create a domain with cdrom device without media connected.
+# ModifyResourceSettings call to change the cdrom media
+#
+
+import sys
+import os
+import pywbem
+
+from CimTest.ReturnCodes import PASS, FAIL, XFAIL, SKIP
+from CimTest.Globals import logger, CIM_USER, CIM_PASS, CIM_NS
+from XenKvmLib.const import do_main, _image_dir
+from XenKvmLib.classes import get_typed_class
+from XenKvmLib.vxml import get_class
+
+supported = ['KVM',]
+
+cim = None
+sys_mgmt_service = None
+domain = None
+
+class CIMDomain(object):
+
+    def __init__(self, name, virt, server):
+        self.name = name
+        self.server = server
+        self.virt = virt
+        self._domain = get_class(virt)(name)
+    #__init__
+
+    def define(self):
+        return self._domain.cim_define(self.server)
+    # define
+
+    def undefine(self):
+        return self._domain.undefine(self.server)
+    # undefine
+
+    def destroy(self):
+        return self._domain.cim_destroy(self.server)
+    #destroy
+# CIMDomain
+
+
+def set_device_addr(inst, address):
+    return """
+instance of %s {
+    InstanceID="%s";
+    ResourceType=%d;
+    PoolID="%s";
+    AllocationUnits="%s";
+    Address="%s";
+    VirtualQuantityUnits="%s";
+    VirtualDevice="%s";
+    EmulatedType=%d;
+    BusType="%s";
+    DriverName="%s";
+    DriverType="%s";
+};""" % (get_typed_class(domain.virt, "DiskResourceAllocationSettingData"),
+        inst["InstanceID"],
+        inst["ResourceType"],
+        inst["PoolID"],
+        inst["AllocationUnits"],
+        address,
+        inst["VirtualQuantityUnits"],
+        inst["VirtualDevice"],
+        inst["EmulatedType"],
+        inst["BusType"],
+        inst["DriverName"],
+        inst["DriverType"],)
+# set_device_addr()
+
+
+def modify_media(cim, inst, addr):
+    logger.info("Setting media addr to '%s'", addr)
+
+    val = set_device_addr(inst, addr)
+    ret = cim.InvokeMethod("ModifyResourceSettings", sys_mgmt_service, **{"ResourceSettings": [val,],})
+
+    if ret[0]:
+        logger.error("Modifying media: %s", ret)
+        return None
+
+    inst = cim.GetInstance(ret[1]["ResultingResourceSettings"][0])
+    new_addr = inst["Address"]
+
+    if new_addr != addr:
+        logger.error("New media '%s' does not match expected '%s'", new_addr, addr)
+        return None
+
+    return inst
+# modify_media()
+
+
+ at do_main(supported)
+def main():
+    options = main.options
+    server = options.ip
+    virt = options.virt
+
+    server_url = "http://%s" % server
+    global cim
+    cim = pywbem.WBEMConnection(server_url, (CIM_USER, CIM_PASS), CIM_NS)
+
+    _class = get_typed_class(virt, "VirtualSystemManagementService")
+    global sys_mgmt_service
+    sys_mgmt_service = cim.EnumerateInstanceNames(_class)[0]
+
+    # Create new domain
+    global domain
+    domain = CIMDomain("cimtest_modify_cdrom", virt, server)
+    if not domain.define():
+        logger.error("Error defining test domain")
+        return FAIL
+
+    logger.info("Domain XML\n%s", domain._domain)
+    # ein KVM_ComputerSystem
+    _class = get_typed_class(virt, "ComputerSystem")
+    computer_system_names = [i for i in cim.EnumerateInstanceNames(_class) if i["Name"] == domain.name]
+
+    logger.info("ComputerSystem Names\n%s", computer_system_names)
+
+    if not computer_system_names:
+        logger.info("Host has no domains defined")
+        return SKIP
+
+    # ain -ac KVM_SystemDevice -arc KVM_LogicalDisk <KVM_ComputerSytem Name>
+    a_class = get_typed_class(virt, "SystemDevice")
+    r_class = get_typed_class(virt, "LogicalDisk")
+    logical_disk_names = []
+
+    for inst_name in computer_system_names:
+        assoc_names = cim.AssociatorNames(inst_name, AssocClass=a_class, ResultClass=r_class)
+        logical_disk_names.extend(assoc_names)
+
+    logger.info("LogicalDisk Names\n%s", logical_disk_names)
+
+    if not logical_disk_names:
+        logger.info("No LogicalDisk instances returned")
+        return FAIL
+
+    # ai -arc KVM_DiskResourceAllocationSettingData <KVM_LogicalDisk Name>
+    rclass = get_typed_class(virt, "DiskResourceAllocationSettingData")
+    disk_rasd_names = []
+
+    for inst_name in logical_disk_names:
+        assoc_names = cim.AssociatorNames(inst_name, ResultClass=rclass)
+        disk_rasd_names.extend(assoc_names)
+
+    logger.info("DiskRASD names\n%s", disk_rasd_names)
+
+    if not disk_rasd_names:
+        logger.info("No DiskRASD instances returned")
+        return FAIL
+
+    cdrom_devices = [i for i in disk_rasd_names if cim.GetInstance(i)["EmulatedType"] == 1]
+
+    logger.info("CDROM devices\n%s", cdrom_devices)
+
+    if not cdrom_devices:
+        logger.info("No CDROM device found")
+        return FAIL
+
+    cdrom = cdrom_devices[0]
+    inst = cim.GetInstance(cdrom)
+
+    for media in ["cdrom01.iso", "cdrom02.iso"]:
+        if not inst:
+            logger.error("Unable to get CDROM device instance")
+            return FAIL
+
+        # Get current media address
+        old_media = inst["Address"]
+
+        logger.info("Current CDROM media: '%s'", old_media)
+
+        if not media and not old_media:
+            logger.info("CDROM device has no media connected")
+            continue
+
+        # Need to eject first?
+        if media and old_media:
+            inst = modify_media(cim, inst, "")
+
+        media_path = os.path.join(_image_dir, media)
+        inst = modify_media(cim, inst, media_path)
+
+    return PASS
+# main()
+
+if __name__ == "__main__":
+    ret = main()
+
+    if domain:
+        domain.destroy()
+        domain.undefine()
+
+    sys.exit(ret)




More information about the Libvirt-cim mailing list