Some more API unit tests

This commit is contained in:
Oliver Walters 2022-04-08 21:59:59 +10:00
parent 8ab54c8e55
commit 78ed5d9cc4

View File

@ -815,6 +815,8 @@ class PartDetailTests(InvenTreeAPITestCase):
Test that the required details are available Test that the required details are available
""" """
p = Part.objects.get(pk=1)
url = reverse('api-part-detail', kwargs={'pk': 1}) url = reverse('api-part-detail', kwargs={'pk': 1})
data = self.get(url, expected_code=200).data data = self.get(url, expected_code=200).data
@ -834,6 +836,7 @@ class PartDetailTests(InvenTreeAPITestCase):
on_order -= line.received on_order -= line.received
self.assertEqual(on_order, data['ordering']) self.assertEqual(on_order, data['ordering'])
self.assertEqual(on_order, p.on_order)
# Some other checks # Some other checks
self.assertEqual(data['in_stock'], 9000) self.assertEqual(data['in_stock'], 9000)
@ -1157,6 +1160,12 @@ class BomItemTest(InvenTreeAPITestCase):
self.assertEqual(len(response.data), 1) self.assertEqual(len(response.data), 1)
self.assertEqual(response.data[0]['pk'], bom_item.pk) self.assertEqual(response.data[0]['pk'], bom_item.pk)
# Each item in response should contain expected keys
for el in response.data:
for key in ['available_stock', 'available_substitute_stock']:
self.assertTrue(key in el)
def test_get_bom_detail(self): def test_get_bom_detail(self):
""" """
Get the detail view for a single BomItem object Get the detail view for a single BomItem object
@ -1166,6 +1175,26 @@ class BomItemTest(InvenTreeAPITestCase):
response = self.get(url, expected_code=200) response = self.get(url, expected_code=200)
expected_values = [
'allow_variants',
'inherited',
'note',
'optional',
'overage',
'pk',
'part',
'quantity',
'reference',
'sub_part',
'substitutes',
'validated',
'available_stock',
'available_substitute_stock',
]
for key in expected_values:
self.assertTrue(key in response.data)
self.assertEqual(int(float(response.data['quantity'])), 25) self.assertEqual(int(float(response.data['quantity'])), 25)
# Increase the quantity # Increase the quantity