diff --git a/smash/web/tests/test_SubjectAddForm.py b/smash/web/tests/test_SubjectAddForm.py index 6e8e81692ce8214511157bb50cc39f9623580990..41fdb32027d0dfe0f14c13344c43b5bbeb5fbe9c 100644 --- a/smash/web/tests/test_SubjectAddForm.py +++ b/smash/web/tests/test_SubjectAddForm.py @@ -3,25 +3,22 @@ from web.forms import SubjectAddForm from web.models import Subject class SubjectAddFormTests(TestCase): - def test_validation(self): - form_data = {'first_name': 'name', - 'last_name': 'name', - 'status' : Subject.STATUS_CHOICES_NOT_CONTACTED, - 'sex' : Subject.SEX_CHOICES_MALE, - 'country' : 'Luxembourg' + def setUp(self): + self.sample_data = {'first_name': 'name', + 'last_name': 'name', + 'status' : Subject.STATUS_CHOICES_NOT_CONTACTED, + 'sex' : Subject.SEX_CHOICES_MALE, + 'country' : 'Luxembourg' } - form = SubjectAddForm(data=form_data) + def test_validation(self): + form = SubjectAddForm(data=self.sample_data) form.is_valid() self.assertTrue(form.is_valid()) def test_invalid(self): - form_data = {'first_name': 'name', - 'last_name': 'name', - 'status' : Subject.STATUS_CHOICES_NOT_CONTACTED, - 'sex' : Subject.SEX_CHOICES_MALE, - 'country' : 'Luxembourg', - 'screening_number' : '123' - } + form_data = self.sample_data + form_data['screening_number'] = "123" + form = SubjectAddForm(data=form_data) form.is_valid() self.assertTrue(form.is_valid()) @@ -33,13 +30,9 @@ class SubjectAddFormTests(TestCase): self.assertTrue("screening_number" in form2.errors) def test_invalid_2(self): - form_data = {'first_name': 'name', - 'last_name': 'name', - 'status' : Subject.STATUS_CHOICES_NOT_CONTACTED, - 'sex' : Subject.SEX_CHOICES_MALE, - 'country' : 'Luxembourg', - 'nd_number' : '123' - } + form_data = self.sample_data + form_data['nd_number'] = "123" + form = SubjectAddForm(data=form_data) form.is_valid() self.assertTrue(form.is_valid())