From 7cef61203742cf07987d2754a3eb3c5cfab4ec58 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=98yvind=20Matheson=20Wergeland?= <oyvind@wergeland.org>
Date: Wed, 14 Aug 2024 21:53:46 +0200
Subject: [PATCH] Test server: support simulating server error and timeout
 (#15)

---
 dumps/test_server.py | 56 +++++++++++++++++++++++++++++---------------
 1 file changed, 37 insertions(+), 19 deletions(-)

diff --git a/dumps/test_server.py b/dumps/test_server.py
index b484e47..74d7839 100644
--- a/dumps/test_server.py
+++ b/dumps/test_server.py
@@ -1,8 +1,15 @@
+import asyncio
+from random import randrange
+
 from aiohttp import web
+import argparse
 import json
 from os import listdir
 from os.path import isfile, join
 
+appliances = {}
+failure_rate = 0
+timeout_rate = 0
 
 async def login(request):
     return web.Response(
@@ -29,12 +36,15 @@ async def token(request):
     )
 
 async def get_appliances(request):
+    if failure_rate > randrange(100):
+        return web.Response(status=500)
+    if timeout_rate > randrange(100):
+        await asyncio.sleep(10.1)
     return web.Response(
         content_type="application/json",
         text=json.dumps(list(appliances.values()))
     )
 
-
 async def update_appliance(request):
     req = await request.json()
     if req["puid"] in appliances:
@@ -57,23 +67,31 @@ async def update_appliance(request):
             text=f'{"resultCode":-1,"kvMap":null,"errorCode":404,"errorDesc":"Unknown puid {req["puid"]}"}'
         )
 
+def main(args):
+    filenames = list(filter(lambda f: f[-5:] == ".json", [f for f in listdir(".") if isfile(join(".", f))]))
+    for filename in filenames:
+        with (open(filename) as f):
+            appliance = json.load(f)
+            appliance["deviceId"] = filename[0:-5]
+            appliance["puid"] = f"puid{appliance['deviceId']}"
+            appliance["deviceNickName"] = f'{appliance["deviceNickName"]} ({appliance["deviceTypeCode"]}-{appliance["deviceFeatureCode"]})'
+            appliances[appliance["puid"]] = appliance
 
+    app = web.Application()
+    app.add_routes([web.post('/accounts.login', login)])
+    app.add_routes([web.post('/accounts.getJWT', get_jwt)])
+    app.add_routes([web.post('/oauth/authorize', authorize)])
+    app.add_routes([web.post('/oauth/token', token)])
+    app.add_routes([web.get('/appliances', get_appliances)])
+    app.add_routes([web.post('/appliances', update_appliance)])
+    web.run_app(app, port=args.port)
 
-filenames = list(filter(lambda f: f[-5:] == ".json", [f for f in listdir(".") if isfile(join(".", f))]))
-appliances = {}
-for filename in filenames:
-    with (open(filename) as f):
-        appliance = json.load(f)
-        appliance["deviceId"] = filename[0:-5]
-        appliance["puid"] = f"puid{appliance['deviceId']}"
-        appliance["deviceNickName"] = f'{appliance["deviceNickName"]} ({appliance["deviceTypeCode"]}-{appliance["deviceFeatureCode"]})'
-        appliances[appliance["puid"]] = appliance
-
-app = web.Application()
-app.add_routes([web.post('/accounts.login', login)])
-app.add_routes([web.post('/accounts.getJWT', get_jwt)])
-app.add_routes([web.post('/oauth/authorize', authorize)])
-app.add_routes([web.post('/oauth/token', token)])
-app.add_routes([web.get('/appliances', get_appliances)])
-app.add_routes([web.post('/appliances', update_appliance)])
-web.run_app(app)
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser(prog='ConnectLife API test server')
+    parser.add_argument('-p', '--port', type=int, default=8080, help='Port on which to serve the web app')
+    parser.add_argument('-f', '--failure_rate', type=int, default=0, help='Failure rate in % for get appliances')
+    parser.add_argument('-t', '--timeout_rate', type=int, default=0, help='Timeout rate in % for get appliances')
+    args = parser.parse_args()
+    failure_rate = args.failure_rate
+    timeout_rate = args.timeout_rate
+    main(args)