From 8299e0e72770dd13d22c372dc6e612fd15dd81a0 Mon Sep 17 00:00:00 2001 From: Matt Hughes Date: Fri, 15 Nov 2019 11:43:36 -0500 Subject: [PATCH] Add test of multiple folks in the upload --- .../commands/tests/test_migrate_saml_uids.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/lms/djangoapps/program_enrollments/management/commands/tests/test_migrate_saml_uids.py b/lms/djangoapps/program_enrollments/management/commands/tests/test_migrate_saml_uids.py index 2003677d32..cdebd435df 100644 --- a/lms/djangoapps/program_enrollments/management/commands/tests/test_migrate_saml_uids.py +++ b/lms/djangoapps/program_enrollments/management/commands/tests/test_migrate_saml_uids.py @@ -6,6 +6,7 @@ from __future__ import absolute_import from django.core.management import call_command from django.test import TestCase +import six from factory import LazyAttributeSequence, SubFactory from factory.django import DjangoModelFactory from lms.djangoapps.program_enrollments.management.commands import migrate_saml_uids @@ -136,3 +137,15 @@ class TestMigrateSamlUids(TestCase): mock_info.assert_any_call('Number of users with {slug} UserSocialAuth records for which there was no mapping in the provided file: 1'.format( slug=self.provider_slug )) + + def test_several_learners(self): + auths = [UserSocialAuthFactory() for _ in range(5)] + new_urn = '9001' + + self._call_command('[{}]'.format( + ','.join([self._format_email_uid_pair(auth.user.email, new_urn + six.text_type(ind)) for ind, auth in enumerate(auths)]) + )) + + for ind, auth in enumerate(auths): + auth.refresh_from_db() + assert auth.uid == self._format_slug_urn_pair(self.provider_slug, new_urn + six.text_type(ind))