style: add _Styles.__len__()

Steve Canny · Steve Canny · commit b8a13ad2ed70 · 2014-02-02T22:33:12.000-08:00
diff --git a/docx/oxml/__init__.py b/docx/oxml/__init__.py
@@ -36,6 +36,9 @@
 register_custom_element_class('w:num', CT_Num)
 register_custom_element_class('w:numbering', CT_Numbering)
 
+from docx.oxml.parts.styles import CT_Styles
+register_custom_element_class('w:styles', CT_Styles)
+
 from docx.oxml.table import CT_Row, CT_Tbl, CT_TblGrid, CT_TblPr, CT_Tc
 register_custom_element_class('w:tbl', CT_Tbl)
 register_custom_element_class('w:tblGrid', CT_TblGrid)
diff --git a/docx/oxml/parts/styles.py b/docx/oxml/parts/styles.py
@@ -4,11 +4,17 @@
 Custom element classes related to the styles part
 """
 
-from docx.oxml.shared import OxmlBaseElement
+from docx.oxml.shared import OxmlBaseElement, qn
 
 
 class CT_Styles(OxmlBaseElement):
     """
     ``<w:styles>`` element, the root element of a styles part, i.e.
     styles.xml
     """
+    @property
+    def style_lst(self):
+        """
+        List of <w:style> child elements.
+        """
+        return self.findall(qn('w:style'))
diff --git a/docx/parts/styles.py b/docx/parts/styles.py
@@ -58,3 +58,6 @@ class _Styles(object):
     def __init__(self, styles_elm):
         super(_Styles, self).__init__()
         self._styles_elm = styles_elm
+
+    def __len__(self):
+        return len(self._styles_elm.style_lst)
diff --git a/features/sty-get-styles-part.feature b/features/sty-get-styles-part.feature
@@ -3,7 +3,6 @@ Feature: Get the document styles part
   As a programmer using the advanced python-docx API
   I need access to the styles part of the document
 
-  @wip
   Scenario: Get an existing styles part from document
     Given a document having a styles part
      When I get the styles part from the document
diff --git a/tests/oxml/unitdata/styles.py b/tests/oxml/unitdata/styles.py
@@ -0,0 +1,27 @@
+# encoding: utf-8
+
+"""
+Test data builders for styles part XML elements
+"""
+
+from ...unitdata import BaseBuilder
+
+
+class CT_StyleBuilder(BaseBuilder):
+    __tag__ = 'w:style'
+    __nspfxs__ = ('w',)
+    __attrs__ = ('w:type', 'w:styleId', 'w:default', 'w:customStyle')
+
+
+class CT_StylesBuilder(BaseBuilder):
+    __tag__ = 'w:styles'
+    __nspfxs__ = ('w',)
+    __attrs__ = ()
+
+
+def a_style():
+    return CT_StyleBuilder()
+
+
+def a_styles():
+    return CT_StylesBuilder()
diff --git a/tests/parts/test_styles.py b/tests/parts/test_styles.py
@@ -15,6 +15,7 @@
 from docx.package import Package
 from docx.parts.styles import StylesPart, _Styles
 
+from ..oxml.unitdata.styles import a_style, a_styles
 from ..unitutil import (
     function_mock, class_mock, initializer_mock, instance_mock, method_mock
 )
@@ -130,3 +131,22 @@ def styles_part_(self, request):
     @pytest.fixture
     def styles_part_load_(self, request):
         return method_mock(request, StylesPart, 'load')
+
+
+class Describe_Styles(object):
+
+    def it_knows_how_many_styles_it_contains(self, len_fixture):
+        styles, style_count = len_fixture
+        assert len(styles) == style_count
+
+    # fixtures -------------------------------------------------------
+
+    @pytest.fixture(params=[0, 1, 2, 3])
+    def len_fixture(self, request):
+        style_count = request.param
+        styles_bldr = a_styles().with_nsdecls()
+        for idx in range(style_count):
+            styles_bldr.with_child(a_style())
+        styles_elm = styles_bldr.element
+        styles = _Styles(styles_elm)
+        return styles, style_count
