flask unittest上传图片或者文件

Posted 2020-10-14 菲菲菲菲菲常新的新手

在flask自动化测试中如何上传图片或者文件

        import StringIO
        picture = StringIO.StringIO(open(‘/home/admin/Pictures/test.jpg‘, ‘rb‘).read())
        data = dict(
            {‘picture‘: (picture, ‘test.jpg‘)},
            timestamp=self.timestamp,
            check = check,
        )
        base_url = ‘/uploadPhoto‘
        rv = self.app.post(base_url, data=data, headers=HEADERS, )
        print rv.data

上传文件也是一样额，把picture改为文件内容即可