Rebuild tree structure before running unit tests!

This commit is contained in:
Oliver Walters 2022-04-02 13:26:39 +11:00
parent fb0b87db3d
commit 271cb3f8b1
2 changed files with 7 additions and 4 deletions

View File

@ -1434,7 +1434,7 @@ class Part(MPTTModel):
- If this part is a "template" (variants exist) then these are counted too - If this part is a "template" (variants exist) then these are counted too
""" """
return self.get_stock_count() return self.get_stock_count(include_variants=True)
def get_bom_item_filter(self, include_inherited=True): def get_bom_item_filter(self, include_inherited=True):
""" """

View File

@ -833,6 +833,9 @@ class PartAPIAggregationTest(InvenTreeAPITestCase):
super().setUp() super().setUp()
# Ensure the part "variant" tree is correctly structured
Part.objects.rebuild()
# Add a new part # Add a new part
self.part = Part.objects.create( self.part = Part.objects.create(
name='Banana', name='Banana',
@ -914,7 +917,7 @@ class PartAPIAggregationTest(InvenTreeAPITestCase):
self.assertEqual(part.build_order_allocation_count(), 0) self.assertEqual(part.build_order_allocation_count(), 0)
self.assertEqual(part.sales_order_allocation_count(), 0) self.assertEqual(part.sales_order_allocation_count(), 0)
self.assertEqual(part.total_stock, in_stock) self.assertEqual(part.total_stock, in_stock)
self.assertEqual(part.available_stock(), in_stock) self.assertEqual(part.available_stock, in_stock)
# Now, let's create a sales order, and allocate some stock # Now, let's create a sales order, and allocate some stock
so = order.models.SalesOrder.objects.create( so = order.models.SalesOrder.objects.create(
@ -1010,7 +1013,7 @@ class PartAPIAggregationTest(InvenTreeAPITestCase):
self.assertEqual(part.build_order_allocation_count(), 10) self.assertEqual(part.build_order_allocation_count(), 10)
self.assertEqual(part.sales_order_allocation_count(), 15) self.assertEqual(part.sales_order_allocation_count(), 15)
self.assertEqual(part.total_stock, 91) self.assertEqual(part.total_stock, 91)
self.assertEqual(part.available_stock(), 66) self.assertEqual(part.available_stock, 66)
# Allocate further stock against the build # Allocate further stock against the build
build.models.BuildItem.objects.create( build.models.BuildItem.objects.create(
@ -1032,7 +1035,7 @@ class PartAPIAggregationTest(InvenTreeAPITestCase):
self.assertEqual(part.build_order_allocation_count(), 20) self.assertEqual(part.build_order_allocation_count(), 20)
self.assertEqual(part.sales_order_allocation_count(), 15) self.assertEqual(part.sales_order_allocation_count(), 15)
self.assertEqual(part.total_stock, 91) self.assertEqual(part.total_stock, 91)
self.assertEqual(part.available_stock(), 56) self.assertEqual(part.available_stock, 56)
class BomItemTest(InvenTreeAPITestCase): class BomItemTest(InvenTreeAPITestCase):