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

Add unit tests for .webp support

Browse Source
This commit is contained in:
Oliver Walters 2022-05-15 23:36:41 +10:00
parent 495798dc98
commit 55f87033b2
1 changed files with 19 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
InvenTree/part/test_api.py
View File

@ -970,12 +970,16 @@ 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, in multiple formats
for fmt in ['jpg', 'png', 'bmp', 'webp']:
fn = f'dummy_image.{fmt}'
# Now try to upload a valid image file
img = PIL.Image.new('RGB', (128, 128), color='red') img = PIL.Image.new('RGB', (128, 128), color='red')
img.save('dummy_image.jpg') img.save(fn)
with open('dummy_image.jpg', 'rb') as dummy_image: with open(fn, 'rb') as dummy_image:
response = upload_client.patch( response = upload_client.patch(
url, url,
{ {
@ -988,6 +992,7 @@ class PartDetailTests(InvenTreeAPITestCase):
# And now check that the image has been set # And now check that the image has been set
p = Part.objects.get(pk=pk) p = Part.objects.get(pk=pk)
self.assertIsNotNone(p.image)
def test_details(self): def test_details(self):
""" """