From defa1f9af42436d38c6a01b3f3c1a48309c8505b Mon Sep 17 00:00:00 2001
From: Tom Price <tom@codedinternet.com>
Date: Wed, 1 Apr 2015 18:32:08 +0100
Subject: [PATCH] Add user registration test

---
 RIGS/test_functional.py | 103 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 100 insertions(+), 3 deletions(-)

diff --git a/RIGS/test_functional.py b/RIGS/test_functional.py
index aa851bf0..89d1fb7d 100644
--- a/RIGS/test_functional.py
+++ b/RIGS/test_functional.py
@@ -1,6 +1,8 @@
 from django.test import LiveServerTestCase
-
+from django.core import mail
 from selenium import webdriver
+from selenium.webdriver.common.keys import Keys
+import re
 
 class UserRegistrationTest(LiveServerTestCase):
 	def setUp(self):
@@ -9,5 +11,100 @@ class UserRegistrationTest(LiveServerTestCase):
 	def tearDown(self):
 		self.browser.quit()
 
-	def test_false(self):
-		self.assertFail("Finish me")
\ No newline at end of file
+	def test_registration(self):
+		# Navigate to the registration page
+		self.browser.get(self.live_server_url + '/user/register/')
+		title_text = self.browser.find_element_by_tag_name('h3').text
+		self.assertIn("User Registration", title_text)
+		
+		# Check the form invites correctly
+		username = self.browser.find_element_by_id('id_username')
+		self.assertEqual(username.get_attribute('placeholder'), 'Username')
+		email = self.browser.find_element_by_id('id_email')
+		self.assertEqual(email.get_attribute('placeholder'), 'E-mail')
+		self.assertEqual(email.get_attribute('type'), 'email') # If this is correct we don't need to test it later
+		password1 = self.browser.find_element_by_id('id_password1')
+		self.assertEqual(password1.get_attribute('placeholder'), 'Password')
+		self.assertEqual(password1.get_attribute('type'), 'password')
+		password2 = self.browser.find_element_by_id('id_password2')
+		self.assertEqual(password2.get_attribute('placeholder'), 'Password (again)')
+		self.assertEqual(password2.get_attribute('type'), 'password')
+		first_name = self.browser.find_element_by_id('id_first_name')
+		self.assertEqual(first_name.get_attribute('placeholder'), 'First name')
+		last_name = self.browser.find_element_by_id('id_last_name')
+		self.assertEqual(last_name.get_attribute('placeholder'), 'Last name')
+		initials = self.browser.find_element_by_id('id_initials')
+		self.assertEqual(initials.get_attribute('placeholder'), 'Initials')
+		phone = self.browser.find_element_by_id('id_phone')
+		self.assertEqual(phone.get_attribute('placeholder'), 'Phone')
+		
+		# Fill the form out incorrectly
+		username.send_keys('TestUsername')
+		email.send_keys('test@example.com')
+		password1.send_keys('correcthorsebatterystaple')
+		password2.send_keys('correcthorsebatterystapleerror') # deliberate mistake
+		first_name.send_keys('John')
+		last_name.send_keys('Smith')
+		initials.send_keys('JS')
+		phone.send_keys('0123456789')
+
+		# Submit incorrect form
+		submit = self.browser.find_element_by_xpath("//input[@type='submit']")
+		submit.click()
+		# Restablish error fields
+		password1 = self.browser.find_element_by_id('id_password1')
+		password2 = self.browser.find_element_by_id('id_password2')
+
+		# Read what the error is
+		alert = self.browser.find_element_by_css_selector('div.alert-danger').text
+		self.assertIn("password fields didn't match", alert)
+
+		# Passwords should be empty
+		self.assertEqual(password1.get_attribute('value'), '')
+		self.assertEqual(password2.get_attribute('value'), '')
+
+		# Correct error
+		password1.send_keys('correcthorsebatterystaple')
+		password2.send_keys('correcthorsebatterystaple')
+
+		# Submit again
+		password2.send_keys(Keys.ENTER)
+
+		# Check we have a success message
+		alert = self.browser.find_element_by_css_selector('div.alert-success').text
+		self.assertIn('register', alert)
+		self.assertIn('email', alert)
+
+		# Check Email
+		self.assertEqual(len(mail.outbox), 1)
+		email = mail.outbox[0]
+		self.assertIn(email.subject, 'TestUsername activation required')
+		urls = re.findall('http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', email.message)
+		self.assertEqual(len(urls), 2)
+
+		mail.outbox = [] # empty this for later
+
+		# Follow link
+		self.browser.get(urls[1]) # go to the second link
+
+		# Complete registration
+		title_text = self.browser.find_element_by_tag_name('h2')
+		self.assertIn('Complete', title_text)
+
+		# Test login
+		self.browser.get(self.live_server_url + '/user/login')
+		username = self.browser.find_element_by_id('id_username')
+		self.assertEqual(username.get_attribute('placeholder'), 'Username')
+		password = self.browser.find_element_by_id('id_password')
+		self.assertEqual(password.get_attribute('placeholder'), 'Password')
+		self.assertEqual(password.get_attribute('type'), 'password')
+
+		username.send_keys('TestUsername')
+		password.send_keys('correcthorsebatterystaple')
+		password.send_keys(Keys.ENTER)
+
+		# Check we are logged in
+		udd = self.browser.find_element_by_id('userdropdown')
+		self.assertIn('Hi John', udd)
+
+		# All is well