diff options
Diffstat (limited to 'archaeological_warehouse/tests.py')
| -rw-r--r-- | archaeological_warehouse/tests.py | 46 |
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) |