feat(md2doc): heading and paragraph renderers with inline formatting
This commit is contained in:
@@ -0,0 +1,109 @@
|
||||
"""Tests for heading and paragraph renderers — verify the EFFECT (text + style land)."""
|
||||
import zipfile
|
||||
import xml.etree.ElementTree as ET
|
||||
from pathlib import Path
|
||||
|
||||
from docx import Document
|
||||
from bs4 import BeautifulSoup
|
||||
|
||||
from md2doc import STYLES, _render_heading, _render_paragraph
|
||||
|
||||
W_NS = "http://schemas.openxmlformats.org/wordprocessingml/2006/main"
|
||||
|
||||
|
||||
def get_doc_xml(doc):
|
||||
tmp = Path(__file__).parent / "_tmp_render.docx"
|
||||
doc.save(tmp)
|
||||
try:
|
||||
with zipfile.ZipFile(tmp) as z:
|
||||
return ET.fromstring(z.read("word/document.xml"))
|
||||
finally:
|
||||
tmp.unlink()
|
||||
|
||||
|
||||
def test_render_h1_uses_heading1_style_and_text():
|
||||
doc = Document()
|
||||
style = STYLES["elegant"]
|
||||
node = BeautifulSoup("<h1>My Title</h1>", "html.parser").find("h1")
|
||||
_render_heading(doc, node, level=1, style=style)
|
||||
root = get_doc_xml(doc)
|
||||
pStyles = [p.text for p in root.iter(f"{{{W_NS}}}pStyle")]
|
||||
assert "Heading1" in pStyles
|
||||
texts = [t.text for t in root.iter(f"{{{W_NS}}}t")]
|
||||
assert "My Title" in texts
|
||||
|
||||
|
||||
def test_render_h1_has_bottom_border():
|
||||
doc = Document()
|
||||
style = STYLES["elegant"]
|
||||
node = BeautifulSoup("<h1>Title</h1>", "html.parser").find("h1")
|
||||
_render_heading(doc, node, level=1, style=style)
|
||||
root = get_doc_xml(doc)
|
||||
bottoms = list(root.iter(f"{{{W_NS}}}bottom"))
|
||||
assert len(bottoms) >= 1, "h1 should have bottom border"
|
||||
|
||||
|
||||
def test_render_h1_applies_style_heading_color():
|
||||
"""Heading color should come from STYLES[style]['heading_color']."""
|
||||
doc = Document()
|
||||
style = STYLES["report"]
|
||||
node = BeautifulSoup("<h1>Title</h1>", "html.parser").find("h1")
|
||||
_render_heading(doc, node, level=1, style=style)
|
||||
root = get_doc_xml(doc)
|
||||
colors = [c.get(f"{{{W_NS}}}val") for c in root.iter(f"{{{W_NS}}}color")]
|
||||
assert "1e3a8a" in colors, f"expected heading color 1e3a8a from 'report' style, got {colors}"
|
||||
|
||||
|
||||
def test_render_heading_applies_keep_with_next():
|
||||
"""COMMON['keep_with_next_headings'] should set paragraph_format.keep_with_next=True."""
|
||||
doc = Document()
|
||||
style = STYLES["elegant"]
|
||||
node = BeautifulSoup("<h2>Section</h2>", "html.parser").find("h2")
|
||||
_render_heading(doc, node, level=2, style=style)
|
||||
root = get_doc_xml(doc)
|
||||
keep_next = list(root.iter(f"{{{W_NS}}}keepNext"))
|
||||
assert len(keep_next) >= 1, "expected w:keepNext for heading"
|
||||
|
||||
|
||||
def test_render_h2_no_bottom_border():
|
||||
"""Only h1 gets border; h2/h3/h4 don't."""
|
||||
doc = Document()
|
||||
style = STYLES["elegant"]
|
||||
node = BeautifulSoup("<h2>Subtitle</h2>", "html.parser").find("h2")
|
||||
_render_heading(doc, node, level=2, style=style)
|
||||
root = get_doc_xml(doc)
|
||||
bottoms = list(root.iter(f"{{{W_NS}}}bottom"))
|
||||
assert len(bottoms) == 0
|
||||
|
||||
|
||||
def test_render_paragraph_emits_text():
|
||||
doc = Document()
|
||||
style = STYLES["elegant"]
|
||||
node = BeautifulSoup("<p>Hello world</p>", "html.parser").find("p")
|
||||
_render_paragraph(doc, node, style=style)
|
||||
root = get_doc_xml(doc)
|
||||
texts = [t.text for t in root.iter(f"{{{W_NS}}}t")]
|
||||
assert "Hello world" in texts
|
||||
|
||||
|
||||
def test_render_paragraph_handles_bold_inline():
|
||||
doc = Document()
|
||||
style = STYLES["elegant"]
|
||||
node = BeautifulSoup("<p>This is <strong>bold</strong> text</p>", "html.parser").find("p")
|
||||
_render_paragraph(doc, node, style=style)
|
||||
root = get_doc_xml(doc)
|
||||
b_el = list(root.iter(f"{{{W_NS}}}b"))
|
||||
assert len(b_el) >= 1
|
||||
|
||||
|
||||
def test_render_paragraph_handles_inline_code_with_shading():
|
||||
"""Inline <code> should have monospace font AND run-level shading."""
|
||||
doc = Document()
|
||||
style = STYLES["elegant"]
|
||||
node = BeautifulSoup("<p>See <code>foo()</code> here</p>", "html.parser").find("p")
|
||||
_render_paragraph(doc, node, style=style)
|
||||
root = get_doc_xml(doc)
|
||||
rPr_list = list(root.iter(f"{{{W_NS}}}rPr"))
|
||||
shd_in_runs = [r.find(f"{{{W_NS}}}shd") for r in rPr_list
|
||||
if r.find(f"{{{W_NS}}}shd") is not None]
|
||||
assert len(shd_in_runs) >= 1, "expected w:shd inside a w:rPr for inline code"
|
||||
Reference in New Issue
Block a user