pythoningLearning
diff --git a/‎opc/oxml.py‎
Lines changed: 14 additions & 0 deletions b/‎opc/oxml.py‎
Lines changed: 14 additions & 0 deletions
diff --git a/‎tests/test_oxml.py‎
Lines changed: 11 additions & 1 deletion b/‎tests/test_oxml.py‎
Lines changed: 11 additions & 1 deletion
diff --git a/‎tests/unitdata.py‎
Lines changed: 53 additions & 2 deletions b/‎tests/unitdata.py‎
Lines changed: 53 additions & 2 deletions
@@ -149,9 +149,23 @@ def target_mode(self):
         return self.get('TargetMode', RTM.INTERNAL)
 
 
+class CT_Types(OxmlBaseElement):
+    """
+    ``<Types>`` element, the container element for Default and Override
+    elements in [Content_Types].xml.
+    """
+    @property
+    def overrides(self):
+        try:
+            return self.Override[:]
+        except AttributeError:
+            return []
+
+
 ct_namespace = element_class_lookup.get_namespace(nsmap['ct'])
 ct_namespace['Default'] = CT_Default
 ct_namespace['Override'] = CT_Override
+ct_namespace['Types'] = CT_Types
 
 pr_namespace = element_class_lookup.get_namespace(nsmap['pr'])
 pr_namespace['Relationship'] = CT_Relationship
@@ -10,8 +10,9 @@
 """Test suite for opc.oxml module."""
 
 from opc.constants import RELATIONSHIP_TARGET_MODE as RTM
+from opc.oxml import CT_Override
 
-from .unitdata import a_Default, an_Override, a_Relationship
+from .unitdata import a_Default, an_Override, a_Relationship, a_Types
 
 
 class DescribeCT_Default(object):
@@ -38,3 +39,12 @@ def it_provides_read_access_to_xml_values(self):
         assert rel.reltype == 'ReLtYpE'
         assert rel.target_ref == 'docProps/core.xml'
         assert rel.target_mode == RTM.INTERNAL
+
+
+class DescribeCT_Types(object):
+
+    def it_provides_access_to_override_child_elements(self):
+        types = a_Types().element
+        assert len(types.overrides) == 3
+        for override in types.overrides:
+            assert isinstance(override, CT_Override)
@@ -22,6 +22,11 @@ def element(self):
         """Return element based on XML generated by builder"""
         return oxml_fromstring(self.xml)
 
+    def with_indent(self, indent):
+        """Add integer *indent* spaces at beginning of element XML"""
+        self._indent = indent
+        return self
+
 
 class CT_DefaultBuilder(BaseBuilder):
     """
@@ -49,14 +54,27 @@ class CT_OverrideBuilder(BaseBuilder):
     def __init__(self):
         """Establish instance variables with default values"""
         self._content_type = 'app/vnd.type'
+        self._indent = 0
         self._namespace = ' xmlns="%s"' % NS.OPC_CONTENT_TYPES
         self._partname = '/part/name.xml'
 
+    def with_content_type(self, content_type):
+        """Set ContentType attribute to *content_type*"""
+        self._content_type = content_type
+        return self
+
+    def with_partname(self, partname):
+        """Set PartName attribute to *partname*"""
+        self._partname = partname
+        return self
+
     @property
     def xml(self):
         """Return Override element"""
-        tmpl = '<Override%s PartName="%s" ContentType="%s"/>\n'
-        return tmpl % (self._namespace, self._partname, self._content_type)
+        tmpl = '%s<Override%s PartName="%s" ContentType="%s"/>\n'
+        indent = ' ' * self._indent
+        return tmpl % (indent, self._namespace, self._partname,
+                       self._content_type)
 
 
 class CT_RelationshipBuilder(BaseBuilder):
@@ -86,6 +104,34 @@ def xml(self):
                        self._target, self.target_mode)
 
 
+class CT_TypesBuilder(BaseBuilder):
+    """
+    Test data builder for CT_Types (<Types>) XML element, the root element in
+    [Content_Types].xml files
+    """
+    def __init__(self):
+        """Establish instance variables with default values"""
+        self._overrides = (
+            ('/docProps/core.xml', 'app/vnd.type1'),
+            ('/ppt/presentation.xml', 'app/vnd.type2'),
+            ('/docProps/thumbnail.jpeg', 'image/jpeg'),
+        )
+
+    @property
+    def xml(self):
+        """
+        Return XML string based on settings accumulated via method calls
+        """
+        xml = '<Types xmlns="%s">\n' % NS.OPC_CONTENT_TYPES
+        for partname, content_type in self._overrides:
+            xml += (an_Override().with_partname(partname)
+                                 .with_content_type(content_type)
+                                 .with_indent(2)
+                                 .xml)
+        xml += '</Types>\n'
+        return xml
+
+
 def a_Default():
     """Return a CT_DefaultBuilder instance"""
     return CT_DefaultBuilder()
@@ -99,3 +145,8 @@ def an_Override():
 def a_Relationship():
     """Return a CT_RelationshipBuilder instance"""
     return CT_RelationshipBuilder()
+
+
+def a_Types():
+    """Return a CT_TypesBuilder instance"""
+    return CT_TypesBuilder()