Python CGI FieldStorage测试工具

时间:2011-12-13 01:00:28

标签: python testing cgi

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

在我看来,FieldStorage(或其背后的对象)是完全不可变的,所以我不知道如何在线束中动态提供CGI数据。

2 个答案:

答案 0 :(得分:4)

您可以使用模拟库(例如Mock)来完成这项工作。例如,假设您要从CGI脚本中测试function_to_test函数,可以像这样编写unittest class

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")

如果我将此代码作为单元测试运行,我会得到:

..
----------------------------------------------------------------------
Ran 2 tests in 0.003s

OK

答案 1 :(得分:2)

如果您不想使用额外的库(如Mock):可以使用一些测试数据设置$scope.$applyAsync(function () { //do some stuff }).then(function(){ //do some more stuff }); 对象。下面的Python3示例假定您期望cgi.FieldStorage输入:

POST

这个想法来自https://bugs.python.org/file9507/cgitest.py。在那里你可以找到其他有趣的例子,例如表格上传文件等。

请注意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` 的{​​{1}}方法没有记录,或者至少我在current cgi module documentation中找不到它。