2
0
mirror of https://github.com/inventree/InvenTree.git synced 2025-05-02 13:28:49 +00:00

Add unit tests for .webp support

This commit is contained in:
Oliver Walters 2022-05-15 23:36:41 +10:00
parent 495798dc98
commit 55f87033b2

View File

@ -970,24 +970,29 @@ class PartDetailTests(InvenTreeAPITestCase):
) )
self.assertEqual(response.status_code, 400) self.assertEqual(response.status_code, 400)
self.assertIn('Upload a valid image', str(response.data))
# Now try to upload a valid image file # Now try to upload a valid image file, in multiple formats
img = PIL.Image.new('RGB', (128, 128), color='red') for fmt in ['jpg', 'png', 'bmp', 'webp']:
img.save('dummy_image.jpg') fn = f'dummy_image.{fmt}'
with open('dummy_image.jpg', 'rb') as dummy_image: img = PIL.Image.new('RGB', (128, 128), color='red')
response = upload_client.patch( img.save(fn)
url,
{
'image': dummy_image,
},
format='multipart',
)
self.assertEqual(response.status_code, 200) with open(fn, 'rb') as dummy_image:
response = upload_client.patch(
url,
{
'image': dummy_image,
},
format='multipart',
)
# And now check that the image has been set self.assertEqual(response.status_code, 200)
p = Part.objects.get(pk=pk)
# And now check that the image has been set
p = Part.objects.get(pk=pk)
self.assertIsNotNone(p.image)
def test_details(self): def test_details(self):
""" """