Wire up code for converting comments

2 files changed, 16 insertions, 0 deletions

diff --git a/tests/mammoth_tests.py b/tests/mammoth_tests.py
index 201df70..8ad9b33 100644
--- a/tests/mammoth_tests.py
+++ b/tests/mammoth_tests.py
@@ -185,6 +185,22 @@ def relationships_are_handled_properly_in_footnotes():
 
 
 @istest
+def when_style_mapping_is_defined_for_comment_references_then_comments_are_included():
+    expected_html = (
+        '<p>Ouch' +
+        '<sup><a href="#doc-42-comment-0" id="doc-42-comment-ref-0">[MW 1]</a></sup>.' +
+        '<sup><a href="#doc-42-comment-2" id="doc-42-comment-ref-2">[MW 2]</a></sup></p>' +
+        '<dl><dt id="doc-42-comment-0">MW 1</dt><dd><p>A tachyon walks into a bar. <a href="#doc-42-comment-ref-0">↑</a></p></dd>' +
+        '<dt id="doc-42-comment-2">MW 2</dt><dd><p>Fin. <a href="#doc-42-comment-ref-2">↑</a></p></dd></dl>'
+    )
+    
+    with open(test_path("comments.docx"), "rb") as fileobj:
+        result = mammoth.convert_to_html(fileobj=fileobj, id_prefix="doc-42-", style_map="comment-reference => sup")
+        assert_equal([], result.messages)
+        assert_equal(expected_html, result.value)
+
+
+@istest
 def text_boxes_are_read():
     with open(test_path("text-box.docx"), "rb") as fileobj:
         result = mammoth.convert_to_html(fileobj=fileobj)
diff --git a/tests/test-data/comments.docx b/tests/test-data/comments.docx
new file mode 100755
index 0000000..32cdae1
--- /dev/null
+++ b/tests/test-data/comments.docx