1025 - Add test_dispatch_to_batch

api/scpca_portal/test/factories.py

@@ -169,4 +169,15 @@ class Meta:
 class ProjectFactory(LeafProjectFactory):
     computed_file1 = factory.RelatedFactory(ProjectComputedFileFactory, "project")
     sample1 = factory.RelatedFactory(SampleFactory, "project")
+    library1 = factory.RelatedFactory(LibraryFactory, "project")
     summary1 = factory.RelatedFactory(ProjectSummaryFactory, factory_related_name="project")
+
+    @factory.post_generation
+    def add_sample_library_relation(self, create, extracted, **kwargs):
+        if not create:
+            return
+
+        sample = self.samples.first()
+        library = self.libraries.first()
+
+        sample.libraries.add(library)

api/scpca_portal/test/management/commands/test_dispatch_to_batch.py

@@ -0,0 +1,42 @@
+from functools import partial
+from unittest.mock import patch
+
+from django.core.management import call_command
+from django.test import TestCase
+
+from scpca_portal.test.factories import ProjectFactory
+
+
+class TestLoadData(TestCase):
+    def setUp(self):
+        self.dispatch_to_batch = partial(call_command, "dispatch_to_batch")
+
+    @patch("scpca_portal.management.commands.dispatch_to_batch.Command.submit_job")
+    def test_only_projects_without_computed_files_submitted(self, mock_submit_job):
+        project_with_files = ProjectFactory()
+
+        project_no_files = ProjectFactory()
+        project_no_files.computed_files.delete()
+
+        # TODO: remove this patch when we get rid of sample computed file generation
+        with patch("scpca_portal.models.Project.samples_to_generate", []):
+            self.dispatch_to_batch()
+        mock_submit_job.assert_called()
+
+        submitted_project_ids = set(
+            call.kwargs["project_id"] for call in mock_submit_job.call_args_list
+        )
+        self.assertIn(project_no_files.scpca_id, submitted_project_ids)
+        self.assertNotIn(project_with_files.scpca_id, submitted_project_ids)
+
+    @patch("scpca_portal.management.commands.dispatch_to_batch.Command.submit_job")
+    def test_project_id_submission(self, mock_submit_job):
+        project = ProjectFactory()
+
+        # TODO: remove this patch when we get rid of sample computed file generation
+        with patch("scpca_portal.models.Project.samples_to_generate", []):
+            self.dispatch_to_batch(project_id=project.scpca_id)
+        mock_submit_job.assert_called()
+
+        submitted_project_id = mock_submit_job.call_args.kwargs["project_id"]
+        self.assertEqual(project.scpca_id, submitted_project_id)