diff options
Diffstat (limited to 'archaeological_finds/tests.py')
| -rw-r--r-- | archaeological_finds/tests.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py index e39ebda32..2c78d3efa 100644 --- a/archaeological_finds/tests.py +++ b/archaeological_finds/tests.py @@ -1207,12 +1207,22 @@ class FindQATest(FindInit, TestCase): codename='change_find')) def test_duplicate(self): + t1, __ = models.Treatment.objects.get_or_create( + label="Treatment 1", + treatment_state=models.TreatmentState.objects.all()[0] + ) + t2, __ = models.Treatment.objects.get_or_create( + label="Treatment 1", + treatment_state=models.TreatmentState.objects.all()[0] + ) find = self.finds[0] default_desc = "Description for duplicate" find.description = default_desc + find.upstream_treatment = t1 find.save() d = Document.objects.create() d.finds.add(find) + find.treatments.add(t2) c = Client() url = reverse('find-qa-duplicate', args=[find.pk]) @@ -1245,6 +1255,10 @@ class FindQATest(FindInit, TestCase): base_bf = find.get_first_base_find() self.assertEqual(new_bf.context_record, base_bf.context_record) self.assertNotIn(d, list(new.documents.all())) + # associated treatments are removed #4850 + self.assertIsNone(new.upstream_treatment) + self.assertIsNone(new.downstream_treatment) + self.assertNotIn(t2, list(new.treatments.all())) def test_bulk_update(self): c = Client() |