From 3488da19b024b94f27ab8d9f71588fe00cf70eca Mon Sep 17 00:00:00 2001 From: Oliver Walters Date: Tue, 17 May 2022 23:47:09 +1000 Subject: [PATCH] Return data as a list of dict objects --- InvenTree/InvenTree/api_tester.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/InvenTree/InvenTree/api_tester.py b/InvenTree/InvenTree/api_tester.py index 4fb6a9e652..1fa2816653 100644 --- a/InvenTree/InvenTree/api_tester.py +++ b/InvenTree/InvenTree/api_tester.py @@ -242,4 +242,15 @@ class InvenTreeAPITestCase(APITestCase): if required_rows is not None: self.assertEqual(len(rows), required_rows) - return (headers, rows) + # Return the file data as a list of dict items, based on the headers + data = [] + + for row in rows: + entry = {} + + for idx, col in enumerate(headers): + entry[col] = row[idx] + + data.append(entry) + + return data