[tests] test secdownload.lua and the lua key-value list lookup

tests/t-secdownload.py

@@ -0,0 +1,49 @@
+# -*- coding: utf-8 -*-
+
+from base import *
+from requests import *
+import time
+from md5 import md5
+
+TEST_TXT="""Hi!
+0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef
+0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef
+0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef
+0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef
+"""
+
+def securl(prefix, path, secret, tstamp = None):
+	if tstamp == None: tstamp = time.time()
+	tstamp = '%x' % int(tstamp)
+	md5content = secret + path + tstamp
+	if prefix[-1] != '/': prefix += '/'
+	return prefix + md5(md5content).hexdigest() + '/' + tstamp + path
+
+class SecdownloadFail(CurlRequest):
+	URL = "/test.txt"
+	EXPECT_RESPONSE_CODE = 403
+
+class SecdownloadSuccess(CurlRequest):
+	EXPECT_RESPONSE_BODY = TEST_TXT
+	EXPECT_RESPONSE_CODE = 200
+
+	def Run(self):
+		self.URL = securl('/', '/test.txt', 'abc')
+		return super(SecdownloadSuccess, self).Run()
+
+class SecdownloadGone(CurlRequest):
+	EXPECT_RESPONSE_CODE = 410
+
+	def Run(self):
+		self.URL = securl('/', '/test.txt', 'abc', time.time() - 800)
+		return super(SecdownloadGone, self).Run()
+
+class Test(GroupTest):
+	group = [SecdownloadFail, SecdownloadSuccess, SecdownloadGone]
+
+	def Prepare(self):
+		self.PrepareVHostFile("test.txt", TEST_TXT)
+		self.config = """
+			secdownload ( "prefix" => "/", "document-root" => "{docroot}", "secret" => "abc", "timeout" => 600 );
+		""".format(docroot = self.vhostdir)
+		self.vhostdir = None