Question

我正在尝试测试一个自编写的FormField AudioFileFormField，它会在存储之前检查文件是否为audiofile。为此我已经覆盖了to_python方法。试图测试这个FormField我遇到了一些困难。

到目前为止，这是我的TestCase：

from django import forms
from django.core.files.uploadedfile import SimpleUploadedFile
from django.test import TestCase

class TestAudioFileFormField(TestCase):
    """ test the formfield to use for AudioFile Uploads """

    class TestForm(forms.Form):
        audiofilefield = AudioFileFormField()


    def setUp(self):

        self.AUDIOFILE_TEST_ROOT = os.path.dirname(__file__) + '/data/'
        self.files = audiofile_files


    def test_to_python(self):
        """ assign some files to a form and validate the form """

        f = file(self.AUDIOFILE_TEST_ROOT + self.files[0]) 
        file_data = {'audiofilefield':SimpleUploadedFile( self.files[0],f.read() )}
        data = {}

        form = self.TestForm(data,f)
        form.is_valid()

行form.is_valid（）引发了一个AttributeError：＆＃39;文件＆＃39;对象没有属性＆＃39; get＆＃39;

当我在form.is_valid（）之前插入调试跟踪时，这就是我在该交互式会话中得到的内容：

ipdb> form.is_valid()
AttributeError: 'file' object has no attribute 'get'
ipdb> suf = file_data['audiofilefield']
ipdb> suf
<SimpleUploadedFile: invalid_format.mp3 (text/plain)>
ipdb> dir(suf)
[lots of info here]
ipdb> form.is_valid()
True

我在交互式会话中究竟发生了什么变化，以便表单验证工作正常？将文件传递给SimpleUploadedFile以避免AttributeError的正确方法是什么？

Answer 1

好的，新面貌可能非常值得。这是我的新test_to_python测试，这次它可以工作：

def test_to_python(self):
    f = file(self.AUDIOFILE_TEST_ROOT + self.files[0])
    file_data = {'audiofilefield':SimpleUploadedFile(self.files[0],f.read())}
    data = {}
    form = self.TestForm(data,file_data)
    self.assertTrue( form.is_valid() )

Answer 2

替代解决方案（因为这个问题是“Django测试模拟上传”的Google最高结果）：Django的内置测试客户端接受打开的文件对象作为POST数据：

# the form
class TestForm(forms.Form):
    audiofilefield = AudioFileFormField()

# the view
def upload_view(request):
    form = TestForm(data=request.POST or None, files=request.FILES or None)
    if request.method == 'POST':
         if form.is_valid():
              return HttpResponse('success')
         else:
              return HttpResponse('fail')

# the test function
class UploadTest(TestCase):
    def test_upload(self):
        c = Client() # django.test.client.Client
        formdata = {}
        with open('/my/audiofile.mp3', 'rb') as f:
            formdata['audiofilefield'] = f
            response = c.post('/my-test-view/', formdata)

模拟用于测试表单的filedata的正确方法是什么？

2 个答案: