From bf986393b992365183417fac213d0c3151726b32 Mon Sep 17 00:00:00 2001
From: Sven Slootweg <jamsoftgamedev@gmail.com>
Date: Fri, 26 Oct 2012 19:06:00 +0200
Subject: [PATCH] Add testing script

---
 test.py | 76 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 76 insertions(+)
 create mode 100644 test.py

diff --git a/test.py b/test.py
new file mode 100644
index 0000000..18d5ca9
--- /dev/null
+++ b/test.py
@@ -0,0 +1,76 @@
+import resolv, urllib, urllib2, argparse
+
+# TODO:
+# http://www.mediafire.com/view/?vxltbkr2l9ycmah => http://www.mediafire.com/?vxltbkr2l9ycmah
+
+suites = {
+	'1channel': {
+		"1channel + PutLocker (video)": "http://www.1channel.ch/external.php?title=Big+Buck+Bunny&url=aHR0cDovL3d3dy5wdXRsb2NrZXIuY29tL2ZpbGUvOTg3RkVCRjVEQjY0NUUyRQ==&domain=cHV0bG9ja2VyLmNvbQ==&loggedin=0"
+	},
+	'putlocker': {
+		"PutLocker (video)": "http://www.putlocker.com/file/987FEBF5DB645E2E",
+		"SockShare (video)": "http://www.sockshare.com/file/88DF2133C85521BD"
+	},
+	'filebox': {
+		"Filebox (video)": "http://www.filebox.com/p0rp8nabrcfk"
+	},
+	'pastebin': {
+		"Pastebin": "http://pastebin.com/imyEc26g"
+	},
+	'mediafire': {
+		"MediaFire": "http://www.mediafire.com/?vxltbkr2l9ycmah",
+		"MediaFire with password (mfddl)": "http://www.mediafire.com/?traa1p0lki9611h"
+	},
+	'youtube': {
+		"YouTube": "http://www.youtube.com/watch?v=XSGBVzeBUbk"
+	}
+}
+
+parser = argparse.ArgumentParser(description='Testing script for the resolv library.')
+
+parser.add_argument('suites', metavar='SUITE', type=str, nargs='*',
+                   help='suites to test (leave empty to test all suites)')
+
+args = parser.parse_args()
+options = vars(args)
+
+to_test = {}
+
+if len(options['suites']) == 0:
+	for suite in suites:
+		for description, url in suite.iteritems():
+			to_test[description] = url
+else:
+	for suite in options['suites']:
+		for description, url in suites[suite].iteritems():
+			to_test[description] = url
+
+def process_result(res):
+	if res.state == "finished":
+		print "Successful!\nType: %s\nResults: %s\nCookie jar: %s" % (res.result_type, str(res.results), str(res.cookiejar))
+	elif res.state == "failed":
+		print "Failed."
+	elif res.state == "invalid":
+		print "Invalid URL."
+	elif res.state == "need_password":
+		pw = raw_input("Password required. Enter password: ")
+		res.verify_password(pw)
+		process_result(res)
+	elif res.state == "password_invalid":
+		pw = raw_input("Password invalid! Try again: ")
+		res.verify_password(pw)
+		process_result(res)
+	else:
+		print "Unknown result state: %s" % res.state
+
+for title, url in to_test.iteritems():
+	print "============ %s ============" % title
+	print "RESOLVE:"
+	res = resolv.resolve(url)
+	process_result(res)
+	print ""
+	print "RECURSE:"
+	res = resolv.recurse(url)
+	process_result(res)
+	print ""
+