/Python/Django/How do I mock an uploaded file in django?

How do I mock an uploaded file in django?

from django.core.files.uploadedfile import SimpleUploadedFile

# To mock a simple raw text file
mock_profile_image = SimpleUploadedFile('simple_file.txt', 'raw-text', content_type='text')

# To mock an image

# Just invoke this function and you'll get a png file
def get_mock_img(name='test.png', ext='png', size=(50, 50), color=(256, 0, 0)):
    file_obj = StringIO()
    image = Image.new("RGB", size=size, color=color)
    image.save(file_obj, ext)
    file_obj.seek(0)
    return File(file_obj, name=name)

mock_profile_image = SimpleUploadedFile('profile_img.png', get_mock_img(), content_type='image/png')

# or
with open(img_path, 'rb') as img_file:
	mock_profile_image = SimpleUploadedFile('profile_img.png', img_file.read(), content_type='image/png')

Similar codes

0 vote393 views1 answer
No module named 'django_hitcounter'
Python Django
0 vote332 views2 answers
MultiValueDictKeyError django
Python Django
1 vote343 views1 answer
Display data from database in django
Python Django
1 vote358 views1 answer
Django model query add annotation field to show duplicate count
Python Django
0 vote357 views1 answer
Generate 50 characters long for django
Python Django
10 votes350 views2 answers
Django import settings
Python Django
0 vote352 views1 answer
Python Django How To Bulk Update
Python Django
2 votes397 views1 answer
Django rest framework viewset perform_update
Python Django
0 vote405 views1 answer
Django connect to a db on heroku
Python Django
0 vote384 views1 answer
Decode in django templates
Python Django