Ticket #783: test_http_post_multipart.2.patch

test_http.py

@@ -10 +10 @@
-
-import httplib
-import cherrypy
+import md5, mimetypes
-
+def encode_multipart_formdata(files):
+    """
+    fields is a sequence of (name, value) elements for regular form fields.
+    files is a sequence of (name, filename, value) elements for data to be uploaded as files
+    Return (content_type, body) ready for httplib.HTTP instance
+    """
+    BOUNDARY = '----------ThIs_Is_tHe_bouNdaRY_$'
+    CRLF = '\r\n'
+    L = []
+    for (key, filename, value) in files:
+        L.append('--' + BOUNDARY)
+        L.append('Content-Disposition: form-data; name="%s"; filename="%s"' % (key, filename))
+        L.append('Content-Type: %s' % get_content_type(filename))
+        L.append('')
+        L.append(value)
+    L.append('--' + BOUNDARY + '--')
+    L.append('')
+    body = CRLF.join(L)
+    content_type = 'multipart/form-data; boundary=%s' % BOUNDARY
+    return content_type, body
+    
+def get_content_type(filename):
+    return mimetypes.guess_type(filename)[0] or 'application/octet-stream'
-
-def setup_server():
-    
@@ -18 +42 @@
-        def index(self, *args, **kwargs):
-            return "Hello world!"
-        index.exposed = True
+        
+        def post_multipart(self, file):
+            # compute and return md5 of posted file
+            contents = file.file.read()
+            return md5.md5(contents).hexdigest()
+        post_multipart.exposed = True
-    
-    cherrypy.tree.mount(Root())
-    cherrypy.config.update({'environment': 'test_suite'})
@@ -35 +65 @@
-        c.request("POST", "/")
-        self.assertEqual(c.getresponse().status, 411)
-
+    def test_post_multipart(self):
+        # generate file contents for a large post
+        contents = "abcdefghijklmnopqrstuvwxyz" * 1000000
+        post_md5 = md5.md5(contents).hexdigest()
-
+        # encode as multipart form data
+        files=[('file', 'file.txt', contents)]
+        content_type, body = encode_multipart_formdata(files)
+
+        # post file
+        if self.scheme == 'https':
+            c = httplib.HTTPS('127.0.0.1:%s' % self.PORT)
+        else:
+            c = httplib.HTTP('127.0.0.1:%s' % self.PORT)
+
+        c.putrequest('POST', '/post_multipart')
+        c.putheader('content-type', content_type)
+        c.putheader('content-length', str(len(body)))
+        c.endheaders()
+        c.send(body)
+        errcode, errmsg, headers = c.getreply()
+
+        response_body = c.file.read()
+        self.assertEquals(post_md5, response_body)
+
-if __name__ == '__main__':
-    setup_server()
-    helper.testmain()