style: add BaseStyle.locked setter

Steve Canny · Steve Canny · commit 67cfed60b3cb · 2015-02-08T00:01:46.000-08:00
diff --git a/docx/oxml/styles.py b/docx/oxml/styles.py
@@ -105,6 +105,13 @@ def locked_val(self):
             return False
         return locked.val
 
+    @locked_val.setter
+    def locked_val(self, value):
+        self._remove_locked()
+        if bool(value) is True:
+            locked = self._add_locked()
+            locked.val = value
+
     @property
     def name_val(self):
         """
diff --git a/docx/styles/style.py b/docx/styles/style.py
@@ -84,6 +84,10 @@ def locked(self):
         """
         return self._element.locked_val
 
+    @locked.setter
+    def locked(self, value):
+        self._element.locked_val = value
+
     @property
     def name(self):
         """
diff --git a/features/sty-style-props.feature b/features/sty-style-props.feature
@@ -59,7 +59,6 @@ Feature: Get and set style properties
       | no setting | False |
 
 
-  @wip
   Scenario Outline: Set locked value
     Given a style having locked set <setting>
      When I assign <new-value> to style.locked
diff --git a/tests/styles/test_style.py b/tests/styles/test_style.py
@@ -163,6 +163,11 @@ def it_knows_whether_its_locked(self, locked_get_fixture):
         style, expected_value = locked_get_fixture
         assert style.locked == expected_value
 
+    def it_can_change_whether_its_locked(self, locked_set_fixture):
+        style, value, expected_xml = locked_set_fixture
+        style.locked = 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()
@@ -245,6 +250,20 @@ def locked_get_fixture(self, request):
         style = BaseStyle(element(style_cxml))
         return style, expected_value
 
+    @pytest.fixture(params=[
+        ('w:style',                   True,  'w:style/w:locked'),
+        ('w:style/w:locked{w:val=0}', True,  'w:style/w:locked'),
+        ('w:style/w:locked{w:val=1}', True,  'w:style/w:locked'),
+        ('w:style',                   False, 'w:style'),
+        ('w:style/w:locked',          False, 'w:style'),
+        ('w:style/w:locked{w:val=1}', False, 'w:style'),
+    ])
+    def locked_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{w:type=table}',                         None),
         ('w:style{w:type=table}/w:name{w:val=Boofar}',    'Boofar'),
