From c7ad1cc5b4a5d190a060e3ddd9274c3065da6708 Mon Sep 17 00:00:00 2001
From: RaNaN <Mast3rRaNaN@hotmail.de>
Date: Fri, 13 Jan 2012 23:24:21 +0100
Subject: plugin unit test, closed #499, #500
---
tests/HosterPluginTester.py | 115 ++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 115 insertions(+)
(limited to 'tests/HosterPluginTester.py')
diff --git a/tests/HosterPluginTester.py b/tests/HosterPluginTester.py
index faaaf799c..e4738ad5e 100644
--- a/tests/HosterPluginTester.py
+++ b/tests/HosterPluginTester.py
@@ -1,3 +1,118 @@
# -*- coding: utf-8 -*-
+from os import remove
+from os.path import dirname
+from logging import log, DEBUG
+from hashlib import md5
+from time import time
+from nose.tools import nottest
+
+from helper.Stubs import Core
+from helper.PluginTester import PluginTester
+
+from module.PyFile import PyFile
+from module.plugins.Base import Fail
+from module.utils import accumulate
+from module.utils.fs import save_join, join, exists
+
+DL_DIR = join("Downloads", "tmp")
+
+class HosterPluginTester(PluginTester):
+
+ files = {}
+
+ def setUp(self):
+ PluginTester.setUp(self)
+ for f in self.files:
+ pass
+ if exists(join(DL_DIR, f)): remove(join(DL_DIR, f))
+
+ @nottest
+ def test_plugin(self, name, url, flag):
+
+ # Print to stdout to see whats going on
+ print "%s: %s" % (name, url)
+ log(DEBUG, "%s: %s", name, url)
+
+ # url and plugin should be only important thing
+ pyfile = PyFile(self.core, -1, url, url, 0, 0, "", name, 0, 0)
+ pyfile.initPlugin()
+
+ self.thread.pyfile = pyfile
+ self.thread.plugin = pyfile.plugin
+
+ try:
+ a = time()
+ pyfile.plugin.preprocessing(self.thread)
+
+
+ log(DEBUG, "downloading took %ds" % (time()-a))
+ log(DEBUG, "size %d kb" % (pyfile.size / 1024))
+
+ if pyfile.name not in self.files:
+ raise Exception("Filename %s wrong." % pyfile.name)
+
+ if not exists(save_join(DL_DIR, pyfile.name)):
+ raise Exception("File %s does not exists." % pyfile.name)
+
+ hash = md5()
+ f = open(save_join(DL_DIR, pyfile.name))
+ while True:
+ buf = f.read(4096)
+ if not buf: break
+ hash.update(buf)
+
+ if hash.hexdigest() != self.files[pyfile.name]:
+ raise Exception("Hash does not match.")
+
+
+
+ except Exception, e:
+ if isinstance(e, Fail) and flag == "fail":
+ pass
+ elif isinstance(e, Fail) and flag == "offline" and e.message == "offline":
+ pass
+ else:
+ raise
+
+
+# setup methods
+
+c = Core()
+
+f = open(join(dirname(__file__), "hosterlinks.txt"))
+links = [x.strip() for x in f.readlines()]
+urls = []
+flags = {}
+
+for l in links:
+ if not l or l.startswith("#"): continue
+ if l.startswith("http"):
+ if "||" in l:
+ l, flag = l.split("||")
+ flags[l] = flag
+ urls.append(l)
+
+ elif len(l.split(" ")) == 2:
+ name, hash = l.split(" ")
+ HosterPluginTester.files[name] = hash
+
+
+hoster, c = c.pluginManager.parseUrls(urls)
+
+plugins = accumulate(hoster)
+for plugin, urls in plugins.iteritems():
+
+ for i, url in enumerate(urls):
+
+
+ def meta(plugin, url, flag, sig):
+ def _test(self):
+ self.test_plugin(plugin, url, flag)
+
+ _test.func_name = sig
+ return _test
+
+ sig = "test_%s_LINK%d" % (plugin, i)
+ setattr(HosterPluginTester, sig, meta(plugin, url, flags.get(url, None), sig))
\ No newline at end of file
--
cgit v1.2.3