add PackageReader.iter_sparts()

Steve Canny · Steve Canny · commit 6bb2b382b563 · 2013-08-05T14:56:07.000-07:00
diff --git a/opc/pkgreader.py b/opc/pkgreader.py
@@ -23,6 +23,7 @@ class PackageReader(object):
     """
     def __init__(self, content_types, pkg_srels, sparts):
         super(PackageReader, self).__init__()
+        self._sparts = sparts
 
     @staticmethod
     def from_file(pkg_file):
@@ -37,6 +38,14 @@ def from_file(pkg_file):
         phys_reader.close()
         return PackageReader(content_types, pkg_srels, sparts)
 
+    def iter_sparts(self):
+        """
+        Generate a 3-tuple `(partname, content_type, blob)` for each of the
+        serialized parts in the package.
+        """
+        for spart in self._sparts:
+            yield (spart.partname, spart.content_type, spart.blob)
+
     @staticmethod
     def _load_serialized_parts(phys_reader, pkg_srels, content_types):
         """
diff --git a/tests/test_pkgreader.py b/tests/test_pkgreader.py
@@ -62,3 +62,18 @@ def it_can_construct_from_pkg_file(self, init, PhysPkgReader_, from_xml,
         phys_reader.close.assert_called_once_with()
         init.assert_called_once_with(content_types, pkg_srels, sparts)
         assert isinstance(pkg_reader, PackageReader)
+
+    def it_can_iterate_over_the_serialized_parts(self):
+        # mockery ----------------------
+        partname, content_type, blob = ('part/name.xml', 'app/vnd.type',
+                                        '<Part_1/>')
+        spart = Mock(name='spart', partname=partname,
+                     content_type=content_type, blob=blob)
+        pkg_reader = PackageReader(None, None, [spart])
+        iter_count = 0
+        # exercise ---------------------
+        for retval in pkg_reader.iter_sparts():
+            iter_count += 1
+        # verify -----------------------
+        assert retval == (partname, content_type, blob)
+        assert iter_count == 1
