diff options
Diffstat (limited to 'archaeological_warehouse/tests.py')
| -rw-r--r-- | archaeological_warehouse/tests.py | 82 |
1 files changed, 69 insertions, 13 deletions
diff --git a/archaeological_warehouse/tests.py b/archaeological_warehouse/tests.py index 2431dc9a5..a14c0fb8c 100644 --- a/archaeological_warehouse/tests.py +++ b/archaeological_warehouse/tests.py @@ -43,8 +43,9 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase): cr2 = self.create_context_record( data={"label": u"CR 2", "operation": ope2} )[1] - self.create_finds(data_base={"context_record": cr})[0] - self.create_finds(data_base={"context_record": cr2})[1] + self.create_finds(data_base={"context_record": cr}) + self.create_finds(data_base={"context_record": cr2}) + self.create_finds(data_base={"context_record": cr2}) w1 = models.Warehouse.objects.create( name="Test1", external_id="test", @@ -52,7 +53,12 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase): ) w2 = models.Warehouse.objects.create( name="Test2", - external_id="test", + external_id="test2", + warehouse_type=models.WarehouseType.objects.all()[0], + ) + w3 = models.Warehouse.objects.create( + name="Test3", + external_id="test3", warehouse_type=models.WarehouseType.objects.all()[0], ) c1 = models.Container.objects.create( @@ -95,6 +101,18 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase): f1.container = c3 f1.container_ref = c4 f1.save() + c5 = models.Container.objects.create( + location=w3, + responsible=w3, + container_type=models.ContainerType.objects.all()[0], + reference="Réf5", + index=1, + external_id="ref5-2" + ) + f2 = self.finds[2] + f2.container = c5 + f2.container_ref = c5 + f2.save() wd1 = models.WarehouseDivision.objects.create( label="Étagère", txt_idx="etagere" ) @@ -109,6 +127,14 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase): warehouse=w1, division=wd2 ) + wl3 = models.WarehouseDivisionLink.objects.create( + warehouse=w2, + division=wd2 + ) + wl4 = models.WarehouseDivisionLink.objects.create( + warehouse=w3, + division=wd1 + ) models.ContainerLocalisation.objects.create( container=c1, division=wl1, @@ -119,6 +145,16 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase): division=wl2, reference="A2" ) + models.ContainerLocalisation.objects.create( + container=c2, + division=wl3, + reference="A4" + ) + models.ContainerLocalisation.objects.create( + container=c3, + division=wl4, + reference="A5" + ) def test_serialization(self): res = self.generic_serialization_test( @@ -126,26 +162,46 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase): warehouse_json = json.loads( res[('warehouse', "archaeological_warehouse__Warehouse")] ) - self.assertEqual(len(warehouse_json), 2) + self.assertEqual(len(warehouse_json), 3) container_json = json.loads( res[('warehouse', "archaeological_warehouse__Container")] ) - self.assertEqual(len(container_json), 4) + self.assertEqual(len(container_json), 5) + div_json = json.loads( + res[('warehouse', + "archaeological_warehouse__WarehouseDivisionLink")] + ) + self.assertEqual(len(div_json), 4) + loca_json = json.loads( + res[('warehouse', + "archaeological_warehouse__ContainerLocalisation")] + ) + self.assertEqual(len(loca_json), 4) - """ result_queryset = Operation.objects.filter(uuid=self.operations[0].uuid) res = self.generic_serialization_test( - serializers.find_serialization, no_test=True, + serializers.warehouse_serialization, no_test=True, kwargs={"operation_queryset": result_queryset} ) - find_json = json.loads( - res[('finds', 'archaeological_finds__Find')] + warehouse_json = json.loads( + res[('warehouse', "archaeological_warehouse__Warehouse")] ) - self.assertEqual(len(find_json), 1) - bfind_json = json.loads( - res[('finds', 'archaeological_finds__BaseFind')] + self.assertEqual(len(warehouse_json), 2) + container_json = json.loads( + res[('warehouse', "archaeological_warehouse__Container")] ) - self.assertEqual(len(bfind_json), 1) + self.assertEqual(len(container_json), 2) + div_json = json.loads( + res[('warehouse', + "archaeological_warehouse__WarehouseDivisionLink")] + ) + self.assertEqual(len(div_json), 3) + loca_json = json.loads( + res[('warehouse', + "archaeological_warehouse__ContainerLocalisation")] + ) + self.assertEqual(len(loca_json), 3) + """ result_queryset = ContextRecord.objects.filter( uuid=self.context_records[0].uuid) |