test: fix test_retry_azure_ocr_success to patch process_with_ocr instead of legacy azure task

Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>
2026-02-24 22:55:46 +00:00
parent 37a3f7aae7
commit 44c9f6dc38
1 changed files with 1 additions and 3 deletions
@@ -950,9 +950,7 @@ class TestRetryPipelineSteps:
        db_session.add(file)
        db_session.commit()

-        with patch(
-            "app.tasks.process_with_azure_document_intelligence.process_with_azure_document_intelligence"
-        ) as mock_task:
+        with patch("app.tasks.process_with_ocr.process_with_ocr") as mock_task:
            mock_task.delay.return_value = Mock(id="task-azure")
            result = _retry_pipeline_step(file, "process_with_azure_document_intelligence", db_session)
            assert result["task_id"] == "task-azure"