Implement performance testing script

1 files changed, 44 insertions, 0 deletions

diff --git a/tools/test-performance.py b/tools/test-performance.py
new file mode 100755
index 000000000..ec31e9b86
--- /dev/null
+++ b/tools/test-performance.py
@@ -0,0 +1,44 @@
+#!/usr/bin/python3
+
+import asyncio
+import random
+
+host = 'localhost'
+port = 5027
+
+messageLogin = bytearray.fromhex('000F313233343536373839303132333435')
+messageLocation = bytearray.fromhex('000000000000002b080100000140d4e3ec6e000cc661d01674a5e0fffc00000900000004020100f0000242322318000000000100007a04')
+
+devices = 100
+period = 1
+
+
+class AsyncClient(asyncio.Protocol):
+
+    def __init__(self, loop):
+        self.loop = loop
+        self.buffer = memoryview(messageLogin)
+
+    def connection_made(self, transport):
+        self.send_message(transport)
+
+    def send_message(self, transport):
+        transport.write(self.buffer)
+        self.buffer = memoryview(messageLocation)
+        delay = period * (0.9 + 0.2 * random.random())
+        self.loop.call_later(delay, self.send_message, transport)
+
+    def data_received(self, data):
+        pass
+
+    def connection_lost(self, exc):
+        self.loop.stop()
+
+
+loop = asyncio.get_event_loop()
+
+for i in range(0, devices):
+    loop.create_task(loop.create_connection(lambda: AsyncClient(loop), host, port))
+
+loop.run_forever()
+loop.close()