parts: add _Document.body

Steve Canny · Steve Canny · commit aeaf77bb2a67 · 2013-12-20T20:10:43.000-08:00
diff --git a/docx/parts.py b/docx/parts.py
@@ -29,9 +29,17 @@ def body(self):
         """
         The |_Body| instance containing the content for this document.
         """
+        return _Body(self._element.body)
 
     @staticmethod
     def load(partname, content_type, blob):
         document_elm = oxml_fromstring(blob)
         document = _Document(partname, content_type, document_elm)
         return document
+
+
+class _Body(object):
+    """
+    Proxy for ``<w:body>`` element in this document, having primarily a
+    container role.
+    """
diff --git a/tests/test_parts.py b/tests/test_parts.py
@@ -15,11 +15,15 @@
 
 from mock import Mock
 
-from .unitutil import function_mock, initializer_mock
+from .unitutil import class_mock, function_mock, initializer_mock
 
 
 class Describe_Document(object):
 
+    @pytest.fixture
+    def _Body_(self, request):
+        return class_mock('docx.parts._Body', request)
+
     @pytest.fixture
     def init(self, request):
         return initializer_mock(_Document, request)
@@ -28,8 +32,8 @@ def init(self, request):
     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):
+    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'),
@@ -42,3 +46,13 @@ def it_can_be_constructed_by_opc_part_factory(self, oxml_fromstring_,
         oxml_fromstring_.assert_called_once_with(blob)
         init.assert_called_once_with(partname, content_type, document_elm)
         assert isinstance(doc, _Document)
+
+    def it_has_a_body(self, init, _Body_):
+        # mockery ----------------------
+        doc = _Document(None, None, None)
+        doc._element = Mock(name='_element')
+        # exercise ---------------------
+        body = doc.body
+        # verify -----------------------
+        _Body_.assert_called_once_with(doc._element.body)
+        assert body is _Body_.return_value
