Mirrors/gallery-dl
1
0
Fork 0
mirror of https://github.com/mikf/gallery-dl.git synced 2025-01-31 19:51:34 +01:00
Code Issues Releases Wiki Activity

[postprocessor:python] add tests

Browse Source
This commit is contained in:
Mike Fährmann 2023-07-24 15:22:57 +02:00
parent 6c9432165e
commit f0203b7559
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88
1 changed files with 34 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
test/test_postprocessor.py
View File

@ -579,6 +579,40 @@ class MtimeTest(BasePostprocessorTest):
self.assertEqual(self.pathfmt.kwdict["_mtime"], 315532800)
class PythonTest(BasePostprocessorTest):
def test_module(self):
path = os.path.join(self.dir.name, "module.py")
self._write_module(path)
sys.path.insert(0, self.dir.name)
try:
self._create({"function": "module:calc"}, {"_value": 123})
finally:
del sys.path[0]
self.assertNotIn("_result", self.pathfmt.kwdict)
self._trigger()
self.assertEqual(self.pathfmt.kwdict["_result"], 246)
def test_path(self):
path = os.path.join(self.dir.name, "module.py")
self._write_module(path)
self._create({"function": path + ":calc"}, {"_value": 12})
self.assertNotIn("_result", self.pathfmt.kwdict)
self._trigger()
self.assertEqual(self.pathfmt.kwdict["_result"], 24)
def _write_module(self, path):
with open(path, "w") as fp:
fp.write("""
def calc(kwdict):
kwdict["_result"] = kwdict["_value"] * 2
""")
class ZipTest(BasePostprocessorTest):
def test_zip_default(self):