add PackageWriter._write_parts()

Steve Canny · Steve Canny · commit e1661785ea98 · 2013-08-06T01:35:04.000-07:00
diff --git a/opc/pkgwriter.py b/opc/pkgwriter.py
@@ -53,7 +53,10 @@ def _write_parts(phys_writer, parts):
         Write the blob of each part in *parts* to the package, along with a
         rels item for its relationships if and only if it has any.
         """
-        raise NotImplementedError()
+        for part in parts:
+            phys_writer.write(part.partname, part.blob)
+            if len(part._rels):
+                phys_writer.write(part.partname.rels_uri, part._rels.xml)
 
     @staticmethod
     def _write_pkg_rels(phys_writer, pkg_rels):
diff --git a/tests/test_pkgwriter.py b/tests/test_pkgwriter.py
@@ -11,7 +11,7 @@
 
 import pytest
 
-from mock import call, Mock, patch
+from mock import call, MagicMock, Mock, patch
 
 from opc.constants import CONTENT_TYPE as CT
 from opc.packuri import PackURI
@@ -90,6 +90,23 @@ def it_can_write_a_pkg_rels_item(self):
         phys_writer.write.assert_called_once_with('/_rels/.rels',
                                                   pkg_rels.xml)
 
+    def it_can_write_a_list_of_parts(self):
+        # mockery ----------------------
+        phys_writer = Mock(name='phys_writer')
+        rels = MagicMock(name='rels')
+        rels.__len__.return_value = 1
+        part1 = Mock(name='part1', _rels=rels)
+        part2 = Mock(name='part2', _rels=[])
+        # exercise ---------------------
+        PackageWriter._write_parts(phys_writer, [part1, part2])
+        # verify -----------------------
+        expected_calls = [
+            call(part1.partname, part1.blob),
+            call(part1.partname.rels_uri, part1._rels.xml),
+            call(part2.partname, part2.blob),
+        ]
+        assert phys_writer.write.mock_calls == expected_calls
+
 
 class Describe_ContentTypesItem(object):
 
