5cc8da127a
Positive test case and improvements including: requires course id to be correctly encoded in the url params, only checks entitlements of the course in question, better source for the basket url, check enrollment upgrade deadline
102 lines
3.6 KiB
Python
102 lines
3.6 KiB
Python
"""
|
|
Tests for experimentation views
|
|
"""
|
|
from __future__ import absolute_import
|
|
|
|
from uuid import uuid4
|
|
import six
|
|
|
|
from django.urls import reverse
|
|
from rest_framework.test import APITestCase
|
|
|
|
from course_modes.tests.factories import CourseModeFactory
|
|
from lms.djangoapps.course_blocks.transformers.tests.helpers import ModuleStoreTestCase
|
|
from student.tests.factories import UserFactory
|
|
|
|
from openedx.core.djangoapps.waffle_utils.testutils import override_waffle_flag
|
|
from xmodule.modulestore.tests.factories import CourseFactory
|
|
|
|
from lms.djangoapps.experiments.views_custom import MOBILE_UPSELL_FLAG
|
|
|
|
CROSS_DOMAIN_REFERER = 'https://ecommerce.edx.org'
|
|
|
|
|
|
class Rev934LoggedOutTests(APITestCase):
|
|
def test_not_logged_in(self):
|
|
"""Test mobile app upsell API is not available if not logged in"""
|
|
url = reverse('api_experiments:rev_934')
|
|
|
|
# Not-logged-in returns 401
|
|
response = self.client.get(url)
|
|
self.assertEqual(response.status_code, 401)
|
|
|
|
|
|
class Rev934Tests(APITestCase, ModuleStoreTestCase):
|
|
"""Test mobile app upsell API"""
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(Rev934Tests, cls).setUpClass()
|
|
cls.url = reverse('api_experiments:rev_934')
|
|
|
|
def setUp(self):
|
|
super(Rev934Tests, self).setUp()
|
|
user = UserFactory(username='robot-mue-1-6pnjv') # Username that hashes to bucket 1
|
|
self.client.login(
|
|
username=user.username,
|
|
password=UserFactory._DEFAULT_PASSWORD, # pylint: disable=protected-access
|
|
)
|
|
|
|
@override_waffle_flag(MOBILE_UPSELL_FLAG, active=False)
|
|
def test_flag_off(self):
|
|
response = self.client.get(self.url)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.data['show_upsell'], False)
|
|
self.assertEqual(response.data['upsell_flag'], False)
|
|
|
|
@override_waffle_flag(MOBILE_UPSELL_FLAG, active=True)
|
|
def test_no_course_id(self):
|
|
response = self.client.get(self.url)
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
@override_waffle_flag(MOBILE_UPSELL_FLAG, active=True)
|
|
def test_bad_course_id(self):
|
|
response = self.client.get(self.url, {'course_id': 'junk'})
|
|
self.assertEqual(response.status_code, 400)
|
|
|
|
@override_waffle_flag(MOBILE_UPSELL_FLAG, active=True)
|
|
def test_simple_course(self):
|
|
course = CourseFactory.create()
|
|
response = self.client.get(self.url, {'course_id': course.id})
|
|
self.assertEqual(response.status_code, 200)
|
|
expected = {
|
|
'show_upsell': False,
|
|
'upsell_flag': True,
|
|
'experiment_bucket': 1,
|
|
'user_upsell': True,
|
|
'basket_link': None, # No sku means no basket link so no upsell
|
|
}
|
|
self.assertEqual(response.data, expected)
|
|
|
|
@override_waffle_flag(MOBILE_UPSELL_FLAG, active=True)
|
|
def test_course(self):
|
|
course = CourseFactory.create(run='test', display_name='test')
|
|
CourseModeFactory.create(
|
|
mode_slug="verified",
|
|
course_id=course.id,
|
|
min_price=10,
|
|
sku=six.text_type(uuid4().hex)
|
|
)
|
|
|
|
response = self.client.get(self.url, {'course_id': course.id})
|
|
self.assertEqual(response.status_code, 200)
|
|
result = response.data
|
|
self.assertIn('basket_url', result)
|
|
self.assertTrue(bool(result['basket_url']))
|
|
expected = {
|
|
'show_upsell': True,
|
|
'price': u'$10',
|
|
'basket_url': result['basket_url'],
|
|
# Example basket_url: u'/verify_student/upgrade/org.0/course_0/test/'
|
|
}
|
|
self.assertEqual(response.data, expected)
|