From 877f17f886e18b95e13a9b749b68048be630a907 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Thu, 5 Mar 2026 17:43:57 +0000 Subject: [PATCH] Fixing some tests --- tests/unit/test_chunking/test_recursive_chunker.py | 7 ++++++- tests/unit/test_decoding/test_pdf_decoder.py | 6 ++++++ 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/tests/unit/test_chunking/test_recursive_chunker.py b/tests/unit/test_chunking/test_recursive_chunker.py index ef4aec69..41286eae 100644 --- a/tests/unit/test_chunking/test_recursive_chunker.py +++ b/tests/unit/test_chunking/test_recursive_chunker.py @@ -176,6 +176,9 @@ class TestRecursiveChunkerSimple(IsolatedAsyncioTestCase): processor = Processor(**config) + # Mock save_child_document to avoid waiting for librarian response + processor.save_child_document = AsyncMock(return_value="mock-doc-id") + # Mock message with TextDocument mock_message = MagicMock() mock_text_doc = MagicMock() @@ -192,11 +195,13 @@ class TestRecursiveChunkerSimple(IsolatedAsyncioTestCase): # Mock consumer and flow with parameter overrides mock_consumer = MagicMock() mock_producer = AsyncMock() + mock_triples_producer = AsyncMock() mock_flow = MagicMock() mock_flow.side_effect = lambda param: { "chunk-size": 1500, "chunk-overlap": 150, - "output": mock_producer + "output": mock_producer, + "triples": mock_triples_producer, }.get(param) # Act diff --git a/tests/unit/test_decoding/test_pdf_decoder.py b/tests/unit/test_decoding/test_pdf_decoder.py index 1e6021d1..bb54a154 100644 --- a/tests/unit/test_decoding/test_pdf_decoder.py +++ b/tests/unit/test_decoding/test_pdf_decoder.py @@ -80,6 +80,9 @@ class TestPdfDecoderProcessor(IsolatedAsyncioTestCase): processor = Processor(**config) + # Mock save_child_document to avoid waiting for librarian response + processor.save_child_document = AsyncMock(return_value="mock-doc-id") + await processor.on_message(mock_msg, None, mock_flow) # Verify output was sent for each page @@ -150,6 +153,9 @@ class TestPdfDecoderProcessor(IsolatedAsyncioTestCase): processor = Processor(**config) + # Mock save_child_document to avoid waiting for librarian response + processor.save_child_document = AsyncMock(return_value="mock-doc-id") + await processor.on_message(mock_msg, None, mock_flow) mock_output_flow.send.assert_called_once()