""" Tests for the Calendar Sync API """


from common.djangoapps.student.tests.factories import UserFactory
from openedx.features.calendar_sync.api import subscribe_user_to_calendar, unsubscribe_user_to_calendar
from openedx.features.calendar_sync.models import UserCalendarSyncConfig
from xmodule.modulestore.tests.django_utils import SharedModuleStoreTestCase  # lint-amnesty, pylint: disable=wrong-import-order
from xmodule.modulestore.tests.factories import CourseFactory  # lint-amnesty, pylint: disable=wrong-import-order

TEST_PASSWORD = 'test'


class TestCalendarSyncAPI(SharedModuleStoreTestCase):
    """ Tests for the Calendar Sync API """
    @classmethod
    def setUpClass(cls):
        """ Set up any course data """
        super().setUpClass()
        cls.course = CourseFactory.create()
        cls.course_key = cls.course.id

    def setUp(self):
        super().setUp()
        self.user = UserFactory(password=TEST_PASSWORD)

    def test_subscribe_to_calendar(self):
        assert UserCalendarSyncConfig.objects.count() == 0
        subscribe_user_to_calendar(self.user, self.course_key)
        assert UserCalendarSyncConfig.objects.count() == 1
        assert UserCalendarSyncConfig.is_enabled_for_course(self.user, self.course_key)

    def test_unsubscribe_to_calendar(self):
        assert UserCalendarSyncConfig.objects.count() == 0
        unsubscribe_user_to_calendar(self.user, self.course_key)
        assert UserCalendarSyncConfig.objects.count() == 0

        UserCalendarSyncConfig.objects.create(user=self.user, course_key=self.course_key, enabled=True)
        assert UserCalendarSyncConfig.is_enabled_for_course(self.user, self.course_key)
        unsubscribe_user_to_calendar(self.user, self.course_key)
        assert not UserCalendarSyncConfig.is_enabled_for_course(self.user, self.course_key)