REVEM-219 Handle invalid course run key

lms/djangoapps/experiments/tests/test_utils.py

@@ -0,0 +1,25 @@
+"""
+Tests of experiment functionality
+"""
+from unittest import TestCase
+from lms.djangoapps.experiments.utils import is_enrolled_in_course_run
+from opaque_keys.edx.keys import CourseKey
+
+
+class ExperimentUtilsTests(TestCase):
+    """
+    Tests of experiment functionality
+    """
+    def test_valid_course_run_key_enrollment(self):
+        course_run = {
+            'key': 'course-v1:DelftX+NGIx+RA0',
+        }
+        enrollment_ids = {CourseKey.from_string('course-v1:DelftX+NGIx+RA0')}
+        self.assertTrue(is_enrolled_in_course_run(course_run, enrollment_ids))
+
+    def test_invalid_course_run_key_enrollment(self):
+        course_run = {
+            'key': 'cr_key',
+        }
+        enrollment_ids = {CourseKey.from_string('course-v1:DelftX+NGIx+RA0')}
+        self.assertFalse(is_enrolled_in_course_run(course_run, enrollment_ids))