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