add _ContentTypeMap.from_xml()

Steve Canny · Steve Canny · commit 321d8a94d977 · 2013-08-05T14:59:31.000-07:00
diff --git a/opc/pkgreader.py b/opc/pkgreader.py
@@ -107,6 +107,15 @@ def from_xml(content_types_xml):
         Return a new |_ContentTypeMap| instance populated with the contents
         of *content_types_xml*.
         """
+        types_elm = oxml_fromstring(content_types_xml)
+        ctmap = _ContentTypeMap()
+        ctmap._overrides = dict(
+            (o.partname, o.content_type) for o in types_elm.overrides
+        )
+        ctmap._defaults = dict(
+            ('.%s' % d.extension, d.content_type) for d in types_elm.defaults
+        )
+        return ctmap
 
 
 class _SerializedPart(object):
diff --git a/tests/test_pkgreader.py b/tests/test_pkgreader.py
@@ -23,6 +23,13 @@
 from .unitutil import class_mock, initializer_mock, method_mock
 
 
+@pytest.fixture
+def oxml_fromstring(request):
+    _patch = patch('opc.pkgreader.oxml_fromstring')
+    request.addfinalizer(_patch.stop)
+    return _patch.start()
+
+
 class DescribePackageReader(object):
 
     @pytest.fixture
@@ -181,6 +188,49 @@ def it_can_retrieve_srels_for_a_source_uri(
         assert retval == srels
 
 
+class Describe_ContentTypeMap(object):
+
+    def it_can_construct_from_types_xml(self, oxml_fromstring):
+        # test data --------------------
+        content_types = (
+            'app/vnd.type1', 'app/vnd.type2', 'app/vnd.type3',
+            'app/vnd.type4',
+        )
+        content_types_xml = '<DontCare/>'
+        extensions = ('rels', 'xml')
+        exts = tuple(['.%s' % extension for extension in extensions])
+        partnames = ('/part/name1.xml', '/part/name2.xml')
+        # mockery ----------------------
+        overrides = (
+            Mock(name='override_elm_1', partname=partnames[0],
+                 content_type=content_types[0]),
+            Mock(name='override_elm_2', partname=partnames[1],
+                 content_type=content_types[1]),
+        )
+        defaults = (
+            Mock(name='default_elm_1', extension=extensions[0],
+                 content_type=content_types[2]),
+            Mock(name='default_elm_2', extension=extensions[1],
+                 content_type=content_types[3]),
+        )
+        types_elm = Mock(
+            name='types_elm', overrides=overrides, defaults=defaults
+        )
+        oxml_fromstring.return_value = types_elm
+        # exercise ---------------------
+        ct_map = _ContentTypeMap.from_xml(content_types_xml)
+        # verify -----------------------
+        expected_overrides = {
+            partnames[0]: content_types[0], partnames[1]: content_types[1]
+        }
+        expected_defaults = {
+            exts[0]: content_types[2], exts[1]: content_types[3]
+        }
+        oxml_fromstring.assert_called_once_with(content_types_xml)
+        assert ct_map._overrides == expected_overrides
+        assert ct_map._defaults == expected_defaults
+
+
 class Describe_SerializedRelationship(object):
 
     def it_remembers_construction_values(self):
