From 5be8d7ce70249b6d5c0523de6a4f8be0261d714a Mon Sep 17 00:00:00 2001 From: Oliver Walters Date: Sat, 28 Dec 2024 01:56:50 +0000 Subject: [PATCH] Fix unit tests --- src/backend/InvenTree/generic/states/tests.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/src/backend/InvenTree/generic/states/tests.py b/src/backend/InvenTree/generic/states/tests.py index f936cd8012..238d1d2d64 100644 --- a/src/backend/InvenTree/generic/states/tests.py +++ b/src/backend/InvenTree/generic/states/tests.py @@ -228,11 +228,13 @@ class ApiTests(InvenTreeAPITestCase): def test_all_states(self): """Test the API endpoint for listing all status models.""" response = self.get(reverse('api-status-all')) - self.assertEqual(len(response.data), 12) + + # 10 built-in state classes, plus the added GeneralState class + self.assertEqual(len(response.data), 11) # Test the BuildStatus model build_status = response.data['BuildStatus'] - self.assertEqual(build_status['class'], 'BuildStatus') + self.assertEqual(build_status['status_class'], 'BuildStatus') self.assertEqual(len(build_status['values']), 5) pending = build_status['values']['PENDING'] self.assertEqual(pending['key'], 10) @@ -241,7 +243,7 @@ class ApiTests(InvenTreeAPITestCase): # Test the StockStatus model (static) stock_status = response.data['StockStatus'] - self.assertEqual(stock_status['class'], 'StockStatus') + self.assertEqual(stock_status['status_class'], 'StockStatus') self.assertEqual(len(stock_status['values']), 8) in_stock = stock_status['values']['OK'] self.assertEqual(in_stock['key'], 10) @@ -249,8 +251,8 @@ class ApiTests(InvenTreeAPITestCase): self.assertEqual(in_stock['label'], 'OK') # MachineStatus model - machine_status = response.data['MachineStatus__LabelPrinterStatus'] - self.assertEqual(machine_status['class'], 'LabelPrinterStatus') + machine_status = response.data['LabelPrinterStatus'] + self.assertEqual(machine_status['status_class'], 'LabelPrinterStatus') self.assertEqual(len(machine_status['values']), 6) connected = machine_status['values']['CONNECTED'] self.assertEqual(connected['key'], 100) @@ -267,10 +269,11 @@ class ApiTests(InvenTreeAPITestCase): reference_status='StockStatus', ) response = self.get(reverse('api-status-all')) - self.assertEqual(len(response.data), 12) + + self.assertEqual(len(response.data), 11) stock_status_cstm = response.data['StockStatus'] - self.assertEqual(stock_status_cstm['class'], 'StockStatus') + self.assertEqual(stock_status_cstm['status_class'], 'StockStatus') self.assertEqual(len(stock_status_cstm['values']), 9) ok_advanced = stock_status_cstm['values']['OK'] self.assertEqual(ok_advanced['key'], 10)