add ZipPkgReader.content_types_xml

Steve Canny · Steve Canny · commit ad52d8d89051 · 2013-08-05T14:56:07.000-07:00
diff --git a/opc/phys_pkg.py b/opc/phys_pkg.py
@@ -26,6 +26,8 @@ class ZipPkgReader(object):
     """
     Implements |PhysPkgReader| interface for a zip file OPC package.
     """
+    _CONTENT_TYPES_MEMBERNAME = '[Content_Types].xml'
+
     def __init__(self, pkg_file):
         super(ZipPkgReader, self).__init__()
         self._zipf = ZipFile(pkg_file, 'r')
@@ -41,3 +43,4 @@ def content_types_xml(self):
         """
         Return the `[Content_Types].xml` blob from the zip package.
         """
+        return self._zipf.read(self._CONTENT_TYPES_MEMBERNAME)
diff --git a/tests/test_phys_pkg.py b/tests/test_phys_pkg.py
@@ -9,13 +9,18 @@
 
 """Test suite for opc.phys_pkg module."""
 
+import hashlib
+
 from opc.phys_pkg import PhysPkgReader, ZipPkgReader
 
 import pytest
 
 from mock import Mock
 
-from .unitutil import class_mock
+from .unitutil import abspath, class_mock
+
+
+test_pptx_path = abspath('test_files/test.pptx')
 
 
 @pytest.fixture
@@ -41,6 +46,12 @@ def it_constructs_a_pkg_reader_instance(self, ZipPkgReader_):
 
 class DescribeZipPkgReader(object):
 
+    @pytest.fixture(scope='class')
+    def phys_reader(self, request):
+        phys_reader = ZipPkgReader(test_pptx_path)
+        request.addfinalizer(phys_reader.close)
+        return phys_reader
+
     def it_opens_pkg_file_zip_on_construction(self, ZipFile_):
         pkg_file = Mock(name='pkg_file')
         ZipPkgReader(pkg_file)
@@ -54,3 +65,7 @@ def it_can_be_closed(self, ZipFile_):
         zip_pkg_reader.close()
         # verify -----------------------
         zipf.close.assert_called_once_with()
+
+    def it_has_the_content_types_xml(self, phys_reader):
+        sha1 = hashlib.sha1(phys_reader.content_types_xml).hexdigest()
+        assert sha1 == '9604a4fb3bf9626f5ad59a4e82029b3a501f106a'
