From 8ab64e5c7042c73b597b7086f4741d2af76545ca Mon Sep 17 00:00:00 2001
From: Piotr Gawron <piotr.gawron@uni.lu>
Date: Wed, 13 Dec 2017 17:27:14 +0100
Subject: [PATCH] unit test fixed after new constraints were introduced

---
 smash/web/tests/forms/test_voucher_forms.py | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/smash/web/tests/forms/test_voucher_forms.py b/smash/web/tests/forms/test_voucher_forms.py
index ef712cfd..c95063e2 100644
--- a/smash/web/tests/forms/test_voucher_forms.py
+++ b/smash/web/tests/forms/test_voucher_forms.py
@@ -47,11 +47,7 @@ class VoucherFormTests(LoggedInWithWorkerTestCase):
         voucher.status = VOUCHER_STATUS_USED
         voucher.save()
 
-        voucher_form = VoucherForm(instance=voucher)
-
-        form_data = {}
-        for key, value in voucher_form.initial.items():
-            form_data[key] = format_form_field(value)
+        form_data = self.get_voucher_form_data(voucher)
         form_data["usage_partner"] = ""
 
         voucher_form = VoucherForm(instance=voucher, data=form_data)
@@ -59,17 +55,22 @@ class VoucherFormTests(LoggedInWithWorkerTestCase):
         self.assertFalse(voucher_form.is_valid())
         self.assertTrue("usage_partner" in voucher_form.errors)
 
+    @staticmethod
+    def get_voucher_form_data(voucher):
+        voucher_form = VoucherForm(instance=voucher)
+        form_data = {}
+        for key, value in voucher_form.initial.items():
+            form_data[key] = format_form_field(value)
+        return form_data
+
     def test_valid_status(self):
         study_subject = create_study_subject()
         voucher = create_voucher(study_subject)
+        self.voucher_partner.voucher_types.add(voucher.voucher_type)
         voucher.usage_partner = self.voucher_partner
         voucher.save()
 
-        voucher_form = VoucherForm(instance=voucher)
-
-        form_data = {}
-        for key, value in voucher_form.initial.items():
-            form_data[key] = format_form_field(value)
+        form_data = self.get_voucher_form_data(voucher)
 
         voucher_form = VoucherForm(instance=voucher, data=form_data)
 
-- 
GitLab