diff --git a/opentech/apply/funds/tests/test_models.py b/opentech/apply/funds/tests/test_models.py index a463b9264de19c96279b6969029ec9558ab92bfc..107d0f35269c79036280b1f4377972a72b63ab0d 100644 --- a/opentech/apply/funds/tests/test_models.py +++ b/opentech/apply/funds/tests/test_models.py @@ -251,14 +251,15 @@ class TestFormSubmission(TestCase): self.assertEqual(ApplicationSubmission.objects.first().user, user) def test_associated_if_another_user_exists(self): + email = 'another@email.com' self.submit_form() # Someone else submits a form - self.submit_form(email='another@email.com') + self.submit_form(email=email) # Lead + 2 x applicant self.assertEqual(self.User.objects.count(), 3) - _, first_user, second_user = self.User.objects.all() + first_user, second_user = self.User.objects.get(email=self.email), self.User.objects.get(email=email) self.assertEqual(ApplicationSubmission.objects.count(), 2) self.assertEqual(ApplicationSubmission.objects.first().user, first_user) self.assertEqual(ApplicationSubmission.objects.last().user, second_user) diff --git a/opentech/apply/users/models.py b/opentech/apply/users/models.py index 7b729c39bbfd7ee5348f0d235715b0ffc36cd343..4441064d4acf0462a76082d90b446df34e682283 100644 --- a/opentech/apply/users/models.py +++ b/opentech/apply/users/models.py @@ -85,3 +85,6 @@ class User(AbstractUser): class Meta: ordering = ('full_name', 'email') + + def __repr__(self): + return f'<{self.__class__.__name__}: {self.full_name} ({self.email})>'