[libvirt] [libvirt-test-API] [PATCH] Add test cases for API defineXMLFlags

[Shanzhi Yu]

Signed-off-by: Shanzhi Yu <shyu at redhat.com>
---
 cases/linux_domain.conf         |  22 +++++++++
 repos/domain/definewithflags.py | 104 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 126 insertions(+)
 create mode 100644 repos/domain/definewithflags.py

diff --git a/cases/linux_domain.conf b/cases/linux_domain.conf
index 6315613..7dd9062 100644
--- a/cases/linux_domain.conf
+++ b/cases/linux_domain.conf
@@ -214,6 +214,28 @@ domain:define
     macaddr
         54:52:00:45:c3:8a
 
+domain:undefine
+    guestname
+        $defaultname
+
+domain:definewithflags
+    guestname
+        $defaultname
+    diskpath
+        /var/lib/libvirt/images/libvirt-test-api
+    vcpu
+        1
+    memory
+        1048576
+    hddriver
+        virtio
+    nicdriver
+        virtio
+    macaddr
+        54:52:00:45:c3:8a
+    flags
+        validate
+
 domain:start
     guestname
         $defaultname
diff --git a/repos/domain/definewithflags.py b/repos/domain/definewithflags.py
new file mode 100644
index 0000000..01a4f15
--- /dev/null
+++ b/repos/domain/definewithflags.py
@@ -0,0 +1,104 @@
+#!/usr/bin/evn python
+
+import os
+import re
+import sys
+import commands
+import string
+import pexpect
+
+import libvirt
+from libvirt import libvirtError
+
+from src import sharedmod
+from utils import utils
+
+required_params = ('guestname', 'diskpath', 'flags',)
+optional_params = {'memory': 1048576,
+                   'vcpu': 1,
+                   'imageformat' : 'raw',
+                   'hddriver' : 'virtio',
+                   'nicdriver': 'virtio',
+                   'macaddr': '52:54:00:97:e4:28',
+                   'uuid' : '05867c1a-afeb-300e-e55e-2673391ae080',
+                   'username': None,
+                   'password': None,
+                   'virt_type': 'kvm',
+                   'xml': 'xmls/kvm_guest_define.xml',
+                   'guestarch': 'x86_64',
+                   'guestmachine': 'pc',
+                  }
+
+def check_define_domain(guestname, virt_type, hostname, username, \
+                        password, logger):
+    """Check define domain result, if define domain is successful,
+       guestname.xml will exist under /etc/libvirt/qemu/
+       and can use virt-xml-validate tool to check the file validity
+    """
+    if "kvm" in virt_type:
+        path = "/etc/libvirt/qemu/%s.xml" % guestname
+    elif "xen" in virt_type:
+        path = "/etc/xen/%s" % guestname
+    else:
+        logger.error("unknown virt type")
+
+    if hostname:
+        cmd = "ls %s" % path
+        ret, output = utils.remote_exec_pexpect(hostname, username, \
+                                               password, cmd)
+        if ret:
+            logger.error("guest %s xml file doesn't exsits" % guestname)
+            return False
+        else:
+            return True
+    else:
+        if os.access(path, os.R_OK):
+            return True
+        else:
+            return False
+
+def definewithflags(params):
+    """Define a domain from xml"""
+    logger = params['logger']
+    guestname = params['guestname']
+
+    xmlstr = params['xml']
+    logger.debug("domain xml:\n%s" % xmlstr)
+
+    conn = sharedmod.libvirtobj['conn']
+    uri = conn.getURI()
+
+    hostname = utils.parse_uri(uri)[1]
+    username = params.get('username', '')
+    password = params.get('password', '')
+    virt_type = params.get('virt_type', 'kvm')
+
+    flags = params['flags']
+    logger.info("The flags are %s" % flags)
+    flags_string = flags.split("|")
+    flags = 0
+
+    logger.info("define domain on %s" % uri)
+
+    for flag in flags_string:
+        if flag == 'validate':
+            flags |= libvirt.VIR_DOMAIN_DEFINE_VALIDATE
+        else:
+            logger.error("unknow flags")
+            return 1
+
+    # Define domain from xml
+    try:
+        conn.defineXMLFlags(xmlstr, flags)
+        if check_define_domain(guestname, virt_type, hostname, \
+                               username, password, logger):
+            logger.info("define a domain from xml is successful")
+        else:
+            logger.error("fail to check define domain")
+            return 1
+    except libvirtError, e:
+        logger.error("API error message: %s, error code is %s" \
+                     % (e.message, e.get_error_code()))
+        return 1
+
+    return 0
-- 
1.8.3.1