From b644cadb709142941198ce9ae693496b3b722f08 Mon Sep 17 00:00:00 2001
From: Zambom <erikzambom@hotmail.com>
Date: Wed, 28 Dec 2016 17:05:01 -0200
Subject: [PATCH] Adding signup and forgot pass tests

---
 users/tests/test_views.py | 37 ++++++++++++++++++++++++++++++++++++-
 users/views.py            |  2 --
 2 files changed, 36 insertions(+), 3 deletions(-)

diff --git a/users/tests/test_views.py b/users/tests/test_views.py
index cfbff12..61347c1 100644
--- a/users/tests/test_views.py
+++ b/users/tests/test_views.py
@@ -1,6 +1,7 @@
 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
diff --git a/users/views.py b/users/views.py
index c70f0df..5373f35 100644
--- a/users/views.py
+++ b/users/views.py
@@ -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!'))
 
--
libgit2 0.21.2