From 800b071e258416c3e16540a91027f0b54fe50a81 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Tue, 10 Jan 2017 16:34:35 +0100 Subject: Container: remove old localisation when changing warehouse --- archaeological_warehouse/tests.py | 33 ++++++++++++++++++++++++++++++++- 1 file changed, 32 insertions(+), 1 deletion(-) (limited to 'archaeological_warehouse/tests.py') 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()) + -- cgit v1.2.3