github/gallery-dl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add 'filesize-min' and 'filesize-max' options (closes #780)

Browse Source
This commit is contained in:
Mike Fährmann
2020-09-01 22:05:17 +02:00
parent 7876a03ece
commit ac3036ef56
6 changed files with 74 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
test/test_downloader.py
View File

@@ -179,6 +179,9 @@ class TestHTTPDownloader(TestDownloaderBase):
server = http.server.HTTPServer(("", port), HttpRequestHandler)
threading.Thread(target=server.serve_forever, daemon=True).start()
def tearDown(self):
self.downloader.minsize = self.downloader.maxsize = None
def test_http_download(self):
self._run_test(self._jpg, None, DATA_JPG, "jpg", "jpg")
self._run_test(self._png, None, DATA_PNG, "png", "png")
@@ -199,6 +202,20 @@ class TestHTTPDownloader(TestDownloaderBase):
self._run_test(self._png, None, DATA_PNG, "gif", "png")
self._run_test(self._gif, None, DATA_GIF, "jpg", "gif")
def test_http_filesize_min(self):
pathfmt = self._prepare_destination(None, extension=None)
self.downloader.minsize = 100
with self.assertLogs(self.downloader.log, "WARNING"):
success = self.downloader.download(self._gif, pathfmt)
self.assertFalse(success)
def test_http_filesize_max(self):
pathfmt = self._prepare_destination(None, extension=None)
self.downloader.maxsize = 100
with self.assertLogs(self.downloader.log, "WARNING"):
success = self.downloader.download(self._jpg, pathfmt)
self.assertFalse(success)
class TestTextDownloader(TestDownloaderBase):