"""
Tests for program enrollment writing Python API.

Currently, we do not directly unit test the functions in api/writing.py extensively.
This is okay for now because they are all used in
`rest_api.v1.views` and is thus tested through `rest_api.v1.tests.test_views`.
Eventually it would be good to directly test the Python API function and just use
mocks in the view tests.
"""


from uuid import UUID

from django.core.cache import cache
from organizations.tests.factories import OrganizationFactory

from lms.djangoapps.program_enrollments.constants import ProgramEnrollmentStatuses as PEStatuses
from lms.djangoapps.program_enrollments.models import ProgramEnrollment
from openedx.core.djangoapps.catalog.cache import PROGRAM_CACHE_KEY_TPL
from openedx.core.djangoapps.catalog.tests.factories import OrganizationFactory as CatalogOrganizationFactory
from openedx.core.djangoapps.catalog.tests.factories import ProgramFactory
from openedx.core.djangolib.testing.utils import CacheIsolationTestCase
from third_party_auth.tests.factories import SAMLProviderConfigFactory

from ..writing import write_program_enrollments


class WritingProgramEnrollmentTest(CacheIsolationTestCase):
    """
    Test cases for program enrollment writing functions.
    """
    ENABLED_CACHES = ['default']

    organization_key = 'test'

    program_uuid_x = UUID('dddddddd-5f48-493d-9910-84e1d36c657f')

    curriculum_uuid_a = UUID('aaaaaaaa-bd26-4370-94b8-b4063858210b')

    user_0 = 'user-0'

    def setUp(self):
        """
        Set up test data
        """
        super(WritingProgramEnrollmentTest, self).setUp()
        catalog_org = CatalogOrganizationFactory.create(key=self.organization_key)
        program = ProgramFactory.create(
            uuid=self.program_uuid_x,
            authoring_organizations=[catalog_org]
        )
        organization = OrganizationFactory.create(short_name=self.organization_key)
        SAMLProviderConfigFactory.create(organization=organization)
        cache.set(PROGRAM_CACHE_KEY_TPL.format(uuid=self.program_uuid_x), program, None)

    def test_write_program_enrollments_status_ended(self):
        """
        Successfully updates program enrollment to status ended if requested
        """
        assert ProgramEnrollment.objects.count() == 0
        write_program_enrollments(self.program_uuid_x, [{
            'external_user_key': self.user_0,
            'status': PEStatuses.PENDING,
            'curriculum_uuid': self.curriculum_uuid_a,
        }], True, False)
        assert ProgramEnrollment.objects.count() == 1
        write_program_enrollments(self.program_uuid_x, [{
            'external_user_key': self.user_0,
            'status': PEStatuses.ENDED,
            'curriculum_uuid': self.curriculum_uuid_a,
        }], False, True)
        assert ProgramEnrollment.objects.count() == 1
        assert ProgramEnrollment.objects.filter(status=PEStatuses.ENDED).exists()