add PackageWriter._write_content_types_stream()

Steve Canny · Steve Canny · commit 7423e5c002e9 · 2013-08-05T15:08:41.000-07:00
diff --git a/opc/pkgwriter.py b/opc/pkgwriter.py
@@ -12,6 +12,7 @@
 Convention (OPC) package, essentially an implementation of OpcPackage.save()
 """
 
+from opc.packuri import CONTENT_TYPES_URI
 from opc.phys_pkg import PhysPkgWriter
 
 
@@ -41,7 +42,7 @@ def _write_content_types_stream(phys_writer, parts):
         Write ``[Content_Types].xml`` part to the physical package with an
         appropriate content type lookup target for each part in *parts*.
         """
-        raise NotImplementedError()
+        phys_writer.write(CONTENT_TYPES_URI, _ContentTypesItem.xml_for(parts))
 
     @staticmethod
     def _write_parts(phys_writer, parts):
@@ -58,3 +59,18 @@ def _write_pkg_rels(phys_writer, pkg_rels):
         package.
         """
         raise NotImplementedError()
+
+
+class _ContentTypesItem(object):
+    """
+    Service class that composes a content types item ([Content_Types].xml)
+    based on a list of parts. Not meant to be instantiated, its single
+    interface method is xml_for(), e.g. ``_ContentTypesItem.xml_for(parts)``.
+    """
+    @staticmethod
+    def xml_for(parts):
+        """
+        Return content types XML mapping each part in *parts* to the
+        appropriate content type and suitable for storage as
+        ``[Content_Types].xml`` in an OPC package.
+        """
diff --git a/tests/test_pkgwriter.py b/tests/test_pkgwriter.py
@@ -13,7 +13,9 @@
 
 from mock import call, Mock, patch
 
-from opc.pkgwriter import PackageWriter
+from opc.pkgwriter import _ContentTypesItem, PackageWriter
+
+from .unitutil import method_mock
 
 
 class DescribePackageWriter(object):
@@ -24,6 +26,10 @@ def PhysPkgWriter_(self, request):
         request.addfinalizer(_patch.stop)
         return _patch.start()
 
+    @pytest.fixture
+    def xml_for(self, request):
+        return method_mock(_ContentTypesItem, 'xml_for', request)
+
     @pytest.fixture
     def _write_methods(self, request):
         """Mock that patches all the _write_* methods of PackageWriter"""
@@ -60,3 +66,14 @@ def it_can_write_a_package(self, PhysPkgWriter_, _write_methods):
         PhysPkgWriter_.assert_called_once_with(pkg_file)
         assert _write_methods.mock_calls == expected_calls
         phys_writer.close.assert_called_once_with()
+
+    def it_can_write_a_content_types_stream(self, xml_for):
+        # mockery ----------------------
+        phys_writer = Mock(name='phys_writer')
+        parts = Mock(name='parts')
+        # exercise ---------------------
+        PackageWriter._write_content_types_stream(phys_writer, parts)
+        # verify -----------------------
+        xml_for.assert_called_once_with(parts)
+        phys_writer.write.assert_called_once_with('/[Content_Types].xml',
+                                                  xml_for.return_value)
