diff options
Diffstat (limited to 'archaeological_warehouse/tests.py')
| -rw-r--r-- | archaeological_warehouse/tests.py | 33 |
1 files changed, 32 insertions, 1 deletions
diff --git a/archaeological_warehouse/tests.py b/archaeological_warehouse/tests.py index efe38b4a3..f4ce52732 100644 --- a/archaeological_warehouse/tests.py +++ b/archaeological_warehouse/tests.py @@ -77,7 +77,7 @@ class ContainerWizardCreationTest(WizardTest, FindInit, TestCase): self.container_number + 1) -class ContainerFormCreationTest(FindInit, TestCase): +class ContainerTest(FindInit, TestCase): fixtures = [settings.ROOT_PATH + '../fixtures/initial_data.json', settings.ROOT_PATH + @@ -110,3 +110,34 @@ class ContainerFormCreationTest(FindInit, TestCase): form.save(self.user) self.assertEqual(models.Container.objects.count(), self.container_number + 1) + + def testChangeLocation(self): + main_warehouse = models.Warehouse.objects.create( + name="Main", + warehouse_type=models.WarehouseType.objects.all()[0] + ) + div = models.WarehouseDivision.objects.create(label='division') + div_link = models.WarehouseDivisionLink.objects.create( + warehouse=main_warehouse, division=div) + + container = models.Container.objects.create( + reference="Test", responsible=main_warehouse, + location=main_warehouse, + container_type=models.ContainerType.objects.all()[0] + ) + models.ContainerLocalisation.objects.create( + container=container, division=div_link, + ) + + self.assertTrue(models.ContainerLocalisation.objects.filter( + division__warehouse=main_warehouse).count()) + # changing location remove unrelevent localisation + other_warehouse = models.Warehouse.objects.create( + name="Other", + warehouse_type=models.WarehouseType.objects.all()[0] + ) + container.location = other_warehouse + container.save() + self.assertFalse(models.ContainerLocalisation.objects.filter( + division__warehouse=main_warehouse).count()) + |