From 071a88d4f3c0c0a3688e9a0c60d30df535007569 Mon Sep 17 00:00:00 2001 From: Rasmus Widing Date: Tue, 2 Sep 2025 17:23:00 +0300 Subject: [PATCH] test: fix tests for cancellation_check parameter MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Update test mocks to include the new cancellation_check parameter added to code extraction methods. 🤖 Generated with Claude Code Co-Authored-By: Claude --- python/tests/test_code_extraction_source_id.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/python/tests/test_code_extraction_source_id.py b/python/tests/test_code_extraction_source_id.py index 5ae87b9f..6cc65440 100644 --- a/python/tests/test_code_extraction_source_id.py +++ b/python/tests/test_code_extraction_source_id.py @@ -27,7 +27,7 @@ class TestCodeExtractionSourceId: # Track what gets passed to the internal extraction method extracted_blocks = [] - async def mock_extract_blocks(crawl_results, source_id, progress_callback=None, start=0, end=100): + async def mock_extract_blocks(crawl_results, source_id, progress_callback=None, start=0, end=100, cancellation_check=None): # Simulate finding code blocks and verify source_id is passed correctly for doc in crawl_results: extracted_blocks.append({ @@ -107,14 +107,15 @@ class TestCodeExtractionSourceId: 100 ) - # Verify the correct source_id was passed + # Verify the correct source_id was passed (now with cancellation_check parameter) mock_extract.assert_called_once_with( crawl_results, url_to_full_document, source_id, # This should be the third argument None, 0, - 100 + 100, + None # cancellation_check parameter ) assert result == 5 @@ -133,7 +134,7 @@ class TestCodeExtractionSourceId: source_ids_seen = [] original_extract = code_service._extract_code_blocks_from_documents - async def track_source_id(crawl_results, source_id, progress_callback=None, start=0, end=100): + async def track_source_id(crawl_results, source_id, progress_callback=None, start=0, end=100, cancellation_check=None): source_ids_seen.append(source_id) return [] # Return empty list to skip further processing