feat: filter image tasks by input count

2026-07-20 09:03:23 +08:00
parent 33d744d6a9
commit 5dc60a204a
6 changed files with 92 additions and 6 deletions
@@ -801,6 +801,61 @@ class ImageGenerationTaskAdminTests(TestCase):

        self.assertEqual(response.status_code, 302)

+    def test_changelist_filters_new_and_legacy_single_image_tasks(self):
+        new_single = self.create_task()
+        self.add_input_image(new_single, ordinal=0, filename="new-single.png")
+        legacy_single = self.create_task()
+        legacy_single.input_image.save("legacy-single.png", ContentFile(b"legacy-image"), save=True)
+        multiple = self.create_task()
+        self.add_input_image(multiple, ordinal=0, filename="multiple-main.png")
+        self.add_input_image(multiple, ordinal=1, filename="multiple-reference.png")
+        no_input = self.create_task()
+
+        response = self.client.get(
+            reverse("admin:api_imagegenerationtask_changelist"),
+            {"input_image_type": "single"},
+        )
+
+        self.assertEqual(response.status_code, 200)
+        self.assertContains(response, "输入图片类型")
+        self.assertContains(response, "单图生图")
+        self.assertContains(response, "多图生图")
+        self.assertContains(response, str(new_single.task_id))
+        self.assertContains(response, str(legacy_single.task_id))
+        self.assertNotContains(response, str(multiple.task_id))
+        self.assertNotContains(response, str(no_input.task_id))
+        self.assertCountEqual(
+            response.context["cl"].queryset.values_list("pk", flat=True),
+            [new_single.pk, legacy_single.pk],
+        )
+
+    def test_changelist_filters_multiple_image_tasks_and_composes_with_status_filter(self):
+        succeeded_multiple = self.create_task(status=ImageGenerationTask.Status.SUCCEEDED)
+        self.add_input_image(succeeded_multiple, ordinal=0, filename="succeeded-main.png")
+        self.add_input_image(succeeded_multiple, ordinal=1, filename="succeeded-reference.png")
+        failed_multiple = self.create_task(status=ImageGenerationTask.Status.FAILED)
+        self.add_input_image(failed_multiple, ordinal=0, filename="failed-main.png")
+        self.add_input_image(failed_multiple, ordinal=1, filename="failed-reference.png")
+        single = self.create_task()
+        self.add_input_image(single, ordinal=0, filename="single.png")
+
+        response = self.client.get(
+            reverse("admin:api_imagegenerationtask_changelist"),
+            {
+                "input_image_type": "multiple",
+                "status__exact": ImageGenerationTask.Status.SUCCEEDED,
+            },
+        )
+
+        self.assertEqual(response.status_code, 200)
+        self.assertContains(response, str(succeeded_multiple.task_id))
+        self.assertNotContains(response, str(failed_multiple.task_id))
+        self.assertNotContains(response, str(single.task_id))
+        self.assertEqual(
+            list(response.context["cl"].queryset.values_list("pk", flat=True)),
+            [succeeded_multiple.pk],
+        )
+

@override_settings(
    PAYMENT_CALLBACK_MODE="mock",