Python Django框架单元测试之文件上传测试示例

时间:2021-10-30 18:59:47

本文实例讲述了python django框架单元测试之文件上传测试。分享给大家供大家参考,具体如下:

submitting files is a special case. to post a file, you need only provide the file field name as a key, and a file handle to the file you wish to upload as a value. for example:

?
1
2
3
>>> c = client()
>>> with open('test.jpg') as fp:
...   c.post('/account/avatar_upload/',{'avatar':fp})

测试文件上传其实没有什么特殊的,只需要指定后端接受请求数据的对应键值即可

(the name avatar here is not relevant; use whatever name your file-processing code expects.)在这里avatar是关联的,对应着具体的后端处理程序代码,eg:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
class useravatar(view):
  def __init__(self):
    self.thumbnail_dir = os.path.join(static_root, 'avatar/thumbnails')
    self.dest_dir = os.path.join(static_root, 'avatar/origin_imgs')
  @method_decorator(login_required)
  def post(self, request):
    nt_id = request.session.get('user_id', 'default')
    user = user.objects.get(pk=nt_id) if user.objects.filter(pk=nt_id).exists() else none
    avatarimg = request.files['avatar']
    if not os.path.exists(self.dest_dir):
      os.mkdir(self.dest_dir)
    dest = os.path.join(self.dest_dir, nt_id+"_avatar.jpg")
    with open(dest, "wb+") as destination:
      for chunk in avatarimg.chunks():
        destination.write(chunk)
    if make_thumb(dest,self.thumbnail_dir):
      avartapath = os.path.join(static_url, 'avatar/thumbnails', nt_id + "_avatar.jpg")
    else:
      avartapath = os.path.join(static_url, 'avatar/origin_imgs', nt_id + "_avatar.jpg")
    user.objects.filter(nt_id=nt_id).update(avatar=avartapath)
    return render(request, 'profile.html', {'user': user})

Python Django框架单元测试之文件上传测试示例

希望本文所述对大家基于django框架的python程序设计有所帮助。

原文链接:https://blog.csdn.net/Lockey23/article/details/80653091