diff options
| -rw-r--r-- | archaeological_warehouse/tests.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/archaeological_warehouse/tests.py b/archaeological_warehouse/tests.py index d84fd5fb9..b329a8432 100644 --- a/archaeological_warehouse/tests.py +++ b/archaeological_warehouse/tests.py @@ -169,15 +169,15 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase): warehouse_json = json.loads( res[("warehouse", "archaeological_warehouse__Warehouse")] ) - self.assertEqual(len(warehouse_json), 3) + self.assertEqual(len(warehouse_json), models.Warehouse.objects.count()) container_json = json.loads( res[("warehouse", "archaeological_warehouse__Container")] ) - self.assertEqual(len(container_json), 5 + 4) + self.assertEqual(len(container_json), models.Container.objects.count()) div_json = json.loads( res[("warehouse", "archaeological_warehouse__WarehouseDivisionLink")] ) - self.assertEqual(len(div_json), 4) + self.assertEqual(len(div_json), models.WarehouseDivisionLink.objects.count()) result_queryset = Operation.objects.filter(uuid=self.operations[0].uuid) res = self.generic_serialization_test( |