[Spice-devel] [PATCH] tests/migrate.py: add a migration test

[Dor Laor]

On 01/13/2011 09:35 AM, Hans de Goede wrote:
> Ack.
>
> On 01/13/2011 06:07 AM, Alon Levy wrote:
>> ---
>> tests/migrate.py | 158
>> ++++++++++++++++++++++++++++++++++++++++++++++++++++++

In the past you started with kvm autotest.
Is your direction changed? We have several live migration tests, why not 
incorporate them together?

>> 1 files changed, 158 insertions(+), 0 deletions(-)
>> create mode 100644 tests/migrate.py
>>
>> diff --git a/tests/migrate.py b/tests/migrate.py
>> new file mode 100644
>> index 0000000..c01b107
>> --- /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
>> +
> _______________________________________________
> Spice-devel mailing list
> Spice-devel at lists.freedesktop.org
> http://lists.freedesktop.org/mailman/listinfo/spice-devel