From 16a47f887447ecccd56fc848aee95fa56af92548 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Wed, 24 Feb 2021 19:14:37 +0100 Subject: get_values: get containers from operation - material_types_code from finds --- archaeological_finds/tests.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'archaeological_finds/tests.py') diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py index bede89b51..afcdc713e 100644 --- a/archaeological_finds/tests.py +++ b/archaeological_finds/tests.py @@ -1004,6 +1004,24 @@ class FindTest(FindInit, TestCase): self.assertEqual(models.BaseFind.objects.filter( pk=self.base_finds[0].pk).count(), 0) + def test_get_material_types(self): + mat0 = models.MaterialType.objects.all()[0] + mat1 = models.MaterialType.objects.all()[1] + self.create_finds() + find0 = self.finds[0] + self.assertEqual(find0.get_material_types(), "") + self.assertEqual(find0.get_material_types_code(), "") + + find0.material_types.add(mat0) + find0.material_types.add(mat1) + + self.assertEqual( + find0.get_material_types(), + ", ".join(sorted([mat0.label, mat1.label]))) + self.assertEqual( + find0.get_material_types_code(), + "|".join(sorted([mat0.code, mat1.code]))) + class FindSearchTest(FindInit, TestCase, SearchText): fixtures = WAREHOUSE_FIXTURES -- cgit v1.2.3