182 lines
6.6 KiB
Python
182 lines
6.6 KiB
Python
from django.test import TestCase
|
|
from student.tests.factories import UserFactory, CourseEnrollmentFactory
|
|
from django_comment_common.models import Role, Permission
|
|
from factories import RoleFactory
|
|
import django_comment_client.utils as utils
|
|
|
|
|
|
class DictionaryTestCase(TestCase):
|
|
def test_extract(self):
|
|
d = {'cats': 'meow', 'dogs': 'woof'}
|
|
k = ['cats', 'dogs', 'hamsters']
|
|
expected = {'cats': 'meow', 'dogs': 'woof', 'hamsters': None}
|
|
self.assertEqual(utils.extract(d, k), expected)
|
|
|
|
def test_strip_none(self):
|
|
d = {'cats': 'meow', 'dogs': 'woof', 'hamsters': None}
|
|
expected = {'cats': 'meow', 'dogs': 'woof'}
|
|
self.assertEqual(utils.strip_none(d), expected)
|
|
|
|
def test_strip_blank(self):
|
|
d = {'cats': 'meow', 'dogs': 'woof', 'hamsters': ' ', 'yetis': ''}
|
|
expected = {'cats': 'meow', 'dogs': 'woof'}
|
|
self.assertEqual(utils.strip_blank(d), expected)
|
|
|
|
def test_merge_dict(self):
|
|
d1 = {'cats': 'meow', 'dogs': 'woof'}
|
|
d2 = {'lions': 'roar', 'ducks': 'quack'}
|
|
expected = {'cats': 'meow', 'dogs': 'woof', 'lions': 'roar', 'ducks': 'quack'}
|
|
self.assertEqual(utils.merge_dict(d1, d2), expected)
|
|
|
|
|
|
class CategorySortTestCase(TestCase):
|
|
def setUp(self):
|
|
self.category_map = {
|
|
'entries': {
|
|
u'General': {
|
|
'sort_key': u'General'
|
|
}
|
|
},
|
|
'subcategories': {
|
|
u'Tests': {
|
|
'sort_key': u'Tests',
|
|
'subcategories': {},
|
|
'entries': {
|
|
u'Quizzes': {
|
|
'sort_key': None
|
|
}, u'All': {
|
|
'sort_key': None
|
|
}, u'Final Exam': {
|
|
'sort_key': None
|
|
},
|
|
}
|
|
},
|
|
u'Assignments': {
|
|
'sort_key': u'Assignments',
|
|
'subcategories': {},
|
|
'entries': {
|
|
u'Homework': {
|
|
'sort_key': None
|
|
},
|
|
u'All': {
|
|
'sort_key': None
|
|
},
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
def test_alpha_sort_true(self):
|
|
expected_true = {
|
|
'entries': {
|
|
u'General': {
|
|
'sort_key': u'General'
|
|
}
|
|
},
|
|
'children': [u'Assignments', u'General', u'Tests'],
|
|
'subcategories': {
|
|
u'Tests': {
|
|
'sort_key': u'Tests',
|
|
'subcategories': {},
|
|
'children': [u'All', u'Final Exam', u'Quizzes'],
|
|
'entries': {
|
|
u'All': {
|
|
'sort_key': 'All'
|
|
}, u'Final Exam': {
|
|
'sort_key': 'Final Exam'
|
|
}, u'Quizzes': {
|
|
'sort_key': 'Quizzes'
|
|
}
|
|
}
|
|
},
|
|
u'Assignments': {
|
|
'sort_key': u'Assignments',
|
|
'subcategories': {},
|
|
'children': [u'All', u'Homework'],
|
|
'entries': {
|
|
u'Homework': {
|
|
'sort_key': 'Homework'
|
|
},
|
|
u'All': {
|
|
'sort_key': 'All'
|
|
},
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
utils.sort_map_entries(self.category_map, True)
|
|
self.assertEqual(self.category_map, expected_true)
|
|
|
|
def test_alpha_sort_false(self):
|
|
expected_false = {
|
|
'entries': {
|
|
u'General': {
|
|
'sort_key': u'General'
|
|
}
|
|
},
|
|
'children': [u'Assignments', u'General', u'Tests'],
|
|
'subcategories': {
|
|
u'Tests': {
|
|
'sort_key': u'Tests',
|
|
'subcategories': {},
|
|
'children': [u'Quizzes', u'All', u'Final Exam'],
|
|
'entries': {
|
|
u'Quizzes': {
|
|
'sort_key': None
|
|
}, u'All': {
|
|
'sort_key': None
|
|
}, u'Final Exam': {
|
|
'sort_key': None
|
|
},
|
|
}
|
|
},
|
|
u'Assignments': {
|
|
'sort_key': u'Assignments',
|
|
'subcategories': {},
|
|
'children': [u'All', u'Homework'],
|
|
'entries': {
|
|
u'Homework': {
|
|
'sort_key': None
|
|
},
|
|
u'All': {
|
|
'sort_key': None
|
|
},
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
utils.sort_map_entries(self.category_map, False)
|
|
self.assertEqual(self.category_map, expected_false)
|
|
|
|
|
|
class AccessUtilsTestCase(TestCase):
|
|
def setUp(self):
|
|
self.course_id = 'edX/toy/2012_Fall'
|
|
self.student_role = RoleFactory(name='Student', course_id=self.course_id)
|
|
self.moderator_role = RoleFactory(name='Moderator', course_id=self.course_id)
|
|
self.student1 = UserFactory(username='student', email='student@edx.org')
|
|
self.student1_enrollment = CourseEnrollmentFactory(user=self.student1)
|
|
self.student_role.users.add(self.student1)
|
|
self.student2 = UserFactory(username='student2', email='student2@edx.org')
|
|
self.student2_enrollment = CourseEnrollmentFactory(user=self.student2)
|
|
self.moderator = UserFactory(username='moderator', email='staff@edx.org', is_staff=True)
|
|
self.moderator_enrollment = CourseEnrollmentFactory(user=self.moderator)
|
|
self.moderator_role.users.add(self.moderator)
|
|
|
|
def test_get_role_ids(self):
|
|
ret = utils.get_role_ids(self.course_id)
|
|
expected = {u'Moderator': [3], u'Student': [1, 2], 'Staff': [3]}
|
|
self.assertEqual(ret, expected)
|
|
|
|
def test_has_forum_access(self):
|
|
ret = utils.has_forum_access('student', self.course_id, 'Student')
|
|
self.assertTrue(ret)
|
|
|
|
ret = utils.has_forum_access('not_a_student', self.course_id, 'Student')
|
|
self.assertFalse(ret)
|
|
|
|
ret = utils.has_forum_access('student', self.course_id, 'NotARole')
|
|
self.assertFalse(ret)
|