🐛 recalculate container weight on find remove (refs #5470)

1 files changed, 11 insertions, 0 deletions

diff --git a/archaeological_warehouse/tests.py b/archaeological_warehouse/tests.py
index 54432f9a5..ebea75217 100644
--- a/archaeological_warehouse/tests.py
+++ b/archaeological_warehouse/tests.py
@@ -1177,11 +1177,22 @@ class ContainerTest(FindInit, TestCase):
         self.assertEqual(container_1.calculated_weight, 1500)
         self.assertEqual(container_1.cached_weight, 1500)
 
+        container_2 = models.Container.objects.create(
+            reference="Test 2", location=self.main_warehouse, container_type=ct
+        )
+        find2.container = container_2
+        find2.save()
+        container_1 = models.Container.objects.get(pk=container_1.pk)
+        self.assertEqual(container_1.calculated_weight, 800)
+        self.assertEqual(container_1.cached_weight, 800)
+
         profile, created = IshtarSiteProfile.objects.get_or_create(
             slug="default", active=True
         )
         profile.calculate_weight_on_full = True
         profile.save()
+        find2.container = container_1
+        find2.save()
 
         container_1.save()
         container_1 = models.Container.objects.get(pk=container_1.pk)