From 994091c2226229fa7fdc1ea9e6cc1bc4e6a27b71 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Fri, 20 Jan 2017 12:25:10 +0100
Subject: Tests: treatment creation wizard

---
 archaeological_finds/tests.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'archaeological_finds/tests.py')

diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py
index b8194cd4d..18744ea3d 100644
--- a/archaeological_finds/tests.py
+++ b/archaeological_finds/tests.py
@@ -207,12 +207,11 @@ class ATreatmentWizardCreationTest(WizardTest, FindInit, TestCase):
         self.assertEqual(models.Treatment.objects.count(),
                          self.treatment_number + 1)
         treat = models.Treatment.objects.order_by('-pk').all()[0]
-        """
+        self.find = models.Find.objects.get(pk=self.find.pk)
         self.assertEqual(models.Find.objects.filter(
             upstream_treatment=treat).count(), 1)
         self.assertEqual(self.find.downstream_treatment,
                          treat)
-        """
 
 
 class ImportFindTest(ImportContextRecordTest):
-- 
cgit v1.2.3