tests/migrate.py: add a migration test

1 files changed, 158 insertions, 0 deletions

diff --git a/tests/migrate.py b/tests/migrate.py
new file mode 100644
index 00000000..c01b1073
--- /dev/null
+++ b/tests/migrate.py
@@ -0,0 +1,158 @@
+"""
+Spice Migration test
+
+Somewhat stressfull test of continuous migration with spice in VGA mode or QXL mode,
+depends on supplying an image in IMAGE variable (if no image is supplied then
+VGA mode since it will just be SeaBIOS).
+
+Dependencies:
+either qmp in python path or running with spice and qemu side by side:
+qemu/QMP/qmp.py
+spice/tests/migrate.py
+
+Will create two temporary unix sockets in /tmp
+Will leave a log file, migrate_test.log, in current directory.
+"""
+
+#
+# start one spiceclient, have two machines (active and target),
+# and repeat:
+#  active wait until it's active
+#  active client_migrate_info
+#  active migrate tcp:localhost:9000
+#  _wait for event of quit
+#  active stop, active<->passive
+# 
+# wait until it's active
+#  command query-status, if running good
+#  if not listen to events until event of running
+
+try:
+    import qmp
+except:
+    import sys
+    sys.path.append("../../qemu/QMP")
+    try:
+        import qmp
+    except:
+        print "can't find qmp"
+        raise SystemExit
+from subprocess import Popen, PIPE
+import os
+import time
+import socket
+import datetime
+import atexit
+
+QMP_1, QMP_2 = "/tmp/migrate_test.1.qmp", "/tmp/migrate_test.2.qmp"
+SPICE_PORT_1, SPICE_PORT_2 = 5911, 6911
+MIGRATE_PORT = 9000
+QEMU = "qemu.upstream"
+LOG_FILENAME = "migrate_log.log"
+IMAGE = "/store/images/f14_regular.qcow2"
+
+qemu_exec = os.popen("which %s" % QEMU).read().strip()
+
+def start_qemu(spice_port, qmp_filename, incoming_port=None):
+    incoming_args = []
+    if incoming_port:
+        incoming_args = ("-incoming tcp::%s" % incoming_port).split()
+    args = ([qemu_exec, "-qmp", "unix:%s,server,nowait" % qmp_filename,
+        "-spice", "disable-ticketing,port=%s" % spice_port]
+        + incoming_args)
+    if os.path.exists(IMAGE):
+        args += ["-m", "512", "-drive",
+                 "file=%s,index=0,media=disk,cache=unsafe" % IMAGE, "-snapshot"]
+    proc = Popen(args, executable=qemu_exec, stdin=PIPE, stdout=PIPE)
+    while not os.path.exists(qmp_filename):
+        time.sleep(0.1)
+    proc.qmp_filename = qmp_filename
+    proc.qmp = qmp.QEMUMonitorProtocol(qmp_filename)
+    while True:
+        try:
+            proc.qmp.connect()
+            break
+        except socket.error, err:
+            pass
+    proc.spice_port = spice_port
+    proc.incoming_port = incoming_port
+    return proc
+
+def start_spicec(spice_port):
+    return Popen(("spicec -h localhost -p %s" % spice_port).split(), executable="spicec")
+
+def wait_active(q, active):
+    events = ["RESUME"] if active else ["STOP"]
+    while True:
+        try:
+            ret = q.cmd("query-status")
+        except:
+            # ValueError
+            time.sleep(0.1)
+            continue
+        if ret and ret.has_key("return"):
+            if ret["return"]["running"] == active:
+                break
+        for e in q.get_events():
+            if e["event"] in events:
+                break
+        time.sleep(0.5)
+
+def wait_for_event(q, event):
+    while True:
+        for e in q.get_events():
+            if e["event"] == event:
+                return
+        time.sleep(0.5)
+
+def cleanup(*args):
+    print "doing cleanup"
+    os.system("killall %s" % qemu_exec)
+    for x in [QMP_1, QMP_2]:
+        if os.path.exists(x):
+            os.unlink(x)
+
+#################### Main #######################
+
+cleanup()
+atexit.register(cleanup)
+
+active = start_qemu(spice_port=SPICE_PORT_1, qmp_filename=QMP_1)
+target = start_qemu(spice_port=SPICE_PORT_2, qmp_filename=QMP_2,
+                    incoming_port=MIGRATE_PORT)
+
+i = 0
+spicec = None
+log = open(LOG_FILENAME, "a+")
+log.write("# "+str(datetime.datetime.now())+"\n")
+
+while True:
+    wait_active(active.qmp, True)
+    wait_active(target.qmp, False)
+    if spicec == None:
+        spicec = start_spicec(spice_port=SPICE_PORT_1)
+        wait_for_event(active.qmp, 'SPICE_INITIALIZED')
+        print "waiting for Enter to start migrations"
+        raw_input()
+    active.qmp.cmd('client_migrate_info', {'protocol':'spice',
+        'hostname':'localhost', 'port':target.spice_port})
+    active.qmp.cmd('migrate', {'uri': 'tcp:localhost:%s' % MIGRATE_PORT})
+    wait_active(active.qmp, False)
+    wait_active(target.qmp, True)
+    wait_for_event(target.qmp, 'SPICE_CONNECTED')
+    dead = active
+    dead.qmp.cmd("quit")
+    dead.qmp.close()
+    dead.wait()
+    new_spice_port = dead.spice_port
+    new_qmp_filename = dead.qmp_filename
+    log.write("# STDOUT dead %s\n" % dead.pid)
+    log.write(dead.stdout.read())
+    del dead
+    active = target
+    target = start_qemu(spice_port=new_spice_port,
+                        qmp_filename=new_qmp_filename,
+                        incoming_port=MIGRATE_PORT)
+    print i
+    i += 1
+