78 lines
3.2 KiB
Python
78 lines
3.2 KiB
Python
"""
|
|
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.
|
|
This also validates history records are created on both create and update.
|
|
"""
|
|
assert ProgramEnrollment.objects.count() == 0
|
|
assert ProgramEnrollment.historical_records.count() == 0 # pylint: disable=no-member
|
|
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
|
|
assert ProgramEnrollment.historical_records.count() == 1 # pylint: disable=no-member
|
|
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.historical_records.count() == 2 # pylint: disable=no-member
|
|
assert ProgramEnrollment.objects.filter(status=PEStatuses.ENDED).exists()
|