151bd13666
* Generate common/djangoapps import shims for LMS * Generate common/djangoapps import shims for Studio * Stop appending project root to sys.path * Stop appending common/djangoapps to sys.path * Import from common.djangoapps.course_action_state instead of course_action_state * Import from common.djangoapps.course_modes instead of course_modes * Import from common.djangoapps.database_fixups instead of database_fixups * Import from common.djangoapps.edxmako instead of edxmako * Import from common.djangoapps.entitlements instead of entitlements * Import from common.djangoapps.pipline_mako instead of pipeline_mako * Import from common.djangoapps.static_replace instead of static_replace * Import from common.djangoapps.student instead of student * Import from common.djangoapps.terrain instead of terrain * Import from common.djangoapps.third_party_auth instead of third_party_auth * Import from common.djangoapps.track instead of track * Import from common.djangoapps.util instead of util * Import from common.djangoapps.xblock_django instead of xblock_django * Add empty common/djangoapps/__init__.py to fix pytest collection * Fix pylint formatting violations * Exclude import_shims/ directory tree from linting
41 lines
1.8 KiB
Python
41 lines
1.8 KiB
Python
""" Tests for the Calendar Sync API """
|
|
|
|
|
|
from openedx.features.calendar_sync.api import subscribe_user_to_calendar, unsubscribe_user_to_calendar
|
|
from openedx.features.calendar_sync.models import UserCalendarSyncConfig
|
|
from common.djangoapps.student.tests.factories import UserFactory
|
|
from xmodule.modulestore.tests.django_utils import SharedModuleStoreTestCase
|
|
from xmodule.modulestore.tests.factories import CourseFactory
|
|
|
|
TEST_PASSWORD = 'test'
|
|
|
|
|
|
class TestCalendarSyncAPI(SharedModuleStoreTestCase):
|
|
""" Tests for the Calendar Sync API """
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
""" Set up any course data """
|
|
super(TestCalendarSyncAPI, cls).setUpClass()
|
|
cls.course = CourseFactory.create()
|
|
cls.course_key = cls.course.id
|
|
|
|
def setUp(self):
|
|
super(TestCalendarSyncAPI, self).setUp()
|
|
self.user = UserFactory(password=TEST_PASSWORD)
|
|
|
|
def test_subscribe_to_calendar(self):
|
|
self.assertEqual(UserCalendarSyncConfig.objects.count(), 0)
|
|
subscribe_user_to_calendar(self.user, self.course_key)
|
|
self.assertEqual(UserCalendarSyncConfig.objects.count(), 1)
|
|
self.assertTrue(UserCalendarSyncConfig.is_enabled_for_course(self.user, self.course_key))
|
|
|
|
def test_unsubscribe_to_calendar(self):
|
|
self.assertEqual(UserCalendarSyncConfig.objects.count(), 0)
|
|
unsubscribe_user_to_calendar(self.user, self.course_key)
|
|
self.assertEqual(UserCalendarSyncConfig.objects.count(), 0)
|
|
|
|
UserCalendarSyncConfig.objects.create(user=self.user, course_key=self.course_key, enabled=True)
|
|
self.assertTrue(UserCalendarSyncConfig.is_enabled_for_course(self.user, self.course_key))
|
|
unsubscribe_user_to_calendar(self.user, self.course_key)
|
|
self.assertFalse(UserCalendarSyncConfig.is_enabled_for_course(self.user, self.course_key))
|