如何創建用於測試目的的 cgi.FieldStorage？

Question

我正在創建一個實用程序來處理基於 webob 的應用程序中的文件上傳。 我想為它寫一些單元測試。

我的問題是 - 由於 webob 使用cgi.FieldStorage上傳文件，我想以一種簡單的方式創建一個FieldStorage實例（不模擬整個請求）。 這是我需要做的最少代碼（沒什么特別的，模擬上傳帶有“Lorem ipsum”內容的文本文件就可以了）。 還是嘲笑它更好？

Answer 1

經過一些研究，我想出了類似的東西：

def _create_fs(mimetype, content):                                              
    fs = cgi.FieldStorage()                                                     
    fs.file = fs.make_file()                                                    
    fs.type = mimetype                                                          
    fs.file.write(content)                                                      
    fs.file.seek(0)                                                             
    return fs             

這足以進行我的單元測試。

Answer 2

你的答案在python3中失敗了。 這是我的修改。 我敢肯定它並不完美，但至少它適用於python2.7和python3.5。

from io import BytesIO

def _create_fs(self, mimetype, content, filename='uploaded.txt', name="file"):
    content = content.encode('utf-8')
    headers = {u'content-disposition': u'form-data; name="{}"; filename="{}"'.format(name, filename),
               u'content-length': len(content),
               u'content-type': mimetype}
    environ = {'REQUEST_METHOD': 'POST'}
    fp = BytesIO(content)
    return cgi.FieldStorage(fp=fp, headers=headers, environ=environ)

Answer 3

我正在做這樣的事情，因為在我的腳本中我正在使用

import cgi

form = cgi.FieldStorage()

>>> # which result:
>>> FieldStorage(None, None, [])

當您的 URL 包含查詢字符串時，它將如下所示：

# URL: https://..../index.cgi?test=only
FieldStorage(None, None, [MiniFieldStorage('test', 'only')])

所以我只是手動將MiniFieldStorage推入form變量

import cgi

fs = cgi.MiniFieldStorage('test', 'only')
>>> fs
MiniFieldStorage('test', 'only')

form = cgi.FieldStorage()
form.list.append(fs)

>>> form
>>> FieldStorage(None, None, [MiniFieldStorage('test', 'only')])

# Now you can call it with same functions
>>> form.getfirst('test')
'only'