86 lines
3.5 KiB
Python
86 lines
3.5 KiB
Python
from django.db import IntegrityError
|
|
from django.test import TestCase
|
|
from xmodule.modulestore.tests.factories import CourseFactory
|
|
from student.tests.factories import UserFactory
|
|
from user_api.tests.factories import UserPreferenceFactory, UserCourseTagFactory, UserOrgTagFactory
|
|
from user_api.models import UserPreference
|
|
|
|
|
|
class UserPreferenceModelTest(TestCase):
|
|
def test_duplicate_user_key(self):
|
|
user = UserFactory.create()
|
|
UserPreferenceFactory.create(user=user, key="testkey", value="first")
|
|
self.assertRaises(
|
|
IntegrityError,
|
|
UserPreferenceFactory.create,
|
|
user=user,
|
|
key="testkey",
|
|
value="second"
|
|
)
|
|
|
|
def test_arbitrary_values(self):
|
|
user = UserFactory.create()
|
|
self._create_and_assert(user=user, key="testkey0", value="")
|
|
self._create_and_assert(user=user, key="testkey1", value="This is some English text!")
|
|
self._create_and_assert(user=user, key="testkey2", value="{'some': 'json'}")
|
|
self._create_and_assert(
|
|
user=user,
|
|
key="testkey3",
|
|
value="\xe8\xbf\x99\xe6\x98\xaf\xe4\xb8\xad\xe5\x9b\xbd\xe6\x96\x87\xe5\xad\x97'"
|
|
)
|
|
|
|
def _create_and_assert(self, user, key, value):
|
|
"""Create a new preference and assert the values. """
|
|
preference = UserPreferenceFactory.create(user=user, key=key, value=value)
|
|
self.assertEqual(preference.user, user)
|
|
self.assertEqual(preference.key, key)
|
|
self.assertEqual(preference.value, value)
|
|
return preference
|
|
|
|
def test_create_user_course_tags(self):
|
|
"""Create user preference tags and confirm properties are set accordingly. """
|
|
user = UserFactory.create()
|
|
course = CourseFactory.create()
|
|
tag = UserCourseTagFactory.create(user=user, course_id=course.id, key="testkey", value="foobar")
|
|
self.assertEquals(tag.user, user)
|
|
self.assertEquals(tag.course_id, course.id)
|
|
self.assertEquals(tag.key, "testkey")
|
|
self.assertEquals(tag.value, "foobar")
|
|
|
|
def test_create_user_org_tags(self):
|
|
"""Create org specific user tags and confirm all properties are set """
|
|
user = UserFactory.create()
|
|
course = CourseFactory.create()
|
|
tag = UserOrgTagFactory.create(user=user, org=course.id.org, key="testkey", value="foobar")
|
|
self.assertEquals(tag.user, user)
|
|
self.assertEquals(tag.org, course.id.org)
|
|
self.assertEquals(tag.key, "testkey")
|
|
self.assertEquals(tag.value, "foobar")
|
|
self.assertIsNotNone(tag.created)
|
|
self.assertIsNotNone(tag.modified)
|
|
|
|
# Modify the tag and save it. Check if the modified timestamp is updated.
|
|
original_modified = tag.modified
|
|
tag.value = "barfoo"
|
|
tag.save()
|
|
self.assertEquals(tag.value, "barfoo")
|
|
self.assertNotEqual(original_modified, tag.modified)
|
|
|
|
def test_get_set_preference(self):
|
|
# Checks that you can set a preference and get that preference later
|
|
# Also, tests that no preference is returned for keys that are not set
|
|
|
|
user = UserFactory.create()
|
|
key = 'testkey'
|
|
value = 'testvalue'
|
|
|
|
# does a round trip
|
|
UserPreference.set_preference(user, key, value)
|
|
pref = UserPreference.get_preference(user, key)
|
|
|
|
self.assertEqual(pref, value)
|
|
|
|
# get preference for key that doesn't exist for user
|
|
pref = UserPreference.get_preference(user, 'testkey_none')
|
|
self.assertIsNone(pref)
|