fix: validate bulk JPEG uploads only once

This commit is contained in:
2026-08-03 15:46:28 +02:00
parent d54fc646ee
commit e1153cfce4
6 changed files with 38 additions and 16 deletions
+21 -3
View File
@@ -3,6 +3,7 @@ from io import BytesIO
from django.core.files.uploadedfile import SimpleUploadedFile
from django.test import SimpleTestCase
from django.utils.datastructures import MultiValueDict
from extras.forms import ImageAttachmentForm
from PIL import Image
from netbox_utilities.forms import BulkImageUploadForm
@@ -10,10 +11,11 @@ from netbox_utilities.forms import BulkImageUploadForm
class BulkImageUploadFormTest(SimpleTestCase):
@staticmethod
def _image(name):
def _image(name, image_format="PNG"):
content = BytesIO()
Image.new("RGB", (2, 2), "white").save(content, format="PNG")
return SimpleUploadedFile(name, content.getvalue(), content_type="image/png")
Image.new("RGB", (2, 2), "white").save(content, format=image_format)
content_type = "image/jpeg" if image_format == "JPEG" else "image/png"
return SimpleUploadedFile(name, content.getvalue(), content_type=content_type)
def test_accepts_multiple_images(self):
files = MultiValueDict({"images": [self._image("front.png"), self._image("rear.png")]})
@@ -24,6 +26,22 @@ class BulkImageUploadFormTest(SimpleTestCase):
self.assertEqual(len(form.cleaned_data["images"]), 2)
self.assertEqual(form.cleaned_data["description"], "Dokumentation")
def test_jpegs_are_validated_once_by_netbox(self):
files = MultiValueDict(
{
"images": [
self._image("front.JPG", "JPEG"),
self._image("rear.jpg", "JPEG"),
]
}
)
form = BulkImageUploadForm(data={}, files=files)
self.assertTrue(form.is_valid(), form.errors)
for image in form.cleaned_data["images"]:
image.seek(0)
self.assertIsNotNone(ImageAttachmentForm.base_fields["image"].clean(image))
def test_requires_at_least_one_image(self):
form = BulkImageUploadForm(data={}, files={})