mirror of
https://github.com/kennethreitz-archive/plac.git
synced 2026-06-05 23:50:18 +00:00
Kenneth Reitz
36 lines
893 B
Python
36 lines
893 B
Python
import multiprocessing, subprocess, time, random
|
|
import plac
|
|
from ishelve2 import ShelveInterface
|
|
|
|
i = plac.Interpreter(ShelveInterface(configfile=None))
|
|
|
|
COMMANDS = ['''\
|
|
.help
|
|
set a 1
|
|
''',
|
|
'''\
|
|
set b 1
|
|
wrong command
|
|
showall
|
|
''']
|
|
|
|
def client_send(commands, port):
|
|
time.sleep(.5) # wait a bit for the server to start
|
|
po = subprocess.Popen(['telnet', 'localhost', str(port)],
|
|
stdin=subprocess.PIPE)
|
|
for cmd in commands.splitlines():
|
|
po.stdin.write(cmd + '\n')
|
|
time.sleep(.1) # wait a bit for the server to answer
|
|
|
|
def test():
|
|
port = random.choice(range(2000, 20000))
|
|
clients = []
|
|
for cmds in COMMANDS:
|
|
cl = multiprocessing.Process(target=client_send, args=(cmds, port))
|
|
clients.append(cl)
|
|
cl.start()
|
|
i.stop_server(wait=1)
|
|
i.start_server(port, timeout=.1)
|
|
for cl in clients:
|
|
cl.join()
|