doc: add Document.part

Steve Canny · Steve Canny · commit fb5e7d798c41 · 2015-02-14T16:42:43.000-08:00
diff --git a/docx/document.py b/docx/document.py
@@ -24,6 +24,13 @@ def __init__(self, element, part):
         self._part = part
         self.__body = None
 
+    @property
+    def part(self):
+        """
+        The |DocumentPart| object of this document.
+        """
+        return self._part
+
     @property
     def _body(self):
         """
diff --git a/tests/test_document.py b/tests/test_document.py
@@ -11,13 +11,18 @@
 import pytest
 
 from docx.document import _Body, Document
+from docx.parts.document import DocumentPart
 
 from .unitutil.cxml import element
 from .unitutil.mock import class_mock, instance_mock
 
 
 class DescribeDocument(object):
 
+    def it_provides_access_to_the_document_part(self, part_fixture):
+        document, part_ = part_fixture
+        assert document.part is part_
+
     def it_provides_access_to_the_document_body(self, body_fixture):
         document, body_elm, _Body_, body_ = body_fixture
         body = document._body
@@ -33,6 +38,11 @@ def body_fixture(self, _Body_, body_):
         document = Document(document_elm, None)
         return document, body_elm, _Body_, body_
 
+    @pytest.fixture
+    def part_fixture(self, document_part_):
+        document = Document(None, document_part_)
+        return document, document_part_
+
     # fixture components ---------------------------------------------
 
     @pytest.fixture
@@ -42,3 +52,7 @@ def _Body_(self, request, body_):
     @pytest.fixture
     def body_(self, request):
         return instance_mock(request, _Body)
+
+    @pytest.fixture
+    def document_part_(self, request):
+        return instance_mock(request, DocumentPart)
