From 4e1f38d8d95b02f8eefb6c01b779cd588d52966a Mon Sep 17 00:00:00 2001
From: theoreticalbts <theoreticalbts@users.noreply.github.com>
Date: Wed, 12 Aug 2015 15:36:35 -0400
Subject: [PATCH] api_stress.py:  Implement simple stress testing of API

---
 tests/intense/api_stress.py | 75 +++++++++++++++++++++++++++++++++++++
 1 file changed, 75 insertions(+)
 create mode 100755 tests/intense/api_stress.py

diff --git a/tests/intense/api_stress.py b/tests/intense/api_stress.py
new file mode 100755
index 00000000..3aa1f2d3
--- /dev/null
+++ b/tests/intense/api_stress.py
@@ -0,0 +1,75 @@
+#!/usr/bin/env python3
+
+import json
+import os
+import random
+import signal
+import struct
+import sys
+import time
+
+try:
+    import asyncio
+except ImportError:
+    print("asyncio module not found (try pip install asyncio, or upgrade to Python 3.4 or later)")
+    sys.exit(1)
+
+try:
+    import websockets
+except ImportError:
+    print("websockets module not found (try pip install websockets)")
+    sys.exit(1)
+
+@asyncio.coroutine
+def mainloop():
+    ws = yield from websockets.connect('ws://localhost:8090/')
+    entropy = struct.unpack("<Q", os.urandom(8))[0]
+    rand = random.Random(entropy)
+    my_account_id = rand.randrange(0, 90000)
+
+    gpo_get = {"id":1, "method":"call", "params":[0, "get_objects", ["2.0.0"]]}
+    dgpo_sub = {"id":2, "method":"call", "params":[0, "subscribe_to_objects", [111, ["2.1.0"]]]}
+    acct_sub = {"id":3, "method":"call", "params":[0, "get_full_accounts", [222, ["1.2."+str(my_account_id)], True]]}
+    yield from ws.send(json.dumps(gpo_get))
+    yield from ws.send(json.dumps(dgpo_sub))
+    yield from ws.send(json.dumps(acct_sub))
+
+    next_call_id = 4
+
+    def peek_random_account():
+        nonlocal next_call_id
+        asyncio.get_event_loop().call_later(rand.uniform(0, 3), peek_random_account)
+        peek_account_id = rand.randrange(0, 90000)
+        acct_peek = {"id" : next_call_id, "method" : "call", "params":
+           [0, "get_objects", [["1.2."+str(peek_account_id)], True]]}
+        next_call_id += 1
+        yield from ws.send(json.dumps(acct_peek))
+        return
+
+    yield from peek_random_account()
+
+    while True:
+        result = yield from ws.recv()
+        #print(result)
+        if result is None:
+           break
+    yield from ws.close()
+
+child_procs = []
+
+# stress test with 200 instances
+while len(child_procs) < 200:
+    pid = os.fork()
+    if pid == 0:
+        asyncio.get_event_loop().run_until_complete(mainloop())
+        break
+    else:
+        child_procs.append(pid)
+
+time.sleep(60)
+for pid in child_procs:
+    os.kill(pid, signal.SIGTERM)
+
+time.sleep(2)
+for pid in child_procs:
+    os.kill(pid, signal.SIGKILL)