Added unit tests for some views

Author: damonmcn

mep_django/linkedin/.tests.py.swp

@@ -1,3 +1,30 @@
 from django.test import TestCase
+#from django.test import Client
+
+class TestHomeView(TestCase):
+    '''Unit test for the home view'''
+    def setUp(self):
+        pass
+
+    def test_status_200(self):
+        '''Return status 200 upon GET request'''
+        response = self.client.get('/')
+        self.assertEqual(response.status_code, 200)
+
+class TestNewsView(TestCase):
+    '''Unit test for the news view'''
+    
+    def setUp(self):
+        self.MY_URL = '/news/'
+
+    def test_status_403(self):
+        '''Return status 403 upon GET request when user is not logged in'''
+        response = self.client.get(self.MY_URL)
+        self.assertEqual(response.status_code, 403)
+
+    def test_status_200(self):
+        '''Return status 200 upon GET request when user IS logged in'''
+        self.client.login(username='user', password='pass')
+        response = self.client.get(self.MY_URL)
+        self.assertEqual(response.status_code, 200)
 
-# Create your tests here.

mep_django/linkedin/tests.py

@@ -3,7 +3,7 @@
 from social.apps.django_app.default.models import UserSocialAuth
 from linkedin import linkedin
 #from django.http import HttpResponse
-
+from django.core.exceptions import PermissionDenied
 
 def logout(request):
     """logs the user out, then redirects to the home page"""
@@ -14,7 +14,6 @@ def home(request):
     """displays the home page"""
     return render(request, 'home.html', {})
 
-
 def about(request):
     """displays the about page"""
     return render(request, 'about.html', {})
@@ -23,7 +22,6 @@ def contact(request):
     """displays the contact page"""
     return render(request, 'contact.html', {})
 
-
 def temp(request):
     """this is used for the popup window which houses the linkedin login page"""
     return render(request, 'temp.html', {})
@@ -59,9 +57,13 @@ def get_access_tokens(user):
 def news(request):
     """Diplays the LinkedIn content, this is the critical view of the app"""
 
-    if not request.user.is_authenticated():
-        return redirect('home')
+    #if not request.user.is_authenticated():
+        #return redirect('home')
 
+    # handle unauthorized access with a 403
+    if not request.user.is_authenticated() or not request.user.is_active:
+        raise PermissionDenied
+
     API_KEY = '75l485e9k29snc'
     API_SECRET = 'iw7fONMpJZcY5HOb'
     USER_KEY, USER_SECRET = get_access_tokens(request.user)
@@ -114,4 +116,4 @@ def news(request):
             #raw_input()
     '''
     return render(request, 'news.html', {'post_list':group_posts['values'], 'update_list': update_list})
-    
+

mep_django/linkedin/views.py

@@ -44,8 +44,11 @@
     'django.contrib.staticfiles',
     'social.apps.django_app.default', # added by chad
     'mep_django.linkedin', # added by chad
+    'django_nose',
 )
 
+TEST_RUNNER = 'django_nose.NoseTestSuiteRunner'
+
 MIDDLEWARE_CLASSES = (
     'django.contrib.sessions.middleware.SessionMiddleware',
     'django.middleware.common.CommonMiddleware',