Đơn vị Kiểm tra một mẫu Django với một FileField

Question

Tôi có một hình thức như:

#forms.py 
from django import forms 

class MyForm(forms.Form): 
    title = forms.CharField() 
    file = forms.FileField() 


#tests.py 
from django.test import TestCase 
from forms import MyForm 

class FormTestCase(TestCase) 
    def test_form(self): 
     upload_file = open('path/to/file', 'r') 
     post_dict = {'title': 'Test Title'} 
     file_dict = {} #?????? 
     form = MyForm(post_dict, file_dict) 
     self.assertTrue(form.is_valid()) 

Làm thế nào để xây dựng file_dict để vượt qua upload_file mẫu?

Answer 1

Cho đến nay tôi đã tìm thấy con đường này mà làm việc

from django.core.files.uploadedfile import SimpleUploadedFile 
... 
def test_form(self): 
     upload_file = open('path/to/file', 'rb') 
     post_dict = {'title': 'Test Title'} 
     file_dict = {'file': SimpleUploadedFile(upload_file.name, upload_file.read())} 
     form = MyForm(post_dict, file_dict) 
     self.assertTrue(form.is_valid()) 

Answer 2

Có thể đây không phải là khá chính xác, nhưng tôi là tạo ra tập tin hình ảnh trong bài kiểm tra đơn vị sử dụng StringIO:

imgfile = StringIO('GIF87a\x01\x00\x01\x00\x80\x01\x00\x00\x00\x00ccc,\x00' 
        '\x00\x00\x00\x01\x00\x01\x00\x00\x02\x02D\x01\x00;') 
imgfile.name = 'test_img_file.gif' 

response = self.client.post(url, {'file': imgfile}) 

Answer 3

Đây là một cách không yêu cầu bạn sử dụng hình ảnh thực.

from PIL import Image 
from StringIO import StringIO # Python 3: from io import StringIO 
from django.core.files.uploadedfile import InMemoryUploadedFile 

... 

def test_form(self): 
    im = Image.new(mode='RGB', size=(200, 200)) # create a new image using PIL 
    im_io = StringIO() # a StringIO object for saving image 
    im.save(im_io, 'JPEG') # save the image to im_io 
    im_io.seek(0) # seek to the beginning 

    image = InMemoryUploadedFile(
     im_io, None, 'random-name.jpg', 'image/jpeg', im_io.len, None 
    ) 

    post_dict = {'title': 'Test Title'} 
    file_dict = {'picture': image} 

    form = MyForm(data=post_dict, files=file_dict)