From 591c75a3eb63714cb8f3a97ab6286a341782ea3f Mon Sep 17 00:00:00 2001
From: cewing <cris@crisewing.com>
Date: Wed, 17 Dec 2014 13:05:20 -0800
Subject: [PATCH] MIT: CCX.  Provide template data for active POCs

Add a utility function to get template data for the POCs to which a user is actively registered

Add tests for the new utility
---
 lms/djangoapps/pocs/tests/test_utils.py | 65 +++++++++++++++++++++++++
 lms/djangoapps/pocs/utils.py            | 31 ++++++++++++
 2 files changed, 96 insertions(+)

diff --git a/lms/djangoapps/pocs/tests/test_utils.py b/lms/djangoapps/pocs/tests/test_utils.py
index 0fe9fbeee0..242c5b98cd 100644
--- a/lms/djangoapps/pocs/tests/test_utils.py
+++ b/lms/djangoapps/pocs/tests/test_utils.py
@@ -11,6 +11,8 @@ from student.roles import CoursePocCoachRole
 from student.tests.factories import (
     AdminFactory,
     UserFactory,
+    CourseEnrollmentFactory,
+    AnonymousUserFactory,
 )
 from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
 from xmodule.modulestore.tests.factories import CourseFactory
@@ -481,3 +483,66 @@ class TestUnenrollEmail(ModuleStoreTestCase):
         self.check_enrollment_state(before, True, self.user, True)
         # no email was sent to the student
         self.assertEqual(len(self.outbox), 0)
+
+
+class TestUserPocList(ModuleStoreTestCase):
+    """Unit tests for poc.utils.get_all_pocs_for_user"""
+
+    def setUp(self):
+        """Create required infrastructure for tests"""
+        self.course = CourseFactory.create()
+        coach = AdminFactory.create()
+        role = CoursePocCoachRole(self.course.id)
+        role.add_users(coach)
+        self.poc = PocFactory(course_id=self.course.id, coach=coach)
+        enrollment = CourseEnrollmentFactory.create(course_id=self.course.id)
+        self.user = enrollment.user
+        self.anonymous = AnonymousUserFactory.create()
+
+    def register_user_in_poc(self, active=False):
+        """create registration of self.user in self.poc
+
+        registration will be inactive unless active=True
+        """
+        PocMembershipFactory(poc=self.poc, student=self.user, active=active)
+
+    def get_course_title(self):
+        from courseware.courses import get_course_about_section
+        return get_course_about_section(self.course, 'title')
+
+    def call_FUT(self, user):
+        from pocs.utils import get_all_pocs_for_user
+        return get_all_pocs_for_user(user)
+
+    def test_anonymous_sees_no_pocs(self):
+        memberships = self.call_FUT(self.anonymous)
+        self.assertEqual(len(memberships), 0)
+
+    def test_unenrolled_sees_no_pocs(self):
+        memberships = self.call_FUT(self.user)
+        self.assertEqual(len(memberships), 0)
+
+    def test_enrolled_inactive_sees_no_pocs(self):
+        self.register_user_in_poc()
+        memberships = self.call_FUT(self.user)
+        self.assertEqual(len(memberships), 0)
+
+    def test_enrolled_sees_a_poc(self):
+        self.register_user_in_poc(active=True)
+        memberships = self.call_FUT(self.user)
+        self.assertEqual(len(memberships), 1)
+
+    def test_data_structure(self):
+        self.register_user_in_poc(active=True)
+        memberships = self.call_FUT(self.user)
+        this_membership = memberships[0]
+        self.assertTrue(this_membership)
+        # structure contains the expected keys
+        for key in ['poc_name', 'poc_url']:
+            self.assertTrue(key in this_membership.keys())
+        url_parts = [self.course.id.to_deprecated_string(), str(self.poc.id)]
+        # all parts of the poc url are present
+        for part in url_parts:
+            self.assertTrue(part in this_membership['poc_url'])
+        actual_name = self.get_course_title()
+        self.assertTrue(actual_name in this_membership['poc_name'])
diff --git a/lms/djangoapps/pocs/utils.py b/lms/djangoapps/pocs/utils.py
index 718c95232b..525deb8de5 100644
--- a/lms/djangoapps/pocs/utils.py
+++ b/lms/djangoapps/pocs/utils.py
@@ -10,7 +10,10 @@ from django.conf import settings
 from django.core.urlresolvers import reverse
 from django.core.mail import send_mail
 
+from courseware.courses import get_course_about_section
+from courseware.courses import get_course_by_id
 from edxmako.shortcuts import render_to_string
+from opaque_keys.edx.locations import SlashSeparatedCourseKey
 
 from microsite_configuration import microsite
 
@@ -214,3 +217,31 @@ def send_mail_to_student(student, param_dict):
             [student],
             fail_silently=False
         )
+
+
+def get_all_pocs_for_user(user):
+    """return all POCS to which the user is registered
+
+    Returns a list of dicts: {
+        poc_name: <formatted title of POC course>
+        poc_url: <url to view this POC>
+    }
+    """
+    if user.is_anonymous():
+        return []
+    active_poc_memberships = PocMembership.objects.filter(
+        student=user, active__exact=True
+    )
+    memberships = []
+    for membership in active_poc_memberships:
+        course = get_course_by_id(membership.poc.course_id)
+        title = 'POC: {}'.format(get_course_about_section(course, 'title'))
+        url = reverse(
+            'switch_active_poc',
+            args=[course.id.to_deprecated_string(), membership.poc.id]
+        )
+        memberships.append({
+            'poc_name': title,
+            'poc_url': url
+        })
+    return memberships