From b5cdf956247527b5b8a06e2d176dff4bfdd521a4 Mon Sep 17 00:00:00 2001 From: Na'aman Hirschfeld Date: Fri, 14 Feb 2025 23:05:44 +0100 Subject: [PATCH] chore: updated pandoc test --- tests/pandoc_test.py | 30 ++++++++++++++++++++++++++++-- 1 file changed, 28 insertions(+), 2 deletions(-) diff --git a/tests/pandoc_test.py b/tests/pandoc_test.py index 5e02b32..7e2ffab 100644 --- a/tests/pandoc_test.py +++ b/tests/pandoc_test.py @@ -135,7 +135,20 @@ async def test_process_file_with_extra_args(mock_subprocess_run: Mock, docx_docu ) assert isinstance(result, ExtractionResult) assert result.content.strip() == "Sample processed content" - assert "--strip-comments" in mock_subprocess_run.call_args[0][0] + + # Get all pandoc commands that were run + pandoc_calls = [call[0][0] for call in mock_subprocess_run.call_args_list if "pandoc" in call[0][0]] + + # First call should be version check + assert pandoc_calls[0] == ["pandoc", "--version"] + + # Second call should be metadata extraction (no extra args) + assert "--to=json" in pandoc_calls[1] + assert "--strip-comments" not in pandoc_calls[1] + + # Third call should be content extraction (with extra args) + assert "--to=markdown" in pandoc_calls[2] + assert "--strip-comments" in pandoc_calls[2] async def test_process_file_error(mock_subprocess_run: Mock, docx_document: Path) -> None: @@ -170,7 +183,20 @@ async def test_process_content_with_extra_args(mock_subprocess_run: Mock) -> Non ) assert isinstance(result, ExtractionResult) assert result.content.strip() == "Sample processed content" - assert "--strip-comments" in mock_subprocess_run.call_args[0][0] + + # Get all pandoc commands that were run + pandoc_calls = [call[0][0] for call in mock_subprocess_run.call_args_list if "pandoc" in call[0][0]] + + # First call should be version check + assert pandoc_calls[0] == ["pandoc", "--version"] + + # Second call should be metadata extraction (no extra args) + assert "--to=json" in pandoc_calls[1] + assert "--strip-comments" not in pandoc_calls[1] + + # Third call should be content extraction (with extra args) + assert "--to=markdown" in pandoc_calls[2] + assert "--strip-comments" in pandoc_calls[2] async def test_extract_metadata_error(mock_subprocess_run: Mock, docx_document: Path) -> None: