pythoningLearning
diff --git a/‎opc/pkgreader.py‎
Lines changed: 12 additions & 0 deletions b/‎opc/pkgreader.py‎
Lines changed: 12 additions & 0 deletions
diff --git a/‎tests/test_pkgreader.py‎
Lines changed: 21 additions & 0 deletions b/‎tests/test_pkgreader.py‎
Lines changed: 21 additions & 0 deletions
@@ -25,6 +25,7 @@ class PackageReader(object):
     """
     def __init__(self, content_types, pkg_srels, sparts):
         super(PackageReader, self).__init__()
+        self._pkg_srels = pkg_srels
         self._sparts = sparts
 
     @staticmethod
@@ -48,6 +49,17 @@ def iter_sparts(self):
         for spart in self._sparts:
             yield (spart.partname, spart.content_type, spart.blob)
 
+    def iter_srels(self):
+        """
+        Generate a 2-tuple `(source_uri, srel)` for each of the relationships
+        in the package.
+        """
+        for srel in self._pkg_srels:
+            yield (PACKAGE_URI, srel)
+        for spart in self._sparts:
+            for srel in spart.srels:
+                yield (spart.partname, srel)
+
     @staticmethod
     def _load_serialized_parts(phys_reader, pkg_srels, content_types):
         """
 
@@ -103,6 +103,27 @@ def it_can_iterate_over_the_serialized_parts(self):
         assert retval == (partname, content_type, blob)
         assert iter_count == 1
 
+    def it_can_iterate_over_all_the_srels(self):
+        # mockery ----------------------
+        pkg_srels = ['srel1', 'srel2']
+        sparts = [
+            Mock(name='spart1', partname='pn1', srels=['srel3', 'srel4']),
+            Mock(name='spart2', partname='pn2', srels=['srel5', 'srel6']),
+        ]
+        pkg_reader = PackageReader(None, pkg_srels, sparts)
+        # exercise ---------------------
+        generated_tuples = [t for t in pkg_reader.iter_srels()]
+        # verify -----------------------
+        expected_tuples = [
+            ('/',   'srel1'),
+            ('/',   'srel2'),
+            ('pn1', 'srel3'),
+            ('pn1', 'srel4'),
+            ('pn2', 'srel5'),
+            ('pn2', 'srel6'),
+        ]
+        assert generated_tuples == expected_tuples
+
     def it_can_load_serialized_parts(self, _SerializedPart_, _walk_phys_parts):
         # test data --------------------
         test_data = (