Fixed the text for the plac server

1 files changed, 34 insertions, 0 deletions

diff --git a/plac/doc/test_server.py b/plac/doc/test_server.py
new file mode 100644
index 0000000..5567f40
--- /dev/null
+++ b/plac/doc/test_server.py
@@ -0,0 +1,34 @@
+import multiprocessing, subprocess, time, random
+import plac
+from ishelve2 import ShelveInterface
+
+i = plac.Interpreter(ShelveInterface())
+
+COMMANDS = ['''\
+.help
+set a 1
+''',
+'''\
+set b 1
+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()