diff options
Diffstat (limited to 'plac/doc/test_server.py')
| -rw-r--r-- | plac/doc/test_server.py | 18 |
1 files changed, 11 insertions, 7 deletions
diff --git a/plac/doc/test_server.py b/plac/doc/test_server.py index a9904ee..81214ee 100644 --- a/plac/doc/test_server.py +++ b/plac/doc/test_server.py @@ -14,22 +14,26 @@ wrong command showall '''] -def client_send(commands, port): +def telnet(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 + try: + for cmd in commands.splitlines(): + po.stdin.write(cmd + '\n') + time.sleep(.1) # wait a bit for the server to answer + finally: + po.stdin.close() def test(): port = random.choice(range(2000, 20000)) clients = [] for cmds in COMMANDS: - cl = multiprocessing.Process(target=client_send, args=(cmds, port)) + cl = multiprocessing.Process(target=telnet, args=(cmds, port)) clients.append(cl) cl.start() - i.stop_server(wait=1) - i.start_server(port, timeout=.1) + + i.start_server(port, timeout=.5) for cl in clients: cl.join() + i.stop_server() |