refs #27: added a test showing how an sdt tag should be handled.

Jason Ward · Jason Ward · commit e30fffde06ad · 2013-05-20T17:01:41.000-04:00
diff --git a/pydocx/tests/document_builder.py b/pydocx/tests/document_builder.py
@@ -9,6 +9,7 @@
     'p': 'p.xml',
     'pict': 'pict.xml',
     'r': 'r.xml',
+    'sdt': 'sdt.xml',
     'sectPr': 'sectPr.xml',
     'smartTag': 'smart_tag.xml',
     'style': 'style.xml',
@@ -85,6 +86,14 @@ def smart_tag(self, run_tags):
         }
         return template.render(**kwargs)
 
+    @classmethod
+    def sdt_tag(self, p_tag):
+        template = env.get_template(templates['sdt'])
+        kwargs = {
+            'p_tag': p_tag,
+        }
+        return template.render(**kwargs)
+
     @classmethod
     def li(self, text, ilvl, numId, bold=False):
         if isinstance(text, str):
diff --git a/pydocx/tests/templates/sdt.xml b/pydocx/tests/templates/sdt.xml
@@ -0,0 +1,16 @@
+<w:sdt>
+	<w:sdtPr>
+		<w:rPr>
+			<w:rFonts w:ascii="Times New Roman" w:hAnsi="Times New Roman" w:cs="Times New Roman"/>
+			<w:sz w:val="22"/>
+		</w:rPr>
+		<w:alias w:val="PolicyTemplateTitle"/>
+		<w:tag w:val="PolicyTemplateTitle"/>
+		<w:id w:val="95087797"/>
+		<w:lock w:val="sdtLocked"/>
+		<w:text/>
+	</w:sdtPr>
+	<w:sdtContent>
+		{{ p_tag }}
+	</w:sdtContent>
+</w:sdt>
diff --git a/pydocx/tests/test_xml.py b/pydocx/tests/test_xml.py
@@ -764,6 +764,7 @@ class SameNumIdInTable(_TranslationTestCase):
         </ol>
     </body></html>
     '''
+
     # Ensure its not failing somewhere and falling back to decimal
     numbering_dict = {
         '1': {
@@ -786,5 +787,28 @@ def get_xml(self):
         lis += table
         lis += DXB.li(text='CCC', ilvl=0, numId=1)
         body = lis
+
+        xml = DXB.xml(body)
+        return xml
+
+
+class SDTTestCase(_TranslationTestCase):
+    expected_output = '''
+    <html><body>
+        <ol data-list-type="decimal">
+            <li>AAA<br/>
+                BBB
+            </li>
+            <li>CCC</li>
+        </ol>
+    </body></html>
+    '''
+
+    def get_xml(self):
+        body = ''
+        body += DXB.li(text='AAA', ilvl=0, numId=0)
+        body += DXB.sdt_tag(p_tag=DXB.p_tag(text='BBB'))
+        body += DXB.li(text='CCC', ilvl=0, numId=0)
+
         xml = DXB.xml(body)
         return xml
