← Back to branch summary

~pythonregexp2.7/python/issue2636-09-01+10

« back to all changes in this revision

Viewing changes to Lib/test/test_cgi.py

  • Committer: Jeffrey C. "The TimeHorse" Jacobs
  • Date: 2008-09-22 21:39:45 UTC
  • mfrom: (39055.1.33 Regexp-2.7)
  • Revision ID: darklord@timehorse.com-20080922213945-23717m5eiqpamcyn
Merged in changes from the Single-Loop Engine branch.

Show diffs side-by-side

added added

removed removed

Lines of Context:
Lib/test/test_cgi.py
55
55
    except StandardError, err:
56
56
        return ComparableException(err)
57
57
 
58
 
# A list of test cases.  Each test case is a a two-tuple that contains
59
 
# a string with the query and a dictionary with the expected result.
60
 
 
61
 
parse_qsl_test_cases = [
62
 
    ("", []),
63
 
    ("&", []),
64
 
    ("&&", []),
65
 
    ("=", [('', '')]),
66
 
    ("=a", [('', 'a')]),
67
 
    ("a", [('a', '')]),
68
 
    ("a=", [('a', '')]),
69
 
    ("a=", [('a', '')]),
70
 
    ("&a=b", [('a', 'b')]),
71
 
    ("a=a+b&b=b+c", [('a', 'a b'), ('b', 'b c')]),
72
 
    ("a=1&a=2", [('a', '1'), ('a', '2')]),
73
 
]
74
 
 
75
58
parse_strict_test_cases = [
76
59
    ("", ValueError("bad query field: ''")),
77
60
    ("&", ValueError("bad query field: ''")),
130
113
def first_second_elts(list):
131
114
    return map(lambda p:(p[0], p[1][0]), list)
132
115
 
 
116
def gen_result(data, environ):
 
117
    fake_stdin = StringIO(data)
 
118
    fake_stdin.seek(0)
 
119
    form = cgi.FieldStorage(fp=fake_stdin, environ=environ)
 
120
 
 
121
    result = {}
 
122
    for k, v in dict(form).items():
 
123
        result[k] = type(v) is list and form.getlist(k) or v.value
 
124
 
 
125
    return result
 
126
 
133
127
class CgiTests(unittest.TestCase):
134
128
 
135
 
    def test_qsl(self):
136
 
        for orig, expect in parse_qsl_test_cases:
137
 
            result = cgi.parse_qsl(orig, keep_blank_values=True)
138
 
            self.assertEqual(result, expect, "Error parsing %s" % repr(orig))
139
 
 
140
129
    def test_strict(self):
141
130
        for orig, expect in parse_strict_test_cases:
142
131
            # Test basic parsing
278
267
                got = getattr(fs.list[x], k)
279
268
                self.assertEquals(got, exp)
280
269
 
 
270
    _qs_result = {
 
271
        'key1': 'value1',
 
272
        'key2': ['value2x', 'value2y'],
 
273
        'key3': 'value3',
 
274
        'key4': 'value4'
 
275
    }
 
276
    def testQSAndUrlEncode(self):
 
277
        data = "key2=value2x&key3=value3&key4=value4"
 
278
        environ = {
 
279
            'CONTENT_LENGTH':   str(len(data)),
 
280
            'CONTENT_TYPE':     'application/x-www-form-urlencoded',
 
281
            'QUERY_STRING':     'key1=value1&key2=value2y',
 
282
            'REQUEST_METHOD':   'POST',
 
283
        }
 
284
        v = gen_result(data, environ)
 
285
        self.assertEqual(self._qs_result, v)
 
286
 
 
287
    def testQSAndFormData(self):
 
288
        data = """
 
289
---123
 
290
Content-Disposition: form-data; name="key2"
 
291
 
 
292
value2y
 
293
---123
 
294
Content-Disposition: form-data; name="key3"
 
295
 
 
296
value3
 
297
---123
 
298
Content-Disposition: form-data; name="key4"
 
299
 
 
300
value4
 
301
---123--
 
302
"""
 
303
        environ = {
 
304
            'CONTENT_LENGTH':   str(len(data)),
 
305
            'CONTENT_TYPE':     'multipart/form-data; boundary=-123',
 
306
            'QUERY_STRING':     'key1=value1&key2=value2x',
 
307
            'REQUEST_METHOD':   'POST',
 
308
        }
 
309
        v = gen_result(data, environ)
 
310
        self.assertEqual(self._qs_result, v)
 
311
 
 
312
    def testQSAndFormDataFile(self):
 
313
        data = """
 
314
---123
 
315
Content-Disposition: form-data; name="key2"
 
316
 
 
317
value2y
 
318
---123
 
319
Content-Disposition: form-data; name="key3"
 
320
 
 
321
value3
 
322
---123
 
323
Content-Disposition: form-data; name="key4"
 
324
 
 
325
value4
 
326
---123
 
327
Content-Disposition: form-data; name="upload"; filename="fake.txt"
 
328
Content-Type: text/plain
 
329
 
 
330
this is the content of the fake file
 
331
 
 
332
---123--
 
333
"""
 
334
        environ = {
 
335
            'CONTENT_LENGTH':   str(len(data)),
 
336
            'CONTENT_TYPE':     'multipart/form-data; boundary=-123',
 
337
            'QUERY_STRING':     'key1=value1&key2=value2x',
 
338
            'REQUEST_METHOD':   'POST',
 
339
        }
 
340
        result = self._qs_result.copy()
 
341
        result.update({
 
342
            'upload': 'this is the content of the fake file\n'
 
343
        })
 
344
        v = gen_result(data, environ)
 
345
        self.assertEqual(result, v)
 
346
 
 
347
    def test_deprecated_parse_qs(self):
 
348
        # this func is moved to urlparse, this is just a sanity check
 
349
        self.assertEqual({'a': ['A1'], 'B': ['B3'], 'b': ['B2']},
 
350
                         cgi.parse_qs('a=A1&b=B2&B=B3'))
 
351
 
 
352
    def test_deprecated_parse_qsl(self):
 
353
        # this func is moved to urlparse, this is just a sanity check
 
354
        self.assertEqual([('a', 'A1'), ('b', 'B2'), ('B', 'B3')],
 
355
                         cgi.parse_qsl('a=A1&b=B2&B=B3'))
 
356
 
 
357
 
281
358
def test_main():
282
359
    run_unittest(CgiTests)
283
360