9cf2f9f298
This will remove imports from __future__ that are no longer needed. https://docs.python.org/3.5/library/2to3.html#2to3fixer-future
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
"""
|
|
Tests access.py
|
|
"""
|
|
|
|
|
|
from django.contrib.auth.models import User
|
|
from django.test import TestCase
|
|
from opaque_keys.edx.locator import CourseLocator
|
|
|
|
from contentstore.views.access import get_user_role
|
|
from student.auth import add_users
|
|
from student.roles import CourseInstructorRole, CourseStaffRole
|
|
from student.tests.factories import AdminFactory
|
|
|
|
|
|
class RolesTest(TestCase):
|
|
"""
|
|
Tests for lti user role serialization.
|
|
"""
|
|
def setUp(self):
|
|
""" Test case setup """
|
|
super(RolesTest, self).setUp()
|
|
|
|
self.global_admin = AdminFactory()
|
|
self.instructor = User.objects.create_user('testinstructor', 'testinstructor+courses@edx.org', 'foo')
|
|
self.staff = User.objects.create_user('teststaff', 'teststaff+courses@edx.org', 'foo')
|
|
self.course_key = CourseLocator('mitX', '101', 'test')
|
|
|
|
def test_get_user_role_instructor(self):
|
|
"""
|
|
Verifies if user is instructor.
|
|
"""
|
|
add_users(self.global_admin, CourseInstructorRole(self.course_key), self.instructor)
|
|
self.assertEqual(
|
|
'instructor',
|
|
get_user_role(self.instructor, self.course_key)
|
|
)
|
|
add_users(self.global_admin, CourseStaffRole(self.course_key), self.staff)
|
|
self.assertEqual(
|
|
'instructor',
|
|
get_user_role(self.instructor, self.course_key)
|
|
)
|
|
|
|
def test_get_user_role_staff(self):
|
|
"""
|
|
Verifies if user is staff.
|
|
"""
|
|
add_users(self.global_admin, CourseStaffRole(self.course_key), self.staff)
|
|
self.assertEqual(
|
|
'staff',
|
|
get_user_role(self.staff, self.course_key)
|
|
)
|