Discovered a case I missed on the original version of this. See https://github.com/edx/edx-platform/pull/5798 JIRA: MA-76 Also fixes checking for mobile_access when course is None JIRA: MOB-978
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
"""
|
|
Tests for mobile API utilities
|
|
"""
|
|
|
|
import ddt
|
|
from rest_framework.test import APITestCase
|
|
|
|
from courseware.tests.factories import UserFactory
|
|
from student import auth
|
|
|
|
from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
|
|
from xmodule.modulestore.tests.factories import CourseFactory
|
|
|
|
from .utils import mobile_available_when_enrolled
|
|
|
|
ROLE_CASES = (
|
|
(auth.CourseBetaTesterRole, True),
|
|
(auth.CourseStaffRole, True),
|
|
(auth.CourseInstructorRole, True),
|
|
(None, False)
|
|
)
|
|
|
|
|
|
@ddt.ddt
|
|
class TestMobileApiUtils(ModuleStoreTestCase, APITestCase):
|
|
"""
|
|
Tests for mobile API utilities
|
|
"""
|
|
|
|
def setUp(self):
|
|
self.user = UserFactory.create()
|
|
|
|
@ddt.data(*ROLE_CASES)
|
|
@ddt.unpack
|
|
def test_mobile_role_access(self, role, should_have_access):
|
|
"""
|
|
Verifies that our mobile access function properly handles using roles to grant access
|
|
"""
|
|
course = CourseFactory.create(mobile_available=False)
|
|
if role:
|
|
role(course.id).add_users(self.user)
|
|
self.assertEqual(should_have_access, mobile_available_when_enrolled(course, self.user))
|
|
|
|
def test_mobile_explicit_access(self):
|
|
"""
|
|
Verifies that our mobile access function listens to the mobile_available flag as it should
|
|
"""
|
|
course = CourseFactory.create(mobile_available=True)
|
|
self.assertTrue(mobile_available_when_enrolled(course, self.user))
|
|
|
|
def test_missing_course(self):
|
|
"""
|
|
Verifies that we handle the case where a course doesn't exist
|
|
"""
|
|
self.assertFalse(mobile_available_when_enrolled(None, self.user))
|