style: add LatentStyles.__len__()

Steve Canny · Steve Canny · commit 391204383712 · 2015-02-08T00:01:47.000-08:00
diff --git a/docx/oxml/__init__.py b/docx/oxml/__init__.py
@@ -113,8 +113,9 @@ def OxmlElement(nsptag_str, attrs=None, nsdecls=None):
 register_element_cls('wp:extent',     CT_PositiveSize2D)
 register_element_cls('wp:inline',     CT_Inline)
 
-from .styles import CT_Style, CT_Styles
+from .styles import CT_LatentStyles, CT_Style, CT_Styles
 register_element_cls('w:basedOn',        CT_String)
+register_element_cls('w:latentStyles',   CT_LatentStyles)
 register_element_cls('w:locked',         CT_OnOff)
 register_element_cls('w:name',           CT_String)
 register_element_cls('w:qFormat',        CT_OnOff)
diff --git a/docx/oxml/styles.py b/docx/oxml/styles.py
@@ -30,6 +30,15 @@ def styleId_from_name(name):
     }.get(name, name.replace(' ', ''))
 
 
+class CT_LatentStyles(BaseOxmlElement):
+    """
+    `w:latentStyles` element, defining behavior defaults for latent styles
+    and containing `w:lsdException` child elements that each override those
+    defaults for a named latent style.
+    """
+    lsdException = ZeroOrMore('w:lsdException', successors=())
+
+
 class CT_Style(BaseOxmlElement):
     """
     A ``<w:style>`` element, representing a style definition
diff --git a/docx/styles/latent.py b/docx/styles/latent.py
@@ -20,6 +20,9 @@ class LatentStyles(ElementProxy):
 
     __slots__ = ()
 
+    def __len__(self):
+        return len(self._element.lsdException_lst)
+
 
 class _LatentStyle(ElementProxy):
     """
diff --git a/features/sty-access-latent-styles.feature b/features/sty-access-latent-styles.feature
@@ -4,7 +4,6 @@ Feature: Access latent styles for a document
   I need access to the latent styles collection
 
 
-  @wip
   Scenario: Access latent styles collection
     Given the style collection of a document
      Then styles.latent_styles is the LatentStyles object for the document
diff --git a/tests/styles/test_latent.py b/tests/styles/test_latent.py
@@ -0,0 +1,34 @@
+# encoding: utf-8
+
+"""
+Unit test suite for the docx.styles.latent module
+"""
+
+from __future__ import (
+    absolute_import, division, print_function, unicode_literals
+)
+
+import pytest
+
+from docx.styles.latent import LatentStyles
+
+from ..unitutil.cxml import element
+
+
+class DescribeLatentStyles(object):
+
+    def it_knows_how_many_latent_styles_it_contains(self, len_fixture):
+        latent_styles, expected_value = len_fixture
+        assert len(latent_styles) == expected_value
+
+    # fixture --------------------------------------------------------
+
+    @pytest.fixture(params=[
+        ('w:latentStyles',                                  0),
+        ('w:latentStyles/w:lsdException',                   1),
+        ('w:latentStyles/(w:lsdException,w:lsdException)',  2),
+    ])
+    def len_fixture(self, request):
+        latentStyles_cxml, count = request.param
+        latent_styles = LatentStyles(element(latentStyles_cxml))
+        return latent_styles, count
