Adding signup and forgot pass tests

Zambom
1 parent f2d885cd
Showing 2 changed files with 36 additions and 3 deletions Show diff stats
users/tests/test_views.py
users/views.py
 from django.test import TestCase, RequestFactory
 from django.core.urlresolvers import reverse_lazy
 from django.contrib.auth.models import AnonymousUser
+from unittest.mock import patch, MagicMock
 from ..models import User
 from .. import views
@@ -50,4 +51,38 @@ class User_Test(TestCase):
 		messages = response.context['messages']
 		self.assertEquals(response.status_code, 200)
-		self.assertIsNotNone(messages) #checking if message was sent
 \ No newline at end of file
+		self.assertIsNotNone(messages) #checking if message was sent
+
+	def test_signup_get(self):
+		request = self.factory.get(reverse_lazy('users:signup'))
+
+		request.user = AnonymousUser()
+
+		response = views.RegisterUser.as_view()(request)
+
+		self.assertEquals(response.status_code, 200)
+
+	@patch('users.models.User.save', MagicMock(name="save"))
+	def test_signup_post(self):
+		data = {
+			'username': 'Teste',
+			'last_name': 'Amadeus',
+			'email': 'teste@amadeus.com.br',
+			'new_password': 'teste',
+			'password2': 'teste'
+		}
+
+		response = self.client.post(reverse_lazy('users:signup'), data)
+
+		self.assertEquals(response.status_code, 302)
+		self.assertTrue(User.save.called)
+		self.assertEquals(User.save.call_count, 2) #call with commit=False first and then saving it
+
+	def test_forgot_pass_get(self):
+		request = self.factory.get(reverse_lazy('users:forgot_pass'))
+
+		request.user = AnonymousUser()
+
+		response = views.ForgotPassword.as_view()(request)
+
+		self.assertEquals(response.status_code, 200)
 \ No newline at end of file
@@ -334,8 +334,6 @@ class RegisterUser(generic.edit.CreateView):
 	def form_valid(self, form):
 		form.save()
-		
-		assign_role(form.instance, 'student')
 		messages.success(self.request, _('User successfully registered!'))
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
	@@ -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