1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
import unittest
import StringIO
from djangopypi.views import parse_distutils_request
def create_post_data(action):
data = {
":action": action,
"metadata_version": "1.0",
"name": "foo",
"version": "0.1.0-pre2",
"summary": "The quick brown fox jumps over the lazy dog.",
"home_page": "http://example.com",
"author": "Foo Bar Baz",
"author_email": "foobarbaz@example.com",
"license": "Apache",
"keywords": "foo bar baz",
"platform": "UNKNOWN",
"classifiers": [
"Development Status :: 3 - Alpha",
"Environment :: Web Environment",
"Framework :: Django",
"Operating System :: OS Independent",
"Intended Audience :: Developers",
"Intended Audience :: System Administrators",
"License :: OSI Approved :: BSD License",
"Topic :: System :: Software Distribution",
"Programming Language :: Python",
],
"download_url": "",
"provides": "",
"requires": "",
"obsoletes": "",
"description": """
=========
FOOBARBAZ
=========
Introduction
------------
``foo`` :class:`bar`
*baz*
[foaoa]
""",
}
return data
def create_request(data):
boundary = '--------------GHSKFJDLGDS7543FJKLFHRE75642756743254'
sep_boundary = '\n--' + boundary
end_boundary = sep_boundary + '--'
body = StringIO.StringIO()
for key, value in data.items():
# handle multiple entries for the same name
if type(value) not in (type([]), type( () )):
value = [value]
for value in value:
value = unicode(value).encode("utf-8")
body.write(sep_boundary)
body.write('\nContent-Disposition: form-data; name="%s"'%key)
body.write("\n\n")
body.write(value)
if value and value[-1] == '\r':
body.write('\n') # write an extra newline (lurve Macs)
body.write(end_boundary)
body.write("\n")
return body.getvalue()
class MockRequest(object):
def __init__(self, raw_post_data):
self.raw_post_data = raw_post_data
self.META = {}
class TestParseWeirdPostData(unittest.TestCase):
def test_weird_post_data(self):
data = create_post_data("submit")
raw_post_data = create_request(data)
post, files = parse_distutils_request(MockRequest(raw_post_data))
self.assertTrue(post)
for key in post.keys():
if isinstance(data[key], list):
self.assertEquals(data[key], post.getlist(key))
elif data[key] == "UNKNOWN":
self.assertTrue(post[key] is None)
else:
self.assertEquals(post[key], data[key])
|
