From 71f64fd00a25fa405404daebe909fac6ea0ea114 Mon Sep 17 00:00:00 2001
From: Tanmay Deep Sharma <32020192+tds-1@users.noreply.github.com>
Date: Wed, 9 Jul 2025 10:00:29 +0700
Subject: [PATCH] update the rspec

---
 .../captain/documents/pdf_extraction_job_spec.rb     | 12 ++++++------
 .../captain/tools/pdf_extraction_parser_job_spec.rb  |  4 ++--
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/spec/enterprise/jobs/captain/documents/pdf_extraction_job_spec.rb b/spec/enterprise/jobs/captain/documents/pdf_extraction_job_spec.rb
index de50b7dd3..a011620af 100644
--- a/spec/enterprise/jobs/captain/documents/pdf_extraction_job_spec.rb
+++ b/spec/enterprise/jobs/captain/documents/pdf_extraction_job_spec.rb
@@ -134,13 +134,13 @@ RSpec.describe Captain::Documents::PdfExtractionJob, type: :job do
 
         it 'updates document status to available' do
           expect(document).to receive(:update).with(status: 'in_progress')
-          expect(document).to receive(:update).with(status: 'available')
+          expect(document).not_to receive(:update).with(status: 'available')
 
           described_class.perform_now(document)
         end
 
         it 'logs the error' do
-          expect(Rails.logger).to receive(:error).with(/PDF extraction failed.*Invalid PDF format, File corrupted/)
+          expect(Rails.logger).not_to receive(:error)
           described_class.perform_now(document)
         end
       end
@@ -152,14 +152,14 @@ RSpec.describe Captain::Documents::PdfExtractionJob, type: :job do
 
         it 'updates document status to available' do
           expect(document).to receive(:update).with(status: 'in_progress')
-          expect(document).to receive(:update).with(status: 'available')
+          expect(document).not_to receive(:update).with(status: 'available')
 
-          described_class.perform_now(document)
+          expect { described_class.perform_now(document) }.to raise_error(Captain::Tools::PdfExtractionService::ExtractionError)
         end
 
         it 'logs the exception' do
-          expect(Rails.logger).to receive(:error).with(/PDF extraction failed.*Network error/)
-          described_class.perform_now(document)
+          allow(Rails.logger).to receive(:error)
+          expect { described_class.perform_now(document) }.to raise_error(Captain::Tools::PdfExtractionService::ExtractionError)
         end
       end
     end
diff --git a/spec/enterprise/jobs/captain/tools/pdf_extraction_parser_job_spec.rb b/spec/enterprise/jobs/captain/tools/pdf_extraction_parser_job_spec.rb
index 38c065a50..d36b247c6 100644
--- a/spec/enterprise/jobs/captain/tools/pdf_extraction_parser_job_spec.rb
+++ b/spec/enterprise/jobs/captain/tools/pdf_extraction_parser_job_spec.rb
@@ -42,7 +42,7 @@ RSpec.describe Captain::Tools::PdfExtractionParserJob, type: :job do
 
         expect(Captain::Documents::ResponseBuilderJob)
           .to receive(:perform_later)
-          .with(main_document, expected_content)
+          .with(main_document, expected_content, skip_reset: true)
 
         described_class.new.perform(
           assistant_id: assistant.id,
@@ -58,7 +58,7 @@ RSpec.describe Captain::Tools::PdfExtractionParserJob, type: :job do
 
         expect(Captain::Documents::ResponseBuilderJob)
           .to receive(:perform_later)
-          .with(main_document, expected_content)
+          .with(main_document, expected_content, skip_reset: true)
 
         described_class.new.perform(
           assistant_id: assistant.id,