auto trim TOC, rewrite test to use actual MPC calls

This commit is contained in:
Willem Melching committed 2026-03-15 13:27:04 +01:00
1 parent 6156a76a91
commit aaa9b83949
8 files changed
+488 -171

No files matched your search

+10 -13
View File
@@ -1,9 +1,6 @@
from pdf_mcp.server import get_pdf_info
class TestGetPdfInfo:
def test_basic_metadata(self, basic_pdf):
info = get_pdf_info(basic_pdf)
def test_basic_metadata(self, basic_pdf, call_tool):
info = call_tool("get_pdf_info", filename=basic_pdf)
assert info["page_count"] == 3
assert info["metadata"]["title"] == "Test Document"
assert info["metadata"]["author"] == "Test Author"
@@ -12,18 +9,18 @@ class TestGetPdfInfo:
assert info["metadata"]["creator"] == "generate.py"
assert info["is_encrypted"] is False
def test_filename_returned(self, basic_pdf):
info = get_pdf_info(basic_pdf)
def test_filename_returned(self, basic_pdf, call_tool):
info = call_tool("get_pdf_info", filename=basic_pdf)
assert info["filename"] == basic_pdf
def test_single_page_pdf(self, diagrams_pdf):
info = get_pdf_info(diagrams_pdf)
def test_single_page_pdf(self, diagrams_pdf, call_tool):
info = call_tool("get_pdf_info", filename=diagrams_pdf)
assert info["page_count"] == 1
def test_multi_page_pdf(self, nested_toc_pdf):
info = get_pdf_info(nested_toc_pdf)
def test_multi_page_pdf(self, nested_toc_pdf, call_tool):
info = call_tool("get_pdf_info", filename=nested_toc_pdf)
assert info["page_count"] == 9
def test_search_targets_page_count(self, search_pdf):
info = get_pdf_info(search_pdf)
def test_search_targets_page_count(self, search_pdf, call_tool):
info = call_tool("get_pdf_info", filename=search_pdf)
assert info["page_count"] == 3