doc: add Document._body

Steve Canny · Steve Canny · commit ce2f4cdecf20 · 2015-02-14T16:16:48.000-08:00
diff --git a/docx/document.py b/docx/document.py
@@ -8,6 +8,7 @@
     absolute_import, division, print_function, unicode_literals
 )
 
+from .blkcntnr import BlockItemContainer
 from .shared import ElementProxy
 
 
@@ -16,8 +17,28 @@ class Document(ElementProxy):
     WordprocessingML (WML) document.
     """
 
-    __slots__ = ('_part',)
+    __slots__ = ('_part', '__body')
 
     def __init__(self, element, part):
         super(Document, self).__init__(element)
         self._part = part
+        self.__body = None
+
+    @property
+    def _body(self):
+        """
+        The |_Body| instance containing the content for this document.
+        """
+        if self.__body is None:
+            self.__body = _Body(self._element.body, self)
+        return self.__body
+
+
+class _Body(BlockItemContainer):
+    """
+    Proxy for ``<w:body>`` element in this document, having primarily a
+    container role.
+    """
+    def __init__(self, body_elm, parent):
+        super(_Body, self).__init__(body_elm, parent)
+        self._body = body_elm
diff --git a/tests/test_document.py b/tests/test_document.py
@@ -0,0 +1,44 @@
+# encoding: utf-8
+
+"""
+Test suite for the docx.document module
+"""
+
+from __future__ import (
+    absolute_import, division, print_function, unicode_literals
+)
+
+import pytest
+
+from docx.document import _Body, Document
+
+from .unitutil.cxml import element
+from .unitutil.mock import class_mock, instance_mock
+
+
+class DescribeDocument(object):
+
+    def it_provides_access_to_the_document_body(self, body_fixture):
+        document, body_elm, _Body_, body_ = body_fixture
+        body = document._body
+        _Body_.assert_called_once_with(body_elm, document)
+        assert body is body_
+
+    # fixtures -------------------------------------------------------
+
+    @pytest.fixture
+    def body_fixture(self, _Body_, body_):
+        document_elm = element('w:document/w:body')
+        body_elm = document_elm[0]
+        document = Document(document_elm, None)
+        return document, body_elm, _Body_, body_
+
+    # fixture components ---------------------------------------------
+
+    @pytest.fixture
+    def _Body_(self, request, body_):
+        return class_mock(request, 'docx.document._Body', return_value=body_)
+
+    @pytest.fixture
+    def body_(self, request):
+        return instance_mock(request, _Body)
