Python CGI现场存储测试工具

如何为Python CGI脚本设置一个小型测试工具？我不想运行服务器来测试它，但我确实想为我的测试提供各种GET/POST输入

---

在我看来，FieldStorage（或其背后的对象）是完全不可变的，因此我不知道如何在一个线束中动态提供CGI数据。

您可以使用模拟库，例如来完成这项工作。例如，假设您想从CGI脚本中测试

函数\u to_test

函数，您可以编写如下代码：

import unittest
import cgi

from mock import patch

def function_to_test():
    form = cgi.FieldStorage()
    if "name" not in form or "addr" not in form:
        return "<H1>Error</H1>\nPlease fill in the name and address.\n"
    text = "<p>name: {0}\n<p>addr: {1}\n"
    return text.format(form["name"].value, form["addr"].value)

@patch('cgi.FieldStorage')
class TestClass(unittest.TestCase):
    class TestField(object):
        def __init__(self, value):
            self.value = value

    FIELDS = { "name" : TestField("Bill"), "addr" : TestField("1 Two Street") }

    def test_cgi(self, MockClass):
        instance = MockClass.return_value
        instance.__getitem__ = lambda s, key: TestClass.FIELDS[key]
        instance.__contains__ = lambda s, key: key in TestClass.FIELDS
        text = function_to_test()
        self.assertEqual(text, "<p>name: Bill\n<p>addr: 1 Two Street\n")

    def test_err(self, MockClass):
        instance = MockClass.return_value
        instance.__contains__ = lambda self, key: False
        text = function_to_test()
        self.assertEqual(text,
            "<H1>Error</H1>\nPlease fill in the name and address.\n")

---

如果您不想使用诸如Mock之类的额外库：可以使用一些测试数据设置

cgi.FieldStorage

对象。下面的Python3示例假设您需要

POST

输入：

import unittest
from io import BytesIO

class TestForm(unittest.TestCase):

    def setUp(self):
        """
        Makes a cgi.FieldStorage object
        with some bogus fields.
        """
        # provide a byte string with the parameters
        # using the format b"name1=value1&name2=value2..."
        urlencode_data = b"firstname=Joe&lastname=Bloggs&email=joe.bloggs@company.com"
        urlencode_environ = {
            'CONTENT_LENGTH':   str(len(urlencode_data)),
            'CONTENT_TYPE':     'application/x-www-form-urlencoded',
            'QUERY_STRING':     '',
            'REQUEST_METHOD':   'POST',
        }
        data = BytesIO(urlencode_data)
        data.seek(0)
        self.fs = cgi.FieldStorage(fp=data, environ=urlencode_environ)

    # unit test methods come here
    # form fields are accessible via `self.fs`

这个想法来自于。在那里你可以找到其他有趣的例子，例如带有文件上传的表单等

请注意，

cgi.FieldStorage

的

\uuuuu init\uuuu

方法没有文档记录，或者至少我在中找不到它

import unittest
from io import BytesIO

class TestForm(unittest.TestCase):

    def setUp(self):
        """
        Makes a cgi.FieldStorage object
        with some bogus fields.
        """
        # provide a byte string with the parameters
        # using the format b"name1=value1&name2=value2..."
        urlencode_data = b"firstname=Joe&lastname=Bloggs&email=joe.bloggs@company.com"
        urlencode_environ = {
            'CONTENT_LENGTH':   str(len(urlencode_data)),
            'CONTENT_TYPE':     'application/x-www-form-urlencoded',
            'QUERY_STRING':     '',
            'REQUEST_METHOD':   'POST',
        }
        data = BytesIO(urlencode_data)
        data.seek(0)
        self.fs = cgi.FieldStorage(fp=data, environ=urlencode_environ)

    # unit test methods come here
    # form fields are accessible via `self.fs`