add ZipPkgWriter.__init__()

Steve Canny · Steve Canny · commit 64932b08a23b · 2013-08-05T15:08:41.000-07:00
diff --git a/opc/phys_pkg.py b/opc/phys_pkg.py
@@ -11,7 +11,7 @@
 Provides a general interface to a *physical* OPC package, such as a zip file.
 """
 
-from zipfile import ZipFile
+from zipfile import ZIP_DEFLATED, ZipFile
 
 
 class PhysPkgReader(object):
@@ -76,3 +76,6 @@ class ZipPkgWriter(object):
     """
     Implements |PhysPkgWriter| interface for a zip file OPC package.
     """
+    def __init__(self, pkg_file):
+        super(ZipPkgWriter, self).__init__()
+        self._zipf = ZipFile(pkg_file, 'w', compression=ZIP_DEFLATED)
diff --git a/tests/test_phys_pkg.py b/tests/test_phys_pkg.py
@@ -11,8 +11,12 @@
 
 import hashlib
 
+from zipfile import ZIP_DEFLATED
+
 from opc.packuri import PACKAGE_URI, PackURI
-from opc.phys_pkg import PhysPkgReader, PhysPkgWriter, ZipPkgReader
+from opc.phys_pkg import (
+    PhysPkgReader, PhysPkgWriter, ZipPkgReader, ZipPkgWriter
+)
 
 import pytest
 
@@ -102,3 +106,12 @@ def it_returns_none_when_part_has_no_rels_xml(self, phys_reader):
         partname = PackURI('/ppt/viewProps.xml')
         rels_xml = phys_reader.rels_xml_for(partname)
         assert rels_xml is None
+
+
+class DescribeZipPkgWriter(object):
+
+    def it_opens_pkg_file_zip_on_construction(self, ZipFile_):
+        pkg_file = Mock(name='pkg_file')
+        ZipPkgWriter(pkg_file)
+        ZipFile_.assert_called_once_with(pkg_file, 'w',
+                                         compression=ZIP_DEFLATED)
