broken: refactor _Document.load() to new signature

Steve Canny · Steve Canny · commit 7ef0bd8b1082 · 2013-12-20T20:10:43.000-08:00
diff --git a/docx/parts.py b/docx/parts.py
@@ -20,7 +20,8 @@ class _Document(Part):
     """
     Main document part of a WordprocessingML (WML) package, aka a .docx file.
     """
-    def __init__(self, document_elm):
+    def __init__(self, partname, content_type, document_elm):
+        super(_Document, self).__init__(partname, content_type)
         self._element = document_elm
 
     @property
@@ -32,6 +33,5 @@ def body(self):
     @staticmethod
     def load(partname, content_type, blob):
         document_elm = oxml_fromstring(blob)
-        document = _Document(document_elm)
-        super(_Document, document).__init__(partname, content_type)
+        document = _Document(partname, content_type, document_elm)
         return document
diff --git a/tests/test_parts.py b/tests/test_parts.py
@@ -40,5 +40,5 @@ def it_can_be_constructed_by_opc_part_factory(self, oxml_fromstring_,
         doc = _Document.load(partname, content_type, blob)
         # verify -----------------------
         oxml_fromstring_.assert_called_once_with(blob)
-        init.assert_called_once_with(document_elm)
+        init.assert_called_once_with(partname, content_type, document_elm)
         assert isinstance(doc, _Document)
