/
test-cluster-deployment-of-session-baskets.py
executable file
·123 lines (101 loc) · 3.53 KB
/
test-cluster-deployment-of-session-baskets.py
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
122
123
#!/usr/bin/env python
import os
import sys
import time
import random
import threading
from twill import get_browser, commands
#from lxml import etree
#from StringIO import StringIO
import subprocess
g_jbosses = [None, None]
g_waitTimeBeforeKillings = 25
def panic(x):
if not x.endswith("\n"):
x += "\n"
sys.stderr.write("\n"+chr(27)+"[32m" + x + chr(27) + "[0m\n")
sys.exit(1)
def mysystem(cmd):
if 0 != os.system(cmd):
panic("Failed to execute '%s'" % cmd)
def listOfRandomWords():
from string import ascii_uppercase
return ["".join(random.choice(ascii_uppercase) for _ in xrange(15)) for unused in xrange(500)]
def KillJBoss(deadBossIndex):
print "Stopping JBoss", deadBossIndex
g_jbosses[deadBossIndex].kill()
print "Confirming shutdown of JBoss", deadBossIndex
g_jbosses[deadBossIndex].wait()
print "Confirmed."
def StartJBoss(newBossIndex):
# Don't use os.system - see my SO question
# http://stackoverflow.com/questions/6773456/why-is-this-code-behaving-differently-across-various-distros-unixes
g_jbosses[newBossIndex] = \
subprocess.Popen("./launch-node" + str(newBossIndex+1), shell=True, executable="/bin/bash")
def SpawnDistributedCluster():
lines = filter(lambda x: "java" in x and "boss" in x and "230" in x, os.popen("COLUMNS=4000 ps aux").readlines())
if 0 != len(lines):
panic("Please stop your JBoss instance in the 230... multicast group")
StartJBoss(0)
StartJBoss(1)
class SpawnKillerThread(threading.Thread):
def run(self):
self.stopKilling = False
while True:
for _ in xrange(g_waitTimeBeforeKillings):
time.sleep(1)
if self.stopKilling:
return
if os.path.exists("/tmp/nokill"):
time.sleep(1)
continue
deadBossIndex = random.randint(0, 1)
KillJBoss(deadBossIndex)
StartJBoss(deadBossIndex)
def main():
joker = SpawnKillerThread()
try:
SpawnDistributedCluster()
print "Waiting for two JBoss instances to start up in cluster mode..."
time.sleep(10)
joker.start()
#commands.debug('http', 1)
#commands.debug('commands',1)
b = get_browser()
import socket
myip = "http://" + socket.gethostbyname(socket.gethostname())
b.go(myip + "/session-basket")
b.submit(5)
wordBatch = listOfRandomWords()
for idx, w in enumerate(wordBatch):
print "Adding word %d: '%s' to the list..." % (idx, w)
commands.formvalue("1", "newItem:j_idt11", w)
b.submit(4)
commands.code(200)
commands.find(w)
time.sleep(0.25)
print wordBatch, len(wordBatch)
b.submit(5)
commands.code(200)
commands.reload()
for w in wordBatch:
commands.find(w)
finally:
joker.stopKilling = True
joker.join()
KillJBoss(0)
KillJBoss(1)
#commands.save_html("file")
#html = b.get_html()
#parser = etree.HTMLParser()
#tree = etree.parse(StringIO(html), parser)
#root = tree.getroot()
#print root.xpath("//span[@id='spot-price-gold-base']")[0].text
#for t in root.xpath("a:file/a:segments/a:segment", namespaces={'a': 'http://path.to/DTD/2003/whatever'}):
# f = t.getparent().getparent()
# s = f.get("subject")
# if not s:
# panic("Missing subject...")
# sizes[s] += long(t.attrib["bytes"])
if __name__ == "__main__":
main()