diff options
Diffstat (limited to 'Biz')
| -rw-r--r-- | Biz/PodcastItLater/Worker.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/Biz/PodcastItLater/Worker.py b/Biz/PodcastItLater/Worker.py index 251f614..ab414ef 100644 --- a/Biz/PodcastItLater/Worker.py +++ b/Biz/PodcastItLater/Worker.py @@ -1972,7 +1972,10 @@ class TestJobProcessing(Test.TestCase): msg = "no job found for %s" raise Test.TestError(msg, self.job_id) - # Mock external calls + def mock_tts(*_args: Any) -> bytes: + shutdown_handler.shutdown_requested.set() + return b"audio-data" + with ( unittest.mock.patch.object( ArticleProcessor, @@ -1987,9 +1990,7 @@ class TestJobProcessing(Test.TestCase): unittest.mock.patch.object( ArticleProcessor, "text_to_speech", - side_effect=lambda *_args: ( - shutdown_handler.shutdown_requested.set() or b"audio-data" # type: ignore[func-returns-value] - ), + side_effect=mock_tts, ), unittest.mock.patch( "Biz.PodcastItLater.Core.Database.update_job_status", |