From 8d3f54ae3ff824503e409886e020c020ac3ce73c Mon Sep 17 00:00:00 2001
From: Tsa6 <tsa6@tsa6.net>
Date: Fri, 29 Sep 2017 17:21:42 -0400
Subject: [PATCH] Added test for successfully logging in new user

---
 tests/LandingPage/__init__.py   |  0
 tests/LandingPage/test_views.py | 46 +++++++++++++++++++++++++++++++++
 tests/__init__.py               |  0
 3 files changed, 46 insertions(+)
 create mode 100644 tests/LandingPage/__init__.py
 create mode 100644 tests/LandingPage/test_views.py
 create mode 100644 tests/__init__.py

diff --git a/tests/LandingPage/__init__.py b/tests/LandingPage/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/LandingPage/test_views.py b/tests/LandingPage/test_views.py
new file mode 100644
index 0000000..fd6709d
--- /dev/null
+++ b/tests/LandingPage/test_views.py
@@ -0,0 +1,46 @@
+from django.test import TestCase,Client,override_settings
+import responses
+from LandingPage.models import User
+from urllib import parse
+
+@override_settings(
+        AUTH_TOKEN_ENDPOINT='http://icynet.test/api/',
+        AUTH_CLIENT_ID='clid',
+        AUTH_B64='Y2xpZDpjbGlzZWM=',
+        AUTH_REDIRECT_URL='http://redirect.test'
+)
+class TestLogin(TestCase):
+   
+    def test_login_new_user(self):
+        # Set up responses to control network flow
+        with responses.RequestsMock() as rm:
+            rm.add(responses.POST,'http://icynet.test/api/token',json={'access_token':'1accesstoken1'})
+            rm.add(responses.GET,'http://icynet.test/api/user',json={'uuid':'935a41b5-b38d-42c3-96ef-653402fc44ca','email':'johnsmith@gmail.com','display_name':'Mr. Smith'})
+
+            # Make initial request to redirect endpoint
+            client = Client()
+            resp = client.get('/login')
+            self.assertEqual(resp.status_code, 302)
+            query = parse.parse_qs(parse.urlparse(resp['Location']).query)
+            state = query['state'][0]
+            self.assertEqual(query['client_id'][0],'clid')
+            self.assertEqual(query['response_type'][0],'code')
+            self.assertEqual(query['redirect_uri'][0],'http://redirect.test')
+            self.assertEqual(query['scope'][0],'email')
+
+            # Make connection to the real endpoint
+            resp = client.get('/login/redirect?state=%s&code=%s'%(state, 'code'))
+            self.assertEqual(resp.status_code, 302)
+
+            # Check that the database is all good
+            users = User.objects.all()
+            self.assertEqual(len(users), 1)
+            user = users[0]
+            self.assertEqual(user.user_id,'935a41b5-b38d-42c3-96ef-653402fc44ca')
+            self.assertEqual(user.email,'johnsmith@gmail.com')
+            self.assertEqual(user.display_name, 'Mr. Smith')
+
+            # Check appropriate values are in the session
+            self.assertEqual(client.session['user_id'], '935a41b5-b38d-42c3-96ef-653402fc44ca')
+            self.assertEqual(client.session['token'],'1accesstoken1')
+            self.assertEqual(client.session['disp_name'], 'Mr. Smith')
diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29