109 lines
3.2 KiB
Python
Executable File
109 lines
3.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
import time
|
|
import random
|
|
import re
|
|
import subprocess
|
|
import signal
|
|
import sys
|
|
import os
|
|
|
|
def run(cmd, verbose=False, **kwargs):
|
|
if verbose:
|
|
print " [s] Running %r" % (cmd,)
|
|
p = subprocess.Popen(cmd.split(),
|
|
stdout=subprocess.PIPE,
|
|
**kwargs)
|
|
p.wait()
|
|
|
|
if verbose:
|
|
for line in p.stdout:
|
|
line = p.stdout.readline().strip()
|
|
if line:
|
|
print ' [s] %s' % (line,)
|
|
print " [s] Done"
|
|
time.sleep(0.1)
|
|
|
|
|
|
def spawn(cmd, verbose=False, **kwargs):
|
|
if verbose:
|
|
print " [r] Waiting for %r" % (cmd,)
|
|
p = subprocess.Popen(cmd.split(),
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
**kwargs)
|
|
time.sleep(0.4)
|
|
return p
|
|
|
|
def wait(p, match, verbose=False):
|
|
os.kill(p.pid, signal.SIGINT)
|
|
p.wait()
|
|
r = False
|
|
for line in p.stdout:
|
|
if re.search(match, line):
|
|
r = True
|
|
if verbose:
|
|
print " [r] %s" % (line.strip(),)
|
|
if verbose:
|
|
print " [r] Done"
|
|
return r
|
|
|
|
|
|
|
|
def gen(prog, arg="", **kwargs):
|
|
Prog = ''.join([w.capitalize() for w in prog.split('_')])
|
|
ctx = {
|
|
'python': kwargs.get('python', prog),
|
|
'erlang': kwargs.get('erlang', prog),
|
|
'java': kwargs.get('java', Prog),
|
|
'dotnet': kwargs.get('dotnet', Prog),
|
|
'ruby': kwargs.get('ruby', prog),
|
|
'php': kwargs.get('php', prog),
|
|
'arg': arg,
|
|
'rubyver': os.environ.get('RUBYVER', '1.8'),
|
|
}
|
|
return [
|
|
('python', './venv/bin/python %(python)s.py %(arg)s' % ctx),
|
|
('erlang', './%(python)s.erl %(arg)s' % ctx),
|
|
('java', 'java -cp .:commons-io-1.2.jar:commons-cli-1.1.jar:'
|
|
'rabbitmq-client.jar %(java)s %(arg)s' % ctx),
|
|
('dotnet', 'env MONO_PATH=lib/bin mono %(dotnet)s.exe %(arg)s' % ctx),
|
|
('ruby', 'env RUBYOPT=-rubygems GEM_HOME=gems/gems RUBYLIB=gems/lib '
|
|
'ruby%(rubyver)s %(ruby)s.rb %(arg)s' % ctx),
|
|
('php', 'php %(php)s.php %(arg)s' % ctx),
|
|
]
|
|
|
|
tests = {
|
|
'tut1': (gen('send'), gen('receive', java='Recv'), 'Hello World!'),
|
|
'tut2': (gen('new_task', arg='%(arg)s'), gen('worker'), '%(arg)s'),
|
|
'tut3': (gen('emit_log', arg='%(arg)s'), gen('receive_logs'), '%(arg)s'),
|
|
}
|
|
|
|
|
|
verbose = len(sys.argv) > 1
|
|
errors = 0
|
|
|
|
for test in sorted(tests.keys()):
|
|
(send_progs, recv_progs, output_mask) = tests[test]
|
|
for scwd, send_cmd in send_progs:
|
|
for rcwd, recv_cmd in recv_progs:
|
|
ctx = {
|
|
'arg': 'rand_%s' % (random.randint(1,100),)
|
|
}
|
|
rcmd = recv_cmd % ctx
|
|
scmd = send_cmd % ctx
|
|
mask = output_mask % ctx
|
|
p = spawn(rcmd, verbose=verbose, cwd=rcwd)
|
|
run(scmd, verbose=verbose, cwd=scwd)
|
|
if wait(p, mask, verbose=verbose):
|
|
print " [+] %s %-20s ok" % (test, scwd+'/'+rcwd)
|
|
else:
|
|
print " [!] %s %-20s FAILED %r %r" % \
|
|
(test, scwd+'/'+rcwd, rcmd, scmd)
|
|
errors += 1
|
|
|
|
if errors:
|
|
print " [!] %s tests failed" % (errors,)
|
|
|
|
sys.exit(errors)
|
|
|