[test] mod_fastcgi: upload 2MiB and check sha1sum

tests/requests.py

@@ -16,6 +16,7 @@ class CurlRequest(TestBase):
 	SCHEME = "http"
 	PORT = 0 # offset to Env.port
 	AUTH = None
+	POST = None
 
 	EXPECT_RESPONSE_BODY = None
 	EXPECT_RESPONSE_CODE = None
@@ -33,6 +34,10 @@ class CurlRequest(TestBase):
 			self.resp_first_line = header
 			return
 		if header == "":
+			if None != self.resp_first_line and self.resp_first_line.split(" ", 2)[1] == '100':
+				if Env.debugRequests:
+					print >> Env.log, "Handling 100 Continue: '%s'" % self.resp_first_line
+				self.resp_first_line = None
 			return
 		try:
 			(key, value) = header.split(":", 1)
@@ -58,6 +63,7 @@ class CurlRequest(TestBase):
 		b = StringIO.StringIO()
 		c.setopt(pycurl.WRITEFUNCTION, b.write)
 		c.setopt(pycurl.HEADERFUNCTION, self._recv_header)
+		if None != self.POST: c.setopt(pycurl.POSTFIELDS, self.POST)
 
 		if None != self.AUTH:
 			c.setopt(pycurl.USERPWD, self.AUTH)

tests/t-cgi.py

@@ -3,6 +3,15 @@
 from base import *
 from requests import *
 from service import FastCGI
+import hashlib
+
+def generate_body(seed, size):
+	i = 0
+	body = ''
+	while len(body) < size:
+		body += hashlib.sha1(seed + str(i)).digest()
+		i += 1
+	return body[:size]
 
 class CGI(FastCGI):
 	name = "fcgi_cgi"
@@ -20,6 +29,15 @@ printf '%s' "${val}"
 
 """
 
+SCRIPT_UPLOADCHECK="""#!/bin/sh
+
+printf 'Status: 200\\r\\nContent-Type: text/plain\\r\\n\\r\\n'
+
+csum=`sha1sum | cut -d' ' -f1`
+printf '%s' "${csum}"
+
+"""
+
 class TestPathInfo1(CurlRequest):
 	URL = "/envcheck.cgi/abc/xyz?PATH_INFO"
 	EXPECT_RESPONSE_BODY = "/abc/xyz"
@@ -30,10 +48,21 @@ class TestRequestUri1(CurlRequest):
 	EXPECT_RESPONSE_BODY = "/envcheck.cgi/abc/xyz?REQUEST_URI"
 	EXPECT_RESPONSE_CODE = 200
 
+BODY = generate_body('hello world', 2*1024*1024)
+BODY_SHA1 = hashlib.sha1(BODY).hexdigest()
+
+class TestUploadLarge1(CurlRequest):
+	URL = "/uploadcheck.cgi"
+	POST = BODY
+	EXPECT_RESPONSE_BODY = BODY_SHA1
+	EXPECT_RESPONSE_CODE = 200
+
+
 class Test(GroupTest):
 	group = [
 		TestPathInfo1,
 		TestRequestUri1,
+		TestUploadLarge1,
 	]
 
 	config = """
@@ -63,3 +92,4 @@ cgi = {{
 
 	def Prepare(self):
 		self.PrepareVHostFile("envcheck.cgi", SCRIPT_ENVCHECK, mode = 0755)
+		self.PrepareVHostFile("uploadcheck.cgi", SCRIPT_UPLOADCHECK, mode = 0755)