From a9679cff69f4442f34df536d9b3664bf58e010af Mon Sep 17 00:00:00 2001 From: Trent Palmer Date: Tue, 20 Apr 2021 19:30:08 -0700 Subject: [PATCH] update accounts/tests/test_register_view.py add functions test_register_user_already_exists and test_register_email_already_exists --- accounts/tests/test_register_view.py | 36 ++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) diff --git a/accounts/tests/test_register_view.py b/accounts/tests/test_register_view.py index 943242b..c145fb1 100644 --- a/accounts/tests/test_register_view.py +++ b/accounts/tests/test_register_view.py @@ -10,8 +10,44 @@ class RegisterViewTestCase(TestCase): user_b = User(email='user_b@example.com', username='user_b') user_b.set_password('123456password') user_b.save() + user_c = User(username='user_c') + user_c.set_password('123456password') + user_c.save() Account.objects.create(user=user_b, twitter_handle='@user_b') + Account.objects.create(user=user_c) EmailWhiteList.objects.create(email='user_a@example.com') + EmailWhiteList.objects.create(email='user_b@example.com') + EmailWhiteList.objects.create(email='user_c@example.com') + + def test_register_user_already_exists(self): + response = self.client.post(reverse('accounts:register'), { + 'username': 'user_c', + 'email': 'user_c@example.com', + 'password1': 'password123456donkey', + 'password2': 'password123456donkey' + }, follow=True) + self.assertEquals(response.status_code, 200) + self.assertTemplateUsed(response, 'base_form.html') + qs = User.objects.filter(username="user_a") + self.assertEquals(len(qs), 0) + self.assertEquals(response.request['PATH_INFO'], '/accounts/register/') + msg = "Try a different username, that one already exists." + self.assertEquals(response.content.decode('utf-8').count(msg), 1) + + def test_register_email_already_exists(self): + response = self.client.post(reverse('accounts:register'), { + 'username': 'user_a', + 'email': 'user_b@example.com', + 'password1': 'password123456donkey', + 'password2': 'password123456donkey' + }, follow=True) + self.assertEquals(response.status_code, 200) + self.assertTemplateUsed(response, 'base_form.html') + qs = User.objects.filter(username="user_a") + self.assertEquals(len(qs), 0) + self.assertEquals(response.request['PATH_INFO'], '/accounts/register/') + msg = "An account already exists with this email address." + self.assertEquals(response.content.decode('utf-8').count(msg), 1) def test_register_already_logged_in(self): self.client.login(username='user_b', password='123456password')