set: configure loader for SettingsPart

Steve Canny · Steve Canny · commit f60fddd0f44d · 2015-03-15T22:49:28.000-07:00
diff --git a/docx/__init__.py b/docx/__init__.py
@@ -14,6 +14,7 @@
 from docx.parts.document import DocumentPart
 from docx.parts.image import ImagePart
 from docx.parts.numbering import NumberingPart
+from docx.parts.settings import SettingsPart
 from docx.parts.styles import StylesPart
 
 
@@ -27,6 +28,7 @@ def part_class_selector(content_type, reltype):
 PartFactory.part_type_for[CT.OPC_CORE_PROPERTIES] = CorePropertiesPart
 PartFactory.part_type_for[CT.WML_DOCUMENT_MAIN] = DocumentPart
 PartFactory.part_type_for[CT.WML_NUMBERING] = NumberingPart
+PartFactory.part_type_for[CT.WML_SETTINGS] = SettingsPart
 PartFactory.part_type_for[CT.WML_STYLES] = StylesPart
 
 del (
diff --git a/tests/parts/test_settings.py b/tests/parts/test_settings.py
@@ -10,15 +10,26 @@
 
 import pytest
 
+from docx.opc.constants import CONTENT_TYPE as CT
+from docx.opc.part import PartFactory
+from docx.package import Package
 from docx.parts.settings import SettingsPart
 from docx.settings import Settings
 
 from ..unitutil.cxml import element
-from ..unitutil.mock import class_mock, instance_mock
+from ..unitutil.mock import class_mock, instance_mock, method_mock
 
 
 class DescribeSettingsPart(object):
 
+    def it_is_used_by_loader_to_construct_settings_part(self, load_fixture):
+        partname, content_type, blob, package_, settings_part_ = load_fixture
+        part = PartFactory(partname, content_type, None, blob, package_)
+        SettingsPart.load.assert_called_once_with(
+            partname, content_type, blob, package_
+        )
+        assert part is settings_part_
+
     def it_provides_access_to_its_settings(self, settings_fixture):
         settings_part, Settings_, settings_ = settings_fixture
         settings = settings_part.settings
@@ -27,6 +38,13 @@ def it_provides_access_to_its_settings(self, settings_fixture):
 
     # fixtures -------------------------------------------------------
 
+    @pytest.fixture
+    def load_fixture(self, load_, package_, settings_part_):
+        partname, blob = 'partname', 'blob'
+        content_type = CT.WML_SETTINGS
+        load_.return_value = settings_part_
+        return partname, content_type, blob, package_, settings_part_
+
     @pytest.fixture
     def settings_fixture(self, Settings_, settings_):
         settings_elm = element('w:settings')
@@ -35,6 +53,14 @@ def settings_fixture(self, Settings_, settings_):
 
     # fixture components ---------------------------------------------
 
+    @pytest.fixture
+    def load_(self, request):
+        return method_mock(request, SettingsPart, 'load')
+
+    @pytest.fixture
+    def package_(self, request):
+        return instance_mock(request, Package)
+
     @pytest.fixture
     def Settings_(self, request, settings_):
         return class_mock(
@@ -44,3 +70,7 @@ def Settings_(self, request, settings_):
     @pytest.fixture
     def settings_(self, request):
         return instance_mock(request, Settings)
+
+    @pytest.fixture
+    def settings_part_(self, request):
+        return instance_mock(request, SettingsPart)
