inventree/InvenTree
2
0
Fork 0
mirror of https://github.com/inventree/InvenTree.git synced 2025-08-08 04:40:57 +00:00
Code Activity

Fix unit tests

Browse Source
This commit is contained in:
Oliver Walters
2024-12-28 01:56:50 +00:00
parent cf9d71f86c
commit 5be8d7ce70
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/backend/InvenTree/generic/states/tests.py
View File

@@ -228,11 +228,13 @@ class ApiTests(InvenTreeAPITestCase):
def test_all_states(self): def test_all_states(self):
"""Test the API endpoint for listing all status models.""" """Test the API endpoint for listing all status models."""
response = self.get(reverse('api-status-all')) 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 # Test the BuildStatus model
build_status = response.data['BuildStatus'] 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) self.assertEqual(len(build_status['values']), 5)
pending = build_status['values']['PENDING'] pending = build_status['values']['PENDING']
self.assertEqual(pending['key'], 10) self.assertEqual(pending['key'], 10)
@@ -241,7 +243,7 @@ class ApiTests(InvenTreeAPITestCase):
# Test the StockStatus model (static) # Test the StockStatus model (static)
stock_status = response.data['StockStatus'] 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) self.assertEqual(len(stock_status['values']), 8)
in_stock = stock_status['values']['OK'] in_stock = stock_status['values']['OK']
self.assertEqual(in_stock['key'], 10) self.assertEqual(in_stock['key'], 10)
@@ -249,8 +251,8 @@ class ApiTests(InvenTreeAPITestCase):
self.assertEqual(in_stock['label'], 'OK') self.assertEqual(in_stock['label'], 'OK')
# MachineStatus model # MachineStatus model
machine_status = response.data['MachineStatus__LabelPrinterStatus'] machine_status = response.data['LabelPrinterStatus']
self.assertEqual(machine_status['class'], 'LabelPrinterStatus') self.assertEqual(machine_status['status_class'], 'LabelPrinterStatus')
self.assertEqual(len(machine_status['values']), 6) self.assertEqual(len(machine_status['values']), 6)
connected = machine_status['values']['CONNECTED'] connected = machine_status['values']['CONNECTED']
self.assertEqual(connected['key'], 100) self.assertEqual(connected['key'], 100)
@@ -267,10 +269,11 @@ class ApiTests(InvenTreeAPITestCase):
reference_status='StockStatus', reference_status='StockStatus',
) )
response = self.get(reverse('api-status-all')) 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'] 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) self.assertEqual(len(stock_status_cstm['values']), 9)
ok_advanced = stock_status_cstm['values']['OK'] ok_advanced = stock_status_cstm['values']['OK']
self.assertEqual(ok_advanced['key'], 10) self.assertEqual(ok_advanced['key'], 10)