From ea5b888ecc7d52e07a6ca8c5aa2491cf069e41c8 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Fri, 3 Nov 2023 10:49:48 +0100
Subject: [PATCH] Repair automatic session restart, which was causing test
 client problems; corrected return value of model query

---
 tests/test_api.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_api.py b/tests/test_api.py
index 3cde9af5..c81c2d02 100644
--- a/tests/test_api.py
+++ b/tests/test_api.py
@@ -19,8 +19,8 @@ class TestServerBaseClass(unittest.TestCase):
             daemon=True
         )
         self.uri = f'http://{host}:{port}/'
-        requests.get(self.uri + 'restart')
         self.server_process.start()
+        requests.get(self.uri + 'restart')
 
     def copy_input_file_to_server(self):
         from shutil import copyfile
@@ -65,7 +65,7 @@ class TestApiFromAutomatedClient(TestServerBaseClass):
         resp_load = requests.put(
             self.uri + f'models/dummy/load',
         )
-        model_id = resp_load.json()['model_id']
+        model_id = list(resp_load.json()['model_id'].keys())[0]
         self.assertEqual(resp_load.status_code, 200, resp_load.json())
         resp_list = requests.get(self.uri + 'models')
         self.assertEqual(resp_list.status_code, 200)
-- 
GitLab