Commit b644cadb709142941198ce9ae693496b3b722f08
1 parent
f2d885cd
Exists in
master
and in
3 other branches
Adding signup and forgot pass tests
Showing
2 changed files
with
36 additions
and
3 deletions
Show diff stats
users/tests/test_views.py
1 | from django.test import TestCase, RequestFactory | 1 | from django.test import TestCase, RequestFactory |
2 | from django.core.urlresolvers import reverse_lazy | 2 | from django.core.urlresolvers import reverse_lazy |
3 | from django.contrib.auth.models import AnonymousUser | 3 | from django.contrib.auth.models import AnonymousUser |
4 | +from unittest.mock import patch, MagicMock | ||
4 | from ..models import User | 5 | from ..models import User |
5 | from .. import views | 6 | from .. import views |
6 | 7 | ||
@@ -50,4 +51,38 @@ class User_Test(TestCase): | @@ -50,4 +51,38 @@ class User_Test(TestCase): | ||
50 | messages = response.context['messages'] | 51 | messages = response.context['messages'] |
51 | 52 | ||
52 | self.assertEquals(response.status_code, 200) | 53 | self.assertEquals(response.status_code, 200) |
53 | - self.assertIsNotNone(messages) #checking if message was sent | ||
54 | \ No newline at end of file | 54 | \ No newline at end of file |
55 | + self.assertIsNotNone(messages) #checking if message was sent | ||
56 | + | ||
57 | + def test_signup_get(self): | ||
58 | + request = self.factory.get(reverse_lazy('users:signup')) | ||
59 | + | ||
60 | + request.user = AnonymousUser() | ||
61 | + | ||
62 | + response = views.RegisterUser.as_view()(request) | ||
63 | + | ||
64 | + self.assertEquals(response.status_code, 200) | ||
65 | + | ||
66 | + @patch('users.models.User.save', MagicMock(name="save")) | ||
67 | + def test_signup_post(self): | ||
68 | + data = { | ||
69 | + 'username': 'Teste', | ||
70 | + 'last_name': 'Amadeus', | ||
71 | + 'email': 'teste@amadeus.com.br', | ||
72 | + 'new_password': 'teste', | ||
73 | + 'password2': 'teste' | ||
74 | + } | ||
75 | + | ||
76 | + response = self.client.post(reverse_lazy('users:signup'), data) | ||
77 | + | ||
78 | + self.assertEquals(response.status_code, 302) | ||
79 | + self.assertTrue(User.save.called) | ||
80 | + self.assertEquals(User.save.call_count, 2) #call with commit=False first and then saving it | ||
81 | + | ||
82 | + def test_forgot_pass_get(self): | ||
83 | + request = self.factory.get(reverse_lazy('users:forgot_pass')) | ||
84 | + | ||
85 | + request.user = AnonymousUser() | ||
86 | + | ||
87 | + response = views.ForgotPassword.as_view()(request) | ||
88 | + | ||
89 | + self.assertEquals(response.status_code, 200) | ||
55 | \ No newline at end of file | 90 | \ No newline at end of file |
users/views.py
@@ -334,8 +334,6 @@ class RegisterUser(generic.edit.CreateView): | @@ -334,8 +334,6 @@ class RegisterUser(generic.edit.CreateView): | ||
334 | 334 | ||
335 | def form_valid(self, form): | 335 | def form_valid(self, form): |
336 | form.save() | 336 | form.save() |
337 | - | ||
338 | - assign_role(form.instance, 'student') | ||
339 | 337 | ||
340 | messages.success(self.request, _('User successfully registered!')) | 338 | messages.success(self.request, _('User successfully registered!')) |
341 | 339 |