style: add BaseStyle.priority setter

Steve Canny · Steve Canny · commit 872a0f2cdf4d · 2015-02-08T00:01:46.000-08:00
diff --git a/docx/oxml/styles.py b/docx/oxml/styles.py
@@ -136,6 +136,13 @@ def uiPriority_val(self):
             return None
         return uiPriority.val
 
+    @uiPriority_val.setter
+    def uiPriority_val(self, value):
+        self._remove_uiPriority()
+        if value is not None:
+            uiPriority = self._add_uiPriority()
+            uiPriority.val = value
+
 
 class CT_Styles(BaseOxmlElement):
     """
diff --git a/docx/styles/style.py b/docx/styles/style.py
@@ -97,6 +97,10 @@ def priority(self):
         """
         return self._element.uiPriority_val
 
+    @priority.setter
+    def priority(self, value):
+        self._element.uiPriority_val = value
+
     @property
     def style_id(self):
         """
diff --git a/features/sty-style-props.feature b/features/sty-style-props.feature
@@ -69,7 +69,6 @@ Feature: Get and set style properties
       | 42         | 42    |
 
 
-  @wip
   Scenario Outline: Set style display sort order
     Given a style having priority of <setting>
      When I assign <new-value> to style.priority
diff --git a/tests/styles/test_style.py b/tests/styles/test_style.py
@@ -136,6 +136,11 @@ def it_knows_its_sort_order(self, priority_get_fixture):
         style, expected_value = priority_get_fixture
         assert style.priority == expected_value
 
+    def it_can_change_its_sort_order(self, priority_set_fixture):
+        style, value, expected_xml = priority_set_fixture
+        style.priority = value
+        assert style._element.xml == expected_xml
+
     def it_can_delete_itself_from_the_document(self, delete_fixture):
         style, styles, expected_xml = delete_fixture
         style.delete()
@@ -237,6 +242,20 @@ def priority_get_fixture(self, request):
         style = BaseStyle(element(style_cxml))
         return style, expected_value
 
+    @pytest.fixture(params=[
+        ('w:style',                        42,
+         'w:style/w:uiPriority{w:val=42}'),
+        ('w:style/w:uiPriority{w:val=42}', 24,
+         'w:style/w:uiPriority{w:val=24}'),
+        ('w:style/w:uiPriority{w:val=24}', None,
+         'w:style'),
+    ])
+    def priority_set_fixture(self, request):
+        style_cxml, value, expected_cxml = request.param
+        style = BaseStyle(element(style_cxml))
+        expected_xml = xml(expected_cxml)
+        return style, value, expected_xml
+
     @pytest.fixture(params=[
         ('w:style',                   WD_STYLE_TYPE.PARAGRAPH),
         ('w:style{w:type=paragraph}', WD_STYLE_TYPE.PARAGRAPH),
