doc: add Document.styles

Steve Canny · Steve Canny · commit 2afdf826708b · 2015-02-15T03:42:28.000-08:00
diff --git a/docx/document.py b/docx/document.py
@@ -147,6 +147,13 @@ def sections(self):
         """
         return Sections(self._element)
 
+    @property
+    def styles(self):
+        """
+        A |Styles| object providing access to the styles in this document.
+        """
+        return self._part.styles
+
     @property
     def _body(self):
         """
diff --git a/tests/test_document.py b/tests/test_document.py
@@ -17,6 +17,7 @@
 from docx.parts.document import DocumentPart, InlineShapes
 from docx.section import Sections
 from docx.shape import InlineShape
+from docx.styles.styles import Styles
 from docx.table import Table
 from docx.text.paragraph import Paragraph
 from docx.text.run import Run
@@ -92,8 +93,7 @@ def it_provides_access_to_its_core_properties(self, core_props_fixture):
 
     def it_provides_access_to_its_inline_shapes(self, inline_shapes_fixture):
         document, inline_shapes_ = inline_shapes_fixture
-        inline_shapes = document.inline_shapes
-        assert inline_shapes is inline_shapes_
+        assert document.inline_shapes is inline_shapes_
 
     def it_provides_access_to_its_paragraphs(self, paragraphs_fixture):
         document, paragraphs_ = paragraphs_fixture
@@ -106,6 +106,10 @@ def it_provides_access_to_its_sections(self, sections_fixture):
         Sections_.assert_called_once_with(document._element)
         assert sections is sections_
 
+    def it_provides_access_to_its_styles(self, styles_fixture):
+        document, styles_ = styles_fixture
+        assert document.styles is styles_
+
     def it_provides_access_to_the_document_part(self, part_fixture):
         document, part_ = part_fixture
         assert document.part is part_
@@ -226,6 +230,12 @@ def sections_fixture(self, Sections_, sections_):
         Sections_.return_value = sections_
         return document, Sections_, sections_
 
+    @pytest.fixture
+    def styles_fixture(self, document_part_, styles_):
+        document = Document(None, document_part_)
+        document_part_.styles = styles_
+        return document, styles_
+
     # fixture components ---------------------------------------------
 
     @pytest.fixture
@@ -284,6 +294,10 @@ def Sections_(self, request):
     def sections_(self, request):
         return instance_mock(request, Sections)
 
+    @pytest.fixture
+    def styles_(self, request):
+        return instance_mock(request, Styles)
+
     @pytest.fixture
     def table_(self, request):
         return instance_mock(request, Table, style='UNASSIGNED')
