No index for stationary containers

1 files changed, 46 insertions, 0 deletions

diff --git a/archaeological_warehouse/tests.py b/archaeological_warehouse/tests.py
index 5ff09e202..93acafa9f 100644
--- a/archaeological_warehouse/tests.py
+++ b/archaeological_warehouse/tests.py
@@ -948,3 +948,49 @@ class ContainerTest(FindInit, TestCase):
         self.assertEqual(container.location, self.alt_warehouse)
         self.assertEqual(container.parent, None)
 
+    def test_index(self):
+        ct = models.ContainerType.objects.all()[0]
+        ct.stationary = False
+        ct.save()
+        ct2 = models.ContainerType.objects.all()[1]
+        ct2.stationary = True
+        ct2.save()
+
+        q = models.Container.objects.filter(
+            location=self.main_warehouse).order_by("-index")
+        if q.count():
+            base_index = q.all()[0].index
+        else:
+            base_index = 0
+        container_1 = models.Container.objects.create(
+            reference="Ref 1",
+            location=self.main_warehouse,
+            container_type=ct)
+        self.assertEqual(models.Container.objects.get(pk=container_1.pk).index,
+                         base_index + 1)
+        container_2 = models.Container.objects.create(
+            reference="Ref 2",
+            location=self.main_warehouse,
+            container_type=ct)
+        self.assertEqual(models.Container.objects.get(pk=container_2.pk).index,
+                         base_index + 2)
+        container_3 = models.Container.objects.create(
+            reference="Ref 3",
+            location=self.main_warehouse,
+            container_type=ct2)
+        self.assertEqual(container_3.index, None)
+        self.assertEqual(models.Container.objects.get(pk=container_3.pk).index,
+                         None)
+        container_4 = models.Container.objects.create(
+            reference="Ref 4",
+            location=self.main_warehouse,
+            container_type=ct2)
+        self.assertEqual(container_4.index, None)
+        self.assertEqual(models.Container.objects.get(pk=container_4.pk).index,
+                         None)
+        container_5 = models.Container.objects.create(
+            reference="Ref 5",
+            location=self.main_warehouse,
+            container_type=ct)
+        self.assertEqual(models.Container.objects.get(pk=container_5.pk).index,
+                         base_index + 3)