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	1	from django.test import TestCase, RequestFactory
2	2	from django.core.urlresolvers import reverse_lazy
3	3	from django.contrib.auth.models import AnonymousUser
	4	+from unittest.mock import patch, MagicMock
4	5	from ..models import User
5	6	from .. import views
6	7
...	...	@@ -50,4 +51,38 @@ class User_Test(TestCase):
50	51	messages = response.context['messages']
51	52
52	53	self.assertEquals(response.status_code, 200)
53		- self.assertIsNotNone(messages) #checking if message was sent
54	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	90	\ No newline at end of file
...	...
...	...	@@ -334,8 +334,6 @@ class RegisterUser(generic.edit.CreateView):
334	334
335	335	def form_valid(self, form):
336	336	form.save()
337		-
338		- assign_role(form.instance, 'student')
339	337
340	338	messages.success(self.request, _('User successfully registered!'))
341	339
...	...