parts: add _Body.paragraphs

Steve Canny · Steve Canny · commit 65d6d1355861 · 2013-12-20T20:10:44.000-08:00
diff --git a/docx/parts.py b/docx/parts.py
@@ -51,3 +51,9 @@ def __init__(self, body_elm):
     def add_paragraph(self):
         p = self._body.add_p()
         return Paragraph(p)
+
+    @property
+    def paragraphs(self):
+        if not hasattr(self._body, 'p'):
+            return ()
+        return tuple([Paragraph(p) for p in self._body.p])
diff --git a/tests/test_parts.py b/tests/test_parts.py
@@ -13,8 +13,9 @@
 
 import pytest
 
-from mock import Mock
+from mock import call, Mock
 
+from .unitdata import a_body
 from .unitutil import class_mock, function_mock, initializer_mock
 
 
@@ -75,3 +76,21 @@ def it_can_add_a_paragraph_to_itself(self, Paragraph_):
         body_elm.add_p.assert_called_once_with()
         Paragraph_.assert_called_once_with(p_elm)
         assert p is Paragraph_.return_value
+
+    def it_provides_access_to_its_paragraphs_as_a_sequence(self, Paragraph_):
+        # mockery ----------------------
+        body_elm = Mock(name='body_elm')
+        p1, p2 = (Mock(name='p1'), Mock(name='p2'))
+        P1, P2 = (Mock(name='Paragraph1'), Mock(name='Paragraph2'))
+        body_elm.p = [p1, p2]
+        body = _Body(body_elm)
+        Paragraph_.side_effect = [P1, P2]
+        # exercise ---------------------
+        paragraphs = body.paragraphs
+        # verify -----------------------
+        assert Paragraph_.mock_calls == [call(p1), call(p2)]
+        assert paragraphs == (P1, P2)
+
+    def it_returns_an_empty_sequence_when_it_contains_no_paragraphs(self):
+        body = _Body(a_body().element)
+        assert body.paragraphs == ()
