Question

我正在尝试为具有POST方法的类编写单元测试，该方法用于将文档上载到基于web的django应用程序。这是我想要编写单元测试的视图类：

class SOP(APIView):
authentication_classes = (authentication.TokenAuthentication,)
def post(self,request):
    returnDict={}
    returnDict['msg']='File not uploaded'
    #if form.is_valid():        
    newdoc = Document(sopFile = request.FILES['sopFile'])
    newdoc.save()

    returnDict['msg']='File uploaded'
    returnDict['fileName']=newdoc.sopFile.name
    # Redirect to the document list after POST
    return Response(returnDict)

由于我的django应用程序正在使用forms.py上传文件，因此我也将此代码与此一起使用：

from django import forms
class DocumentForm(forms.Form):
docfile = forms.FileField(
    label='Select a file',
    help_text='max. 42 megabytes'
)

我曾尝试使用RequestFactory（）和TestCase（）编写测试用例，但我无法想出如何为这种类/视图编写单元测试......

Answer 1

您可以使用Django的the test client。这很容易使用。

Django docs的示例：

>>> c = Client()
>>> with open('wishlist.doc') as fp:
...     c.post('/customers/wishes/', {'name': 'fred', 'attachment': fp})

Answer 2

您可以尝试使用this等内容模拟您需要的内容。

如何在django中编写文件上传类的单元测试？

2 个答案: