diff options
Diffstat (limited to 'djangopypi/tests.py')
-rw-r--r-- | djangopypi/tests.py | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/djangopypi/tests.py b/djangopypi/tests.py index c6ebce2..2d8c305 100644 --- a/djangopypi/tests.py +++ b/djangopypi/tests.py | |||
@@ -1,6 +1,6 @@ | |||
1 | import unittest | 1 | import unittest |
2 | import StringIO | 2 | import StringIO |
3 | from djangopypi.views import parse_weird_post_data | 3 | from djangopypi.views import parse_distutils_request |
4 | 4 | ||
5 | def create_post_data(action): | 5 | def create_post_data(action): |
6 | data = { | 6 | data = { |
@@ -67,12 +67,20 @@ def create_request(data): | |||
67 | return body.getvalue() | 67 | return body.getvalue() |
68 | 68 | ||
69 | 69 | ||
70 | class MockRequest(object): | ||
71 | |||
72 | def __init__(self, raw_post_data): | ||
73 | self.raw_post_data = raw_post_data | ||
74 | self.META = {} | ||
75 | |||
76 | |||
70 | class TestParseWeirdPostData(unittest.TestCase): | 77 | class TestParseWeirdPostData(unittest.TestCase): |
71 | 78 | ||
72 | def test_weird_post_data(self): | 79 | def test_weird_post_data(self): |
73 | data = create_post_data("submit") | 80 | data = create_post_data("submit") |
74 | raw_post_data = create_request(data) | 81 | raw_post_data = create_request(data) |
75 | post, files = parse_weird_post_data(raw_post_data) | 82 | post, files = parse_distutils_request(MockRequest(raw_post_data)) |
83 | print("post: %s files: %s" % (post, files)) | ||
76 | self.assertTrue(post) | 84 | self.assertTrue(post) |
77 | 85 | ||
78 | for key in post.keys(): | 86 | for key in post.keys(): |