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

[common] include duration in 'wait()' output

Browse Source
This commit is contained in:
Mike Fährmann
2026-02-13 20:44:46 +01:00
parent d2477a94af
commit f1da162d72
2 changed files with 31 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
test/test_extractor.py
View File

@@ -217,7 +217,23 @@ class TestExtractorWait(unittest.TestCase):
calls = log.info.mock_calls
self.assertEqual(len(calls), 1)
self._assert_isotime(calls[0][1][1], until)
self.assertEqual(calls[0][1][1], "6 seconds")
self._assert_isotime(calls[0][1][2], until)
def test_wait_seconds_long(self):
extr = extractor.find("generic:https://example.org/")
seconds = 5000
until = time.time() + seconds
with patch("time.sleep") as sleep, patch.object(extr, "log") as log:
extr.wait(seconds=seconds)
sleep.assert_called_once_with(5001.0)
calls = log.info.mock_calls
self.assertEqual(len(calls), 1)
self.assertEqual(calls[0][1][1], "1h 23min")
self._assert_isotime(calls[0][1][2], until)
def test_wait_until(self):
extr = extractor.find("generic:https://example.org/")
@@ -232,7 +248,8 @@ class TestExtractorWait(unittest.TestCase):
calls = log.info.mock_calls
self.assertEqual(len(calls), 1)
self._assert_isotime(calls[0][1][1], until)
self.assertEqual(calls[0][1][1], "5 seconds")
self._assert_isotime(calls[0][1][2], until)
def test_wait_until_datetime(self):
extr = extractor.find("generic:https://example.org/")
@@ -251,7 +268,8 @@ class TestExtractorWait(unittest.TestCase):
calls = log.info.mock_calls
self.assertEqual(len(calls), 1)
self._assert_isotime(calls[0][1][1], until_local)
self.assertEqual(calls[0][1][1], "5 seconds")
self._assert_isotime(calls[0][1][2], until_local)
def _assert_isotime(self, output, until):
if not isinstance(until, dt.datetime):