hdr: add BaseStoryPart.get_style()

Author: scanny

docx/parts/story.py

@@ -16,6 +16,14 @@ class BaseStoryPart(XmlPart):
     `.add_paragraph()`, `.add_table()` etc.
     """
 
+    def get_style(self, style_id, style_type):
+        """Return the style in this document matching *style_id*.
+
+        Returns the default style for *style_type* if *style_id* is |None| or does not
+        match a defined style of *style_type*.
+        """
+        return self._document_part.get_style(style_id, style_type)
+
     def get_style_id(self, style_or_name, style_type):
         """Return str style_id for *style_or_name* of *style_type*.
 

features/hdr-header-footer.feature

@@ -34,7 +34,6 @@ Feature: Header and footer behaviors
       And header_2.is_linked_to_previous is True
 
 
-  @wip
   Scenario: _Header text accepts style assignment
     Given a _Header object with a header definition as header
      When I assign "Normal" to header.paragraphs[0].style
@@ -78,7 +77,6 @@ Feature: Header and footer behaviors
       And footer_2.is_linked_to_previous is True
 
 
-  @wip
   Scenario: _Footer text accepts style assignment
     Given a _Footer object with a footer definition as footer
      When I assign "Normal" to footer.paragraphs[0].style

tests/parts/test_story.py

@@ -17,6 +17,20 @@
 
 class DescribeBaseStoryPart(object):
 
+    def it_can_get_a_style_by_id_and_type(
+        self, _document_part_prop_, document_part_, style_
+    ):
+        style_id = "BodyText"
+        style_type = WD_STYLE_TYPE.PARAGRAPH
+        _document_part_prop_.return_value = document_part_
+        document_part_.get_style.return_value = style_
+        story_part = BaseStoryPart(None, None, None, None)
+
+        style = story_part.get_style(style_id, style_type)
+
+        document_part_.get_style.assert_called_once_with(style_id, style_type)
+        assert style is style_
+
     def it_can_get_a_style_id_by_style_or_name_and_type(
         self, _document_part_prop_, document_part_, style_
     ):