2012-04-24 85 views
5

我想测试一个自写的FormField AudioFileFormField,它在存储之前检查一个文件是否是audiofile文件。为此我覆盖了to_python方法。试图测试这个FormField我遇到了一些困难。模拟提交测试表单的正确方法是什么?

这是迄今为止我的测试用例:

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:“文件”对象有没有属性“得到”

当我插入一个调试跟踪权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的正确方法是什么?

回答

7

好,有焕然一新的感觉可能是值得很多。这是我的新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()) 
+2

难道你不觉得这会污染你的'upload /'目录吗? – 2016-10-23 10:22:55

5

替代解决方案(因为这个问题是顶级的谷歌结果为“Django的测试模拟上传”):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) 
相关问题