aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_postprocessor.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/test_postprocessor.py')
-rw-r--r--test/test_postprocessor.py19
1 files changed, 15 insertions, 4 deletions
diff --git a/test/test_postprocessor.py b/test/test_postprocessor.py
index 354f9ff..5da3131 100644
--- a/test/test_postprocessor.py
+++ b/test/test_postprocessor.py
@@ -12,12 +12,14 @@ import sys
import unittest
from unittest.mock import Mock, mock_open, patch
+import logging
import zipfile
import tempfile
from datetime import datetime, timezone as tz
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-from gallery_dl import postprocessor, extractor, util, config # noqa E402
+from gallery_dl import extractor, output, util # noqa E402
+from gallery_dl import postprocessor, util, config # noqa E402
from gallery_dl.postprocessor.common import PostProcessor # noqa E402
@@ -25,6 +27,15 @@ class MockPostprocessorModule(Mock):
__postprocessor__ = "mock"
+class FakeJob():
+
+ def __init__(self):
+ self.extractor = extractor.find("test:")
+ self.pathfmt = util.PathFormat(self.extractor)
+ self.out = output.NullOutput()
+ self.get_logger = logging.getLogger
+
+
class TestPostprocessorModule(unittest.TestCase):
def setUp(self):
@@ -58,9 +69,9 @@ class BasePostprocessorTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
- cls.extractor = extractor.find("test:")
cls.dir = tempfile.TemporaryDirectory()
config.set((), "base-directory", cls.dir.name)
+ cls.job = FakeJob()
@classmethod
def tearDownClass(cls):
@@ -74,12 +85,12 @@ class BasePostprocessorTest(unittest.TestCase):
if data is not None:
kwdict.update(data)
- self.pathfmt = util.PathFormat(self.extractor)
+ self.pathfmt = self.job.pathfmt
self.pathfmt.set_directory(kwdict)
self.pathfmt.set_filename(kwdict)
pp = postprocessor.find(self.__class__.__name__[:-4].lower())
- return pp(self.pathfmt, options)
+ return pp(self.job, options)
class ClassifyTest(BasePostprocessorTest):