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