Add test for specifying locator to migrate_to_split command

cms/djangoapps/contentstore/management/commands/tests/test_migrate_to_split.py

@@ -10,8 +10,8 @@ from contentstore.management.commands.migrate_to_split import Command
 from contentstore.tests.modulestore_config import TEST_MODULESTORE
 from xmodule.modulestore.tests.django_utils import ModuleStoreTestCase
 from xmodule.modulestore.tests.factories import CourseFactory
-from xmodule.modulestore.django import modulestore
-from xmodule.modulestore.django import loc_mapper
+from xmodule.modulestore.django import modulestore, loc_mapper
+from xmodule.modulestore.locator import CourseLocator
 # pylint: disable=E1101
 
 
@@ -57,7 +57,7 @@ class TestMigrateToSplit(ModuleStoreTestCase):
         self.user = User.objects.create_user(uname, email, password)
         self.course = CourseFactory()
 
-    def test_happy_path_email(self):
+    def test_user_email(self):
         call_command(
             "migrate_to_split",
             str(self.course.location),
@@ -67,7 +67,7 @@ class TestMigrateToSplit(ModuleStoreTestCase):
         course_from_split = modulestore('split').get_course(locator)
         self.assertIsNotNone(course_from_split)
 
-    def test_happy_path_user_id(self):
+    def test_user_id(self):
         call_command(
             "migrate_to_split",
             str(self.course.location),
@@ -76,3 +76,14 @@ class TestMigrateToSplit(ModuleStoreTestCase):
         locator = loc_mapper().translate_location(self.course.id, self.course.location)
         course_from_split = modulestore('split').get_course(locator)
         self.assertIsNotNone(course_from_split)
+
+    def test_locator_string(self):
+        call_command(
+            "migrate_to_split",
+            str(self.course.location),
+            str(self.user.id),
+            "org.dept.name.run",
+        )
+        locator = CourseLocator(package_id="org.dept.name.run", branch="published")
+        course_from_split = modulestore('split').get_course(locator)
+        self.assertIsNotNone(course_from_split)