1 from test
.test_support
import verify
, verbose
7 # The regression test will have real values in sys.argv, which
8 # will completely confuse the test of the cgi module
12 cgi
.sys
= HackedSysModule()
15 from cStringIO
import StringIO
17 from StringIO
import StringIO
19 class ComparableException
:
20 def __init__(self
, err
):
26 def __cmp__(self
, anExc
):
27 if not isinstance(anExc
, Exception):
29 x
= cmp(self
.err
.__class
__, anExc
.__class
__)
32 return cmp(self
.err
.args
, anExc
.args
)
34 def __getattr__(self
, attr
):
35 return getattr(self
.err
, attr
)
37 def do_test(buf
, method
):
41 env
['REQUEST_METHOD'] = 'GET'
42 env
['QUERY_STRING'] = buf
43 elif method
== "POST":
45 env
['REQUEST_METHOD'] = 'POST'
46 env
['CONTENT_TYPE'] = 'application/x-www-form-urlencoded'
47 env
['CONTENT_LENGTH'] = str(len(buf
))
49 raise ValueError, "unknown method: %s" % method
51 return cgi
.parse(fp
, env
, strict_parsing
=1)
52 except StandardError, err
:
53 return ComparableException(err
)
55 # A list of test cases. Each test case is a a two-tuple that contains
56 # a string with the query and a dictionary with the expected result.
58 parse_qsl_test_cases
= [
67 ("&a=b", [('a', 'b')]),
68 ("a=a+b&b=b+c", [('a', 'a b'), ('b', 'b c')]),
69 ("a=1&a=2", [('a', '1'), ('a', '2')]),
72 parse_strict_test_cases
= [
73 ("", ValueError("bad query field: ''")),
74 ("&", ValueError("bad query field: ''")),
75 ("&&", ValueError("bad query field: ''")),
76 (";", ValueError("bad query field: ''")),
77 (";&;", ValueError("bad query field: ''")),
78 # Should the next few really be valid?
82 # This rest seem to make sense
84 ("&=a", ValueError("bad query field: ''")),
85 ("=a&", ValueError("bad query field: ''")),
86 ("=&a", ValueError("bad query field: 'a'")),
87 ("b=a", {'b': ['a']}),
88 ("b+=a", {'b ': ['a']}),
89 ("a=b=a", {'a': ['b=a']}),
90 ("a=+b=a", {'a': [' b=a']}),
91 ("&b=a", ValueError("bad query field: ''")),
92 ("b&=a", ValueError("bad query field: 'b'")),
93 ("a=a+b&b=b+c", {'a': ['a b'], 'b': ['b c']}),
94 ("a=a+b&a=b+a", {'a': ['a b', 'b a']}),
95 ("x=1&y=2.0&z=2-3.%2b0", {'x': ['1'], 'y': ['2.0'], 'z': ['2-3.+0']}),
96 ("x=1;y=2.0&z=2-3.%2b0", {'x': ['1'], 'y': ['2.0'], 'z': ['2-3.+0']}),
97 ("x=1;y=2.0;z=2-3.%2b0", {'x': ['1'], 'y': ['2.0'], 'z': ['2-3.+0']}),
98 ("Hbc5161168c542333633315dee1182227:key_store_seqid=400006&cuyer=r&view=bustomer&order_id=0bb2e248638833d48cb7fed300000f1b&expire=964546263&lobale=en-US&kid=130003.300038&ss=env",
99 {'Hbc5161168c542333633315dee1182227:key_store_seqid': ['400006'],
101 'expire': ['964546263'],
102 'kid': ['130003.300038'],
104 'order_id': ['0bb2e248638833d48cb7fed300000f1b'],
106 'view': ['bustomer'],
109 ("group_id=5470&set=custom&_assigned_to=31392&_status=1&_category=100&SUBMIT=Browse",
110 {'SUBMIT': ['Browse'],
111 '_assigned_to': ['31392'],
112 '_category': ['100'],
114 'group_id': ['5470'],
120 if type(list) == type([]):
124 def first_elts(list):
125 return map(lambda x
:x
[0], list)
127 def first_second_elts(list):
128 return map(lambda p
:(p
[0], p
[1][0]), list)
131 for orig
, expect
in parse_qsl_test_cases
:
132 result
= cgi
.parse_qsl(orig
, keep_blank_values
=True)
133 print repr(orig
), '=>', result
134 verify(result
== expect
, "Error parsing %s" % repr(orig
))
136 for orig
, expect
in parse_strict_test_cases
:
139 d
= do_test(orig
, "GET")
140 verify(d
== expect
, "Error parsing %s" % repr(orig
))
141 d
= do_test(orig
, "POST")
142 verify(d
== expect
, "Error parsing %s" % repr(orig
))
144 env
= {'QUERY_STRING': orig
}
145 fcd
= cgi
.FormContentDict(env
)
146 sd
= cgi
.SvFormContentDict(env
)
147 fs
= cgi
.FieldStorage(environ
=env
)
148 if type(expect
) == type({}):
149 # test dict interface
150 verify(len(expect
) == len(fcd
))
151 verify(norm(expect
.keys()) == norm(fcd
.keys()))
152 verify(norm(expect
.values()) == norm(fcd
.values()))
153 verify(norm(expect
.items()) == norm(fcd
.items()))
154 verify(fcd
.get("nonexistent field", "default") == "default")
155 verify(len(sd
) == len(fs
))
156 verify(norm(sd
.keys()) == norm(fs
.keys()))
157 verify(fs
.getvalue("nonexistent field", "default") == "default")
158 # test individual fields
159 for key
in expect
.keys():
160 expect_val
= expect
[key
]
161 verify(fcd
.has_key(key
))
162 verify(norm(fcd
[key
]) == norm(expect
[key
]))
163 verify(fcd
.get(key
, "default") == fcd
[key
])
164 verify(fs
.has_key(key
))
165 if len(expect_val
) > 1:
172 verify(not single_value
)
173 verify(fs
.getvalue(key
) == expect_val
)
176 verify(val
== expect_val
[0])
177 verify(fs
.getvalue(key
) == expect_val
[0])
178 verify(norm(sd
.getlist(key
)) == norm(expect_val
))
180 verify(norm(sd
.values()) == \
181 first_elts(norm(expect
.values())))
182 verify(norm(sd
.items()) == \
183 first_second_elts(norm(expect
.items())))
185 # Test the weird FormContentDict classes
186 env
= {'QUERY_STRING': "x=1&y=2.0&z=2-3.%2b0&1=1abc"}
187 expect
= {'x': 1, 'y': 2.0, 'z': '2-3.+0', '1': '1abc'}
188 d
= cgi
.InterpFormContentDict(env
)
189 for k
, v
in expect
.items():
191 for k
, v
in d
.items():
192 verify(expect
[k
] == v
)
193 verify(norm(expect
.values()) == norm(d
.values()))
197 cgi
.logfp
= sys
.stdout
198 cgi
.initlog("%s", "Testing initlog 1")
199 cgi
.log("%s", "Testing log 2")
200 if os
.path
.exists("/dev/null"):
202 cgi
.logfile
= "/dev/null"
203 cgi
.initlog("%s", "Testing log 3")
204 cgi
.log("Testing log 4")