diff options
Diffstat (limited to 'Biz/PodcastItLater')
| -rw-r--r-- | Biz/PodcastItLater/Test.py | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/Biz/PodcastItLater/Test.py b/Biz/PodcastItLater/Test.py index 86b04f4..8b41e79 100644 --- a/Biz/PodcastItLater/Test.py +++ b/Biz/PodcastItLater/Test.py @@ -22,7 +22,9 @@ import Biz.PodcastItLater.Core as Core import Biz.PodcastItLater.UI as UI import Biz.PodcastItLater.Web as Web -import Biz.PodcastItLater.Worker as Worker +import Biz.PodcastItLater.Worker.Core as WorkerCore +import Biz.PodcastItLater.Worker.Jobs as WorkerJobs +import Biz.PodcastItLater.Worker.Processor as WorkerProcessor import Omni.App as App import Omni.Test as Test import pathlib @@ -93,15 +95,15 @@ class TestEndToEnd(BaseWebTest): self.assertEqual(job["user_id"], self.user_id) # Step 2: Process the job with mocked external services - shutdown_handler = Worker.ShutdownHandler() - processor = Worker.ArticleProcessor(shutdown_handler) + shutdown_handler = WorkerCore.ShutdownHandler() + processor = WorkerProcessor.ArticleProcessor(shutdown_handler) # Mock external dependencies mock_audio_data = b"fake-mp3-audio-content-12345" with ( unittest.mock.patch.object( - Worker.ArticleProcessor, + WorkerProcessor.ArticleProcessor, "extract_article_content", return_value=( "Great Article Title", @@ -109,11 +111,11 @@ class TestEndToEnd(BaseWebTest): ), ), unittest.mock.patch( - "Biz.PodcastItLater.Worker.prepare_text_for_tts", + "Biz.PodcastItLater.Worker.TextProcessing.prepare_text_for_tts", return_value=["This is the article content."], ), unittest.mock.patch( - "Biz.PodcastItLater.Worker.check_memory_usage", + "Biz.PodcastItLater.Worker.Processor.check_memory_usage", return_value=50.0, ), unittest.mock.patch.object( @@ -148,7 +150,7 @@ class TestEndToEnd(BaseWebTest): mock_audio_segment.return_value = mock_segment # Process the pending job - Worker.process_pending_jobs(processor) + WorkerJobs.process_pending_jobs(processor) # Step 3: Verify job was marked completed job = Core.Database.get_job_by_id(job_id) |