check course end date and verified mode for first purchase offer

openedx/features/discounts/tests/test_applicability.py

@@ -1,11 +1,16 @@
 """Tests of openedx.features.discounts.applicability"""
 # -*- coding: utf-8 -*-
 
+from datetime import timedelta
+from django.utils.timezone import now
+
+from course_modes.tests.factories import CourseModeFactory
+from openedx.core.djangoapps.waffle_utils.testutils import override_waffle_flag
 from student.tests.factories import UserFactory
 from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
 from xmodule.modulestore.tests.factories import CourseFactory
 
-from ..applicability import can_receive_discount
+from ..applicability import can_receive_discount, DISCOUNT_APPLICABILITY_FLAG
 
 
 class TestApplicability(ModuleStoreTestCase):
@@ -18,8 +23,25 @@ class TestApplicability(ModuleStoreTestCase):
         super(TestApplicability, self).setUp()
         self.user = UserFactory.create()
         self.course = CourseFactory.create(run='test', display_name='test')
+        CourseModeFactory.create(course_id=self.course.id, mode_slug='verified')
 
     def test_can_receive_discount(self):
         # Right now, no one should be able to receive the discount
-        applicability = can_receive_discount(user=self.user, course_key_string=self.course.id)
+        applicability = can_receive_discount(user=self.user, course=self.course)
+        self.assertEqual(applicability, False)
+
+    @override_waffle_flag(DISCOUNT_APPLICABILITY_FLAG, active=True)
+    def test_can_receive_discount_course_requirements(self):
+        """
+        Ensure first purchase offer banner only displays for courses with a non-expired verified mode
+        """
+        applicability = can_receive_discount(user=self.user, course=self.course)
+        self.assertEqual(applicability, True)
+
+        no_verified_mode_course = CourseFactory(end=now() + timedelta(days=30))
+        applicability = can_receive_discount(user=self.user, course=no_verified_mode_course)
+        self.assertEqual(applicability, False)
+
+        course_that_has_ended = CourseFactory(end=now() - timedelta(days=30))
+        applicability = can_receive_discount(user=self.user, course=course_that_has_ended)
         self.assertEqual(applicability, False)