get_values: get containers from operation - material_types_code from finds

1 files changed, 18 insertions, 0 deletions

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