1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
#!/usr/bin/python
"""
Example usage:
./test_multiple.py test_display_no_ssl --log test.log
Resulting test.log file (single test, stopped with Ctrl-C with 15 clients):
0 162212
1 154156
2 154424
3 154952
4 155352
5 155616
6 156280
7 222084
8 222612
9 230820
10 230820
11 230820
12 230820
13 296356
14 296356
"""
import argparse
import os
import sys
import subprocess
import atexit
import socket
import time
def killall(procs):
for p in procs:
print "killing %d" % p.pid
p.kill()
p.wait()
def cleanup():
killall(clients + [test_process])
def wait_for_port(port):
if not port:
return
# TODO: do this without actually opening the port - maybe just look at /proc/qemu_process_id/fd?
s = socket.socket(socket.AF_INET)
while True:
try:
s.connect(('localhost', port))
s.close()
break
except:
time.sleep(1)
pass
def get_vm_size(pid):
""" read from /proc/<pid>/status, VmSize, in KiloBytes """
return int([x for x in open('/proc/%s/status' % pid).readlines() if 'VmSize' in x][0].split()[1])
parser = argparse.ArgumentParser()
parser.add_argument('--client', default='spicy')
parser.add_argument('--start-count', default=1, type=int)
parser.add_argument('--end-count', default=50, type=int)
parser.add_argument('--log', default='-')
parser.add_argument('--sleep', default=3, type=int)
args, rest = parser.parse_known_args(sys.argv[1:])
client = os.popen('which %s' % args.client).read().strip()
if not os.path.exists(client):
print "supply a valid client. %s does not exist" % (args.client)
sys.exit(1)
if not rest or len(rest) < 1 or not os.path.exists(rest[0]):
print "supply one argument that is the tester you wish to run"
sys.exit(1)
prog = rest[0]
port = {
'test_display_no_ssl': 5912,
'test_display_streaming': 5912,
'test_just_sockets_no_ssl': 5912,
'test_playback': 5701,
}.get(prog, None)
if args.log == '-':
log = sys.stdout
else:
log = open(args.log, 'a+')
log.write('#%s\n' % time.time())
# kill leftovers from previous invocation
os.system('killall lt-%s' % prog)
if prog[0] != '/':
prog = os.path.join('.', prog)
if not port:
print "unknown port for %r" % prog
print "prog = %r" % prog
print "client = %r" % client
print "range = %d..%d" % (args.start_count, args.end_count)
atexit.register(cleanup)
os.environ['SPICE_DEBUG_ALLOW_MC'] = '1'
test_process = subprocess.Popen([prog], executable=prog)
wait_for_port(port)
for count in xrange(args.start_count, args.end_count):
print "starting %d clients" % count
clients = [subprocess.Popen(args=[client, '-h', 'localhost', '-p', str(port)],
executable=client) for i in xrange(count)]
print "sleeping %d" % (args.sleep * count)
time.sleep(args.sleep * count)
vmsize = "%d %d" % (i, get_vm_size(test_process.pid))
print vmsize
log.write(vmsize + '\n')
log.flush()
killall(clients)
test_process.wait()
|
