style: add LatentStyles.add_style()

Steve Canny · Steve Canny · commit f3fa3a90226f · 2015-02-08T00:01:48.000-08:00
diff --git a/docx/styles/latent.py b/docx/styles/latent.py
@@ -35,6 +35,16 @@ def __iter__(self):
     def __len__(self):
         return len(self._element.lsdException_lst)
 
+    def add_latent_style(self, name):
+        """
+        Return a newly added |_LatentStyle| object to override the inherited
+        defaults defined in this latent styles object for the built-in style
+        having *name*.
+        """
+        lsdException = self._element.add_lsdException()
+        lsdException.name = name
+        return _LatentStyle(lsdException)
+
     @property
     def default_priority(self):
         """
diff --git a/features/sty-latent-add-del.feature b/features/sty-latent-add-del.feature
@@ -4,7 +4,6 @@ Feature: Add or delete a latent style
   I need a way to add and delete a latent style
 
 
-  @wip
   Scenario: Add a latent style
     Given a document having known styles
      When I add a latent style named 'Foobar'
diff --git a/tests/styles/test_latent.py b/tests/styles/test_latent.py
@@ -117,6 +117,15 @@ def priority_set_fixture(self, request):
 
 class DescribeLatentStyles(object):
 
+    def it_can_add_a_latent_style(self, add_fixture):
+        latent_styles, name, expected_xml = add_fixture
+
+        latent_style = latent_styles.add_latent_style(name)
+
+        assert latent_styles.element.xml == expected_xml
+        assert isinstance(latent_style, _LatentStyle)
+        assert latent_style.element is latent_styles.element[0]
+
     def it_knows_how_many_latent_styles_it_contains(self, len_fixture):
         latent_styles, expected_value = len_fixture
         assert len(latent_styles) == expected_value
@@ -169,6 +178,13 @@ def it_can_change_its_boolean_properties(self, bool_prop_set_fixture):
 
     # fixtures -------------------------------------------------------
 
+    @pytest.fixture
+    def add_fixture(self):
+        latent_styles = LatentStyles(element('w:latentStyles'))
+        name = 'Foobar'
+        expected_xml = xml('w:latentStyles/w:lsdException{w:name=Foobar}')
+        return latent_styles, name, expected_xml
+
     @pytest.fixture(params=[
         ('w:latentStyles', 'default_to_hidden',           False),
         ('w:latentStyles', 'default_to_locked',           False),
