Added testcases for model and view.

3 files changed, 160 insertions, 5 deletions

diff --git a/yaksh/test_models.py b/yaksh/test_models.py
index 8bd2dda..94178a0 100644
--- a/yaksh/test_models.py
+++ b/yaksh/test_models.py
@@ -252,6 +252,18 @@ class QuizTestCases(unittest.TestCase):
                          )
         self.assertEqual(trial_quiz.time_between_attempts, 0)
 
+    def test_view_answerpaper(self):
+        self.assertFalse(self.quiz1.view_answerpaper)
+        self.assertFalse(self.quiz2.view_answerpaper)
+
+        # When
+        self.quiz1.view_answerpaper = True
+        self.quiz1.save()
+
+        # Then
+        self.assertTrue(self.quiz1.view_answerpaper)
+
+
 
 ###############################################################################
 class QuestionPaperTestCases(unittest.TestCase):
diff --git a/yaksh/test_views.py b/yaksh/test_views.py
index 6e59e26..4dd905e 100644
--- a/yaksh/test_views.py
+++ b/yaksh/test_views.py
@@ -5,6 +5,7 @@ from django.contrib.auth.models import Group
 from django.core.urlresolvers import reverse
 from django.test import TestCase
 from django.test import Client
+from django.utils import timezone
 
 from yaksh.models import User, Profile, Question, Quiz, QuestionPaper,\
     QuestionSet, AnswerPaper, Answer, Course, StandardTestCase,\
@@ -632,7 +633,7 @@ class TestRemoveTeacher(TestCase):
             target_status_code=301
         )
         for t_id in teacher_id_list:
-            teacher = User.objects.get(id=t_id) 
+            teacher = User.objects.get(id=t_id)
             self.assertNotIn(teacher, self.course.teachers.all())
 
 
@@ -640,7 +641,7 @@ class TestCourses(TestCase):
     def setUp(self):
         self.client = Client()
 
-        self.mod_group = Group.objects.create(name='moderator') 
+        self.mod_group = Group.objects.create(name='moderator')
 
         # Create Moderator with profile
         self.user1_plaintext_pass = 'demo1'
@@ -754,7 +755,7 @@ class TestCourseDetail(TestCase):
     def setUp(self):
         self.client = Client()
 
-        self.mod_group = Group.objects.create(name='moderator') 
+        self.mod_group = Group.objects.create(name='moderator')
 
         # Create Moderator with profile
         self.user1_plaintext_pass = 'demo1'
@@ -848,7 +849,7 @@ class TestCourseDetail(TestCase):
         """
         self.client.login(
             username=self.user2.username,
-            password=self.user2_plaintext_pass 
+            password=self.user2_plaintext_pass
         )
         response = self.client.get(reverse('yaksh:course_detail',
                 kwargs={'course_id': self.user1_course.id}
@@ -980,6 +981,148 @@ class TestEnrollRequest(TestCase):
         )
         self.assertRedirects(response, '/exam/manage/')
 
+class TestViewAnswerPaper(TestCase):
+    def setUp(self):
+        self.client = Client()
+        self.plaintext_pass = 'demo'
+
+        for i in range(1, 4):
+            User.objects.create_user(
+                username='demo_user{0}'.format(i),
+                password=self.plaintext_pass,
+                first_name='first_name',
+                last_name='last_name',
+                email='demo@test.com'
+            )
+
+        self.user1 = User.objects.get(pk=1)
+
+        self.course = Course.objects.create(name="Python Course",
+                                       enrollment="Enroll Request",
+                                       creator=self.user1)
+
+        self.question = Question.objects.create(summary='Dummy', points=1,
+                                               type='code', user=self.user1)
+
+        self.quiz = Quiz.objects.create(time_between_attempts=0, course=self.course,
+                                        description='demo quiz', language='Python')
+
+        self.question_paper = QuestionPaper.objects.create(quiz=self.quiz,
+            total_marks=1.0)
+
+        self.question_paper.fixed_questions.add(self.question)
+        self.question_paper.save()
+
+        AnswerPaper.objects.create(user_id=3,
+                attempt_number=1, question_paper=self.question_paper,
+                start_time=timezone.now(), user_ip='101.0.0.1',
+                end_time=timezone.now()+timezone.timedelta(minutes=20))
+
+    def tearDown(self):
+        User.objects.all().delete()
+        Course.objects.all().delete()
+        Question.objects.all().delete()
+        Quiz.objects.all().delete()
+        QuestionPaper.objects.all().delete()
+        AnswerPaper.objects.all().delete()
+
+    def test_anonymous_user(self):
+        # Given, user not logged in
+        redirect_destination = ('/exam/login/?next=/exam'
+            '/view_answerpaper/{0}/'.format(self.question_paper.id))
+
+        # When
+        response = self.client.get(reverse('yaksh:view_answerpaper',
+                kwargs={'questionpaper_id': self.question_paper.id}
+            ),
+            follow=True
+        )
+
+        # Then
+        self.assertRedirects(response, redirect_destination)
+
+    def test_cannot_view(self):
+        # Given, enrolled user tries to view when not permitted by moderator
+        user2 = User.objects.get(pk=2)
+        self.course.students.add(user2)
+        self.course.save()
+        self.quiz.view_answerpaper = False
+        self.quiz.save()
+        self.client.login(
+            username=user2.username,
+            password=self.plaintext_pass
+        )
+
+        # When
+        response = self.client.get(reverse('yaksh:view_answerpaper',
+                kwargs={'questionpaper_id': self.question_paper.id}
+            ),
+            follow=True
+        )
+
+        # Then
+        self.assertRedirects(response, '/exam/quizzes/')
+
+    def test_can_view(self):
+        # Given, user enrolled and can view
+        user3 = User.objects.get(pk=3)
+        self.course.students.add(user3)
+        self.course.save()
+        answerpaper = AnswerPaper.objects.get(pk=1)
+        self.quiz.view_answerpaper = True
+        self.quiz.save()
+        self.client.login(
+            username=user3.username,
+            password=self.plaintext_pass
+        )
+
+        # When
+        response = self.client.get(reverse('yaksh:view_answerpaper',
+                kwargs={'questionpaper_id': self.question_paper.id}
+            ),
+            follow=True
+        )
+
+        # Then
+        self.assertEqual(response.status_code, 200)
+        self.assertTrue('data' in response.context)
+        self.assertTrue('quiz' in response.context)
+        self.assertTemplateUsed(response, 'yaksh/view_answerpaper.html')
+
+
+        # When, wrong question paper id
+        response = self.client.get(reverse('yaksh:view_answerpaper',
+                kwargs={'questionpaper_id': 190}
+            ),
+            follow=True
+        )
+
+        # Then
+        self.assertEqual(response.status_code, 404)
+
+
+    def test_view_when_not_enrolled(self):
+        # Given, user tries to view when not enrolled in the course
+        user2 = User.objects.get(pk=2)
+        self.client.login(
+            username=user2.username,
+            password=self.plaintext_pass
+        )
+        self.course.students.remove(user2)
+        self.course.save()
+        self.quiz.view_answerpaper = True
+        self.quiz.save()
+
+        # When
+        response = self.client.get(reverse('yaksh:view_answerpaper',
+                kwargs={'questionpaper_id': self.question_paper.id}
+            ),
+            follow=True
+        )
+
+        # Then
+        self.assertRedirects(response, '/exam/quizzes/')
+
 
 class TestSelfEnroll(TestCase):
     def setUp(self):
diff --git a/yaksh/views.py b/yaksh/views.py
index 9e0edac..9d1ab99 100644
--- a/yaksh/views.py
+++ b/yaksh/views.py
@@ -1279,7 +1279,7 @@ def test_quiz(request, mode, quiz_id):
 def view_answerpaper(request, questionpaper_id):
     user = request.user
     quiz = get_object_or_404(QuestionPaper, pk=questionpaper_id).quiz
-    if quiz.view_answerpaper:
+    if quiz.view_answerpaper and user in quiz.course.students.all():
         data = AnswerPaper.objects.get_user_data(user, questionpaper_id)
         context = {'data': data, 'quiz': quiz}
         return my_render_to_response('yaksh/view_answerpaper.html', context)