wehub-resource-sync
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
import zipfile
|
|
|
|
import pytest
|
|
|
|
from services.office_document_service import (
|
|
OfficeDocumentError,
|
|
extract_office_document_text,
|
|
)
|
|
|
|
|
|
def _write_zip(path, files):
|
|
with zipfile.ZipFile(path, "w") as archive:
|
|
for name, content in files.items():
|
|
archive.writestr(name, content)
|
|
|
|
|
|
def test_extracts_pptx_slide_text_in_slide_order(tmp_path):
|
|
path = tmp_path / "deck.pptx"
|
|
_write_zip(
|
|
path,
|
|
{
|
|
"ppt/slides/slide10.xml": "<p:sld xmlns:p='p' xmlns:a='a'><a:t>Ten</a:t></p:sld>",
|
|
"ppt/slides/slide2.xml": "<p:sld xmlns:p='p' xmlns:a='a'><a:t>Two</a:t></p:sld>",
|
|
"ppt/slides/slide1.xml": "<p:sld xmlns:p='p' xmlns:a='a'><a:t>One</a:t></p:sld>",
|
|
},
|
|
)
|
|
|
|
assert extract_office_document_text(str(path)) == "One\n\nTwo\n\nTen"
|
|
|
|
|
|
def test_extracts_docx_paragraphs(tmp_path):
|
|
path = tmp_path / "document.docx"
|
|
_write_zip(
|
|
path,
|
|
{
|
|
"word/document.xml": (
|
|
"<w:document xmlns:w='w'><w:body>"
|
|
"<w:p><w:r><w:t>Hello</w:t></w:r><w:r><w:t>world</w:t></w:r></w:p>"
|
|
"<w:p><w:r><w:t>Second paragraph</w:t></w:r></w:p>"
|
|
"</w:body></w:document>"
|
|
)
|
|
},
|
|
)
|
|
|
|
assert extract_office_document_text(str(path)) == "Hello world\nSecond paragraph"
|
|
|
|
|
|
def test_rejects_legacy_binary_office_formats(tmp_path):
|
|
path = tmp_path / "legacy.ppt"
|
|
path.write_bytes(b"legacy")
|
|
|
|
with pytest.raises(OfficeDocumentError, match="external office conversion engine"):
|
|
extract_office_document_text(str(path))
|