parts: integrate _Document with opc part factory

Steve Canny · Steve Canny · commit 041d9013d71c · 2013-12-20T20:10:43.000-08:00
diff --git a/docx/__init__.py b/docx/__init__.py
@@ -10,3 +10,12 @@
 from docx.api import Document  # noqa
 
 __version__ = '0.3.0d1'
+
+
+from opc import PartFactory
+from opc.constants import CONTENT_TYPE as CT
+
+from docx.parts import _Document
+
+
+PartFactory.part_type_for[CT.WML_DOCUMENT_MAIN] = _Document
diff --git a/docx/parts.py b/docx/parts.py
@@ -13,13 +13,25 @@
 
 from opc import Part
 
+from docx.oxml.base import oxml_fromstring
+
 
 class _Document(Part):
     """
     Main document part of a WordprocessingML (WML) package, aka a .docx file.
     """
+    def __init__(self, document_elm):
+        self._element = document_elm
+
     @property
     def body(self):
         """
         The |_Body| instance containing the content for this document.
         """
+
+    @staticmethod
+    def load(partname, content_type, blob):
+        document_elm = oxml_fromstring(blob)
+        document = _Document(document_elm)
+        super(_Document, document).__init__(partname, content_type)
+        return document
diff --git a/tests/test_parts.py b/tests/test_parts.py
@@ -0,0 +1,44 @@
+# -*- coding: utf-8 -*-
+#
+# test_parts.py
+#
+# Copyright (C) 2013 Steve Canny scanny@cisco.com
+#
+# This module is part of python-docx and is released under the MIT License:
+# http://www.opensource.org/licenses/mit-license.php
+
+"""Test suite for the docx.parts module."""
+
+from docx.parts import _Document
+
+import pytest
+
+from mock import Mock
+
+from .unitutil import function_mock, initializer_mock
+
+
+class Describe_Document(object):
+
+    @pytest.fixture
+    def init(self, request):
+        return initializer_mock(_Document, request)
+
+    @pytest.fixture
+    def oxml_fromstring_(self, request):
+        return function_mock('docx.parts.oxml_fromstring', request)
+
+    def it_can_be_constructed_by_opc_part_factory(self, oxml_fromstring_,
+                                                  init):
+        # mockery ----------------------
+        partname, content_type, blob, document_elm = (
+            Mock(name='partname'), Mock(name='content_type'),
+            Mock(name='blob'), Mock(name='document_elm')
+        )
+        oxml_fromstring_.return_value = document_elm
+        # exercise ---------------------
+        doc = _Document.load(partname, content_type, blob)
+        # verify -----------------------
+        oxml_fromstring_.assert_called_once_with(blob)
+        init.assert_called_once_with(document_elm)
+        assert isinstance(doc, _Document)
