From 8da022930409a61f4844c17473307ac0b80fab10 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Thu, 5 Mar 2026 17:47:43 +0000 Subject: [PATCH] Fixing some tests --- tests/unit/test_decoding/test_pdf_decoder.py | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/tests/unit/test_decoding/test_pdf_decoder.py b/tests/unit/test_decoding/test_pdf_decoder.py index bb54a154..1b6e219f 100644 --- a/tests/unit/test_decoding/test_pdf_decoder.py +++ b/tests/unit/test_decoding/test_pdf_decoder.py @@ -69,9 +69,13 @@ class TestPdfDecoderProcessor(IsolatedAsyncioTestCase): mock_msg = MagicMock() mock_msg.value.return_value = mock_document - # Mock flow + # Mock flow - separate mocks for output and triples mock_output_flow = AsyncMock() - mock_flow = MagicMock(return_value=mock_output_flow) + mock_triples_flow = AsyncMock() + mock_flow = MagicMock(side_effect=lambda name: { + "output": mock_output_flow, + "triples": mock_triples_flow, + }.get(name)) config = { 'id': 'test-pdf-decoder', @@ -87,6 +91,8 @@ class TestPdfDecoderProcessor(IsolatedAsyncioTestCase): # Verify output was sent for each page assert mock_output_flow.send.call_count == 2 + # Verify triples were sent for each page (provenance) + assert mock_triples_flow.send.call_count == 2 @patch('trustgraph.base.chunking_service.Consumer') @patch('trustgraph.base.chunking_service.Producer') @@ -143,8 +149,13 @@ class TestPdfDecoderProcessor(IsolatedAsyncioTestCase): mock_msg = MagicMock() mock_msg.value.return_value = mock_document + # Mock flow - separate mocks for output and triples mock_output_flow = AsyncMock() - mock_flow = MagicMock(return_value=mock_output_flow) + mock_triples_flow = AsyncMock() + mock_flow = MagicMock(side_effect=lambda name: { + "output": mock_output_flow, + "triples": mock_triples_flow, + }.get(name)) config = { 'id': 'test-pdf-decoder',