add message_size to test suites api response

att · Jan 31, 2024 · 210e933 · 210e933
1 parent 3ca7010
commit 210e933
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/api/src/utils/test_suite_serializer.py b/api/src/utils/test_suite_serializer.py
@@ -35,6 +35,7 @@ def __get_test_runs_metrics(test_runs):
             "id": test_run.id,
             "algorithm": test_run.algorithm,
             "iterations": test_run.iterations,
+            "message_size": test_run.message_size,
             "results": {
                 "averageCPU": round(cpu_avg, 2),
                 "averageMemory": int(memory_avg),

diff --git a/api/tests/test_tests_api.py b/api/tests/test_tests_api.py
@@ -70,7 +70,7 @@ def test_get_test_suite(self):
         self.app.database_manager.get_by_id.return_value = test_suite
         response = self.client.get(TEST_SUITES_GET_URL)
         result = json.loads(response.data)
-        expected = {'code_release': '1.1.0', 'description': 'description', 'end_time': None, 'environment_info': {'cpu': None, 'cpuArchitecture': None, 'cpuClockSpeed': None, 'cpuCores': None, 'nodeSize': None, 'operatingSystem': None, 'resourceName': None}, 'id': None, 'name': 'name', 'start_time': None, 'test_runs': [{'algorithm': None, 'id': 1, 'iterations': None, 'results': {'averageCPU': 9.0, 'averageMemory': 14}}]}
+        expected = {'code_release': '1.1.0', 'description': 'description', 'end_time': None, 'environment_info': {'cpu': None, 'cpuArchitecture': None, 'cpuClockSpeed': None, 'cpuCores': None, 'nodeSize': None, 'operatingSystem': None, 'resourceName': None}, 'id': None, 'name': 'name', 'start_time': None, 'test_runs': [{'algorithm': None, 'id': 1, 'iterations': None, 'message_size': None, 'results': {'averageCPU': 9.0, 'averageMemory': 14}}]}
         self.assertEqual(result, expected)
 
     def test_get_test_suite_return_not_found(self):