+++ /dev/null
-from test.test_support import run_unittest, check_warnings\r
-import cgi\r
-import os\r
-import sys\r
-import tempfile\r
-import unittest\r
-\r
-class HackedSysModule:\r
- # The regression test will have real values in sys.argv, which\r
- # will completely confuse the test of the cgi module\r
- argv = []\r
- stdin = sys.stdin\r
-\r
-cgi.sys = HackedSysModule()\r
-\r
-try:\r
- from cStringIO import StringIO\r
-except ImportError:\r
- from StringIO import StringIO\r
-\r
-class ComparableException:\r
- def __init__(self, err):\r
- self.err = err\r
-\r
- def __str__(self):\r
- return str(self.err)\r
-\r
- def __cmp__(self, anExc):\r
- if not isinstance(anExc, Exception):\r
- return -1\r
- x = cmp(self.err.__class__, anExc.__class__)\r
- if x != 0:\r
- return x\r
- return cmp(self.err.args, anExc.args)\r
-\r
- def __getattr__(self, attr):\r
- return getattr(self.err, attr)\r
-\r
-def do_test(buf, method):\r
- env = {}\r
- if method == "GET":\r
- fp = None\r
- env['REQUEST_METHOD'] = 'GET'\r
- env['QUERY_STRING'] = buf\r
- elif method == "POST":\r
- fp = StringIO(buf)\r
- env['REQUEST_METHOD'] = 'POST'\r
- env['CONTENT_TYPE'] = 'application/x-www-form-urlencoded'\r
- env['CONTENT_LENGTH'] = str(len(buf))\r
- else:\r
- raise ValueError, "unknown method: %s" % method\r
- try:\r
- return cgi.parse(fp, env, strict_parsing=1)\r
- except StandardError, err:\r
- return ComparableException(err)\r
-\r
-parse_strict_test_cases = [\r
- ("", ValueError("bad query field: ''")),\r
- ("&", ValueError("bad query field: ''")),\r
- ("&&", ValueError("bad query field: ''")),\r
- (";", ValueError("bad query field: ''")),\r
- (";&;", ValueError("bad query field: ''")),\r
- # Should the next few really be valid?\r
- ("=", {}),\r
- ("=&=", {}),\r
- ("=;=", {}),\r
- # This rest seem to make sense\r
- ("=a", {'': ['a']}),\r
- ("&=a", ValueError("bad query field: ''")),\r
- ("=a&", ValueError("bad query field: ''")),\r
- ("=&a", ValueError("bad query field: 'a'")),\r
- ("b=a", {'b': ['a']}),\r
- ("b+=a", {'b ': ['a']}),\r
- ("a=b=a", {'a': ['b=a']}),\r
- ("a=+b=a", {'a': [' b=a']}),\r
- ("&b=a", ValueError("bad query field: ''")),\r
- ("b&=a", ValueError("bad query field: 'b'")),\r
- ("a=a+b&b=b+c", {'a': ['a b'], 'b': ['b c']}),\r
- ("a=a+b&a=b+a", {'a': ['a b', 'b a']}),\r
- ("x=1&y=2.0&z=2-3.%2b0", {'x': ['1'], 'y': ['2.0'], 'z': ['2-3.+0']}),\r
- ("x=1;y=2.0&z=2-3.%2b0", {'x': ['1'], 'y': ['2.0'], 'z': ['2-3.+0']}),\r
- ("x=1;y=2.0;z=2-3.%2b0", {'x': ['1'], 'y': ['2.0'], 'z': ['2-3.+0']}),\r
- ("Hbc5161168c542333633315dee1182227:key_store_seqid=400006&cuyer=r&view=bustomer&order_id=0bb2e248638833d48cb7fed300000f1b&expire=964546263&lobale=en-US&kid=130003.300038&ss=env",\r
- {'Hbc5161168c542333633315dee1182227:key_store_seqid': ['400006'],\r
- 'cuyer': ['r'],\r
- 'expire': ['964546263'],\r
- 'kid': ['130003.300038'],\r
- 'lobale': ['en-US'],\r
- 'order_id': ['0bb2e248638833d48cb7fed300000f1b'],\r
- 'ss': ['env'],\r
- 'view': ['bustomer'],\r
- }),\r
-\r
- ("group_id=5470&set=custom&_assigned_to=31392&_status=1&_category=100&SUBMIT=Browse",\r
- {'SUBMIT': ['Browse'],\r
- '_assigned_to': ['31392'],\r
- '_category': ['100'],\r
- '_status': ['1'],\r
- 'group_id': ['5470'],\r
- 'set': ['custom'],\r
- })\r
- ]\r
-\r
-def first_elts(list):\r
- return map(lambda x:x[0], list)\r
-\r
-def first_second_elts(list):\r
- return map(lambda p:(p[0], p[1][0]), list)\r
-\r
-def gen_result(data, environ):\r
- fake_stdin = StringIO(data)\r
- fake_stdin.seek(0)\r
- form = cgi.FieldStorage(fp=fake_stdin, environ=environ)\r
-\r
- result = {}\r
- for k, v in dict(form).items():\r
- result[k] = isinstance(v, list) and form.getlist(k) or v.value\r
-\r
- return result\r
-\r
-class CgiTests(unittest.TestCase):\r
-\r
- def test_strict(self):\r
- for orig, expect in parse_strict_test_cases:\r
- # Test basic parsing\r
- d = do_test(orig, "GET")\r
- self.assertEqual(d, expect, "Error parsing %s" % repr(orig))\r
- d = do_test(orig, "POST")\r
- self.assertEqual(d, expect, "Error parsing %s" % repr(orig))\r
-\r
- env = {'QUERY_STRING': orig}\r
- fcd = cgi.FormContentDict(env)\r
- sd = cgi.SvFormContentDict(env)\r
- fs = cgi.FieldStorage(environ=env)\r
- if isinstance(expect, dict):\r
- # test dict interface\r
- self.assertEqual(len(expect), len(fcd))\r
- self.assertItemsEqual(expect.keys(), fcd.keys())\r
- self.assertItemsEqual(expect.values(), fcd.values())\r
- self.assertItemsEqual(expect.items(), fcd.items())\r
- self.assertEqual(fcd.get("nonexistent field", "default"), "default")\r
- self.assertEqual(len(sd), len(fs))\r
- self.assertItemsEqual(sd.keys(), fs.keys())\r
- self.assertEqual(fs.getvalue("nonexistent field", "default"), "default")\r
- # test individual fields\r
- for key in expect.keys():\r
- expect_val = expect[key]\r
- self.assertTrue(fcd.has_key(key))\r
- self.assertItemsEqual(fcd[key], expect[key])\r
- self.assertEqual(fcd.get(key, "default"), fcd[key])\r
- self.assertTrue(fs.has_key(key))\r
- if len(expect_val) > 1:\r
- single_value = 0\r
- else:\r
- single_value = 1\r
- try:\r
- val = sd[key]\r
- except IndexError:\r
- self.assertFalse(single_value)\r
- self.assertEqual(fs.getvalue(key), expect_val)\r
- else:\r
- self.assertTrue(single_value)\r
- self.assertEqual(val, expect_val[0])\r
- self.assertEqual(fs.getvalue(key), expect_val[0])\r
- self.assertItemsEqual(sd.getlist(key), expect_val)\r
- if single_value:\r
- self.assertItemsEqual(sd.values(),\r
- first_elts(expect.values()))\r
- self.assertItemsEqual(sd.items(),\r
- first_second_elts(expect.items()))\r
-\r
- def test_weird_formcontentdict(self):\r
- # Test the weird FormContentDict classes\r
- env = {'QUERY_STRING': "x=1&y=2.0&z=2-3.%2b0&1=1abc"}\r
- expect = {'x': 1, 'y': 2.0, 'z': '2-3.+0', '1': '1abc'}\r
- d = cgi.InterpFormContentDict(env)\r
- for k, v in expect.items():\r
- self.assertEqual(d[k], v)\r
- for k, v in d.items():\r
- self.assertEqual(expect[k], v)\r
- self.assertItemsEqual(expect.values(), d.values())\r
-\r
- def test_log(self):\r
- cgi.log("Testing")\r
-\r
- cgi.logfp = StringIO()\r
- cgi.initlog("%s", "Testing initlog 1")\r
- cgi.log("%s", "Testing log 2")\r
- self.assertEqual(cgi.logfp.getvalue(), "Testing initlog 1\nTesting log 2\n")\r
- if os.path.exists("/dev/null"):\r
- cgi.logfp = None\r
- cgi.logfile = "/dev/null"\r
- cgi.initlog("%s", "Testing log 3")\r
- cgi.log("Testing log 4")\r
-\r
- def test_fieldstorage_readline(self):\r
- # FieldStorage uses readline, which has the capacity to read all\r
- # contents of the input file into memory; we use readline's size argument\r
- # to prevent that for files that do not contain any newlines in\r
- # non-GET/HEAD requests\r
- class TestReadlineFile:\r
- def __init__(self, file):\r
- self.file = file\r
- self.numcalls = 0\r
-\r
- def readline(self, size=None):\r
- self.numcalls += 1\r
- if size:\r
- return self.file.readline(size)\r
- else:\r
- return self.file.readline()\r
-\r
- def __getattr__(self, name):\r
- file = self.__dict__['file']\r
- a = getattr(file, name)\r
- if not isinstance(a, int):\r
- setattr(self, name, a)\r
- return a\r
-\r
- f = TestReadlineFile(tempfile.TemporaryFile())\r
- f.write('x' * 256 * 1024)\r
- f.seek(0)\r
- env = {'REQUEST_METHOD':'PUT'}\r
- fs = cgi.FieldStorage(fp=f, environ=env)\r
- # if we're not chunking properly, readline is only called twice\r
- # (by read_binary); if we are chunking properly, it will be called 5 times\r
- # as long as the chunksize is 1 << 16.\r
- self.assertTrue(f.numcalls > 2)\r
-\r
- def test_fieldstorage_multipart(self):\r
- #Test basic FieldStorage multipart parsing\r
- env = {'REQUEST_METHOD':'POST', 'CONTENT_TYPE':'multipart/form-data; boundary=---------------------------721837373350705526688164684', 'CONTENT_LENGTH':'558'}\r
- postdata = """-----------------------------721837373350705526688164684\r
-Content-Disposition: form-data; name="id"\r
-\r
-1234\r
------------------------------721837373350705526688164684\r
-Content-Disposition: form-data; name="title"\r
-\r
-\r
------------------------------721837373350705526688164684\r
-Content-Disposition: form-data; name="file"; filename="test.txt"\r
-Content-Type: text/plain\r
-\r
-Testing 123.\r
-\r
------------------------------721837373350705526688164684\r
-Content-Disposition: form-data; name="submit"\r
-\r
- Add\x20\r
------------------------------721837373350705526688164684--\r
-"""\r
- fs = cgi.FieldStorage(fp=StringIO(postdata), environ=env)\r
- self.assertEqual(len(fs.list), 4)\r
- expect = [{'name':'id', 'filename':None, 'value':'1234'},\r
- {'name':'title', 'filename':None, 'value':''},\r
- {'name':'file', 'filename':'test.txt','value':'Testing 123.\n'},\r
- {'name':'submit', 'filename':None, 'value':' Add '}]\r
- for x in range(len(fs.list)):\r
- for k, exp in expect[x].items():\r
- got = getattr(fs.list[x], k)\r
- self.assertEqual(got, exp)\r
-\r
- _qs_result = {\r
- 'key1': 'value1',\r
- 'key2': ['value2x', 'value2y'],\r
- 'key3': 'value3',\r
- 'key4': 'value4'\r
- }\r
- def testQSAndUrlEncode(self):\r
- data = "key2=value2x&key3=value3&key4=value4"\r
- environ = {\r
- 'CONTENT_LENGTH': str(len(data)),\r
- 'CONTENT_TYPE': 'application/x-www-form-urlencoded',\r
- 'QUERY_STRING': 'key1=value1&key2=value2y',\r
- 'REQUEST_METHOD': 'POST',\r
- }\r
- v = gen_result(data, environ)\r
- self.assertEqual(self._qs_result, v)\r
-\r
- def testQSAndFormData(self):\r
- data = """\r
----123\r
-Content-Disposition: form-data; name="key2"\r
-\r
-value2y\r
----123\r
-Content-Disposition: form-data; name="key3"\r
-\r
-value3\r
----123\r
-Content-Disposition: form-data; name="key4"\r
-\r
-value4\r
----123--\r
-"""\r
- environ = {\r
- 'CONTENT_LENGTH': str(len(data)),\r
- 'CONTENT_TYPE': 'multipart/form-data; boundary=-123',\r
- 'QUERY_STRING': 'key1=value1&key2=value2x',\r
- 'REQUEST_METHOD': 'POST',\r
- }\r
- v = gen_result(data, environ)\r
- self.assertEqual(self._qs_result, v)\r
-\r
- def testQSAndFormDataFile(self):\r
- data = """\r
----123\r
-Content-Disposition: form-data; name="key2"\r
-\r
-value2y\r
----123\r
-Content-Disposition: form-data; name="key3"\r
-\r
-value3\r
----123\r
-Content-Disposition: form-data; name="key4"\r
-\r
-value4\r
----123\r
-Content-Disposition: form-data; name="upload"; filename="fake.txt"\r
-Content-Type: text/plain\r
-\r
-this is the content of the fake file\r
-\r
----123--\r
-"""\r
- environ = {\r
- 'CONTENT_LENGTH': str(len(data)),\r
- 'CONTENT_TYPE': 'multipart/form-data; boundary=-123',\r
- 'QUERY_STRING': 'key1=value1&key2=value2x',\r
- 'REQUEST_METHOD': 'POST',\r
- }\r
- result = self._qs_result.copy()\r
- result.update({\r
- 'upload': 'this is the content of the fake file\n'\r
- })\r
- v = gen_result(data, environ)\r
- self.assertEqual(result, v)\r
-\r
- def test_deprecated_parse_qs(self):\r
- # this func is moved to urlparse, this is just a sanity check\r
- with check_warnings(('cgi.parse_qs is deprecated, use urlparse.'\r
- 'parse_qs instead', PendingDeprecationWarning)):\r
- self.assertEqual({'a': ['A1'], 'B': ['B3'], 'b': ['B2']},\r
- cgi.parse_qs('a=A1&b=B2&B=B3'))\r
-\r
- def test_deprecated_parse_qsl(self):\r
- # this func is moved to urlparse, this is just a sanity check\r
- with check_warnings(('cgi.parse_qsl is deprecated, use urlparse.'\r
- 'parse_qsl instead', PendingDeprecationWarning)):\r
- self.assertEqual([('a', 'A1'), ('b', 'B2'), ('B', 'B3')],\r
- cgi.parse_qsl('a=A1&b=B2&B=B3'))\r
-\r
- def test_parse_header(self):\r
- self.assertEqual(\r
- cgi.parse_header("text/plain"),\r
- ("text/plain", {}))\r
- self.assertEqual(\r
- cgi.parse_header("text/vnd.just.made.this.up ; "),\r
- ("text/vnd.just.made.this.up", {}))\r
- self.assertEqual(\r
- cgi.parse_header("text/plain;charset=us-ascii"),\r
- ("text/plain", {"charset": "us-ascii"}))\r
- self.assertEqual(\r
- cgi.parse_header('text/plain ; charset="us-ascii"'),\r
- ("text/plain", {"charset": "us-ascii"}))\r
- self.assertEqual(\r
- cgi.parse_header('text/plain ; charset="us-ascii"; another=opt'),\r
- ("text/plain", {"charset": "us-ascii", "another": "opt"}))\r
- self.assertEqual(\r
- cgi.parse_header('attachment; filename="silly.txt"'),\r
- ("attachment", {"filename": "silly.txt"}))\r
- self.assertEqual(\r
- cgi.parse_header('attachment; filename="strange;name"'),\r
- ("attachment", {"filename": "strange;name"}))\r
- self.assertEqual(\r
- cgi.parse_header('attachment; filename="strange;name";size=123;'),\r
- ("attachment", {"filename": "strange;name", "size": "123"}))\r
-\r
-\r
-def test_main():\r
- run_unittest(CgiTests)\r
-\r
-if __name__ == '__main__':\r
- test_main()\r