doc: add PartFactory call test for DocumentPart

Steve Canny · Steve Canny · commit 64d6a55ba8d4 · 2013-12-28T21:31:02.000-08:00
diff --git a/tests/opc/test_package.py b/tests/opc/test_package.py
@@ -460,14 +460,6 @@ def CustomPartClass_(self, request, part_of_custom_type_):
         CustomPartClass_.load.return_value = part_of_custom_type_
         return CustomPartClass_
 
-    @pytest.fixture
-    def part_of_custom_type_(self, request):
-        return instance_mock(request, Part)
-
-    @pytest.fixture
-    def part_of_default_type_(self, request):
-        return instance_mock(request, Part)
-
     @pytest.fixture
     def DefaultPartClass_(self, request, part_of_default_type_):
         DefaultPartClass_ = cls_attr_mock(
@@ -484,6 +476,14 @@ def package_(self, request):
     def package_2_(self, request):
         return instance_mock(request, OpcPackage)
 
+    @pytest.fixture
+    def part_of_custom_type_(self, request):
+        return instance_mock(request, Part)
+
+    @pytest.fixture
+    def part_of_default_type_(self, request):
+        return instance_mock(request, Part)
+
     @pytest.fixture
     def partname_(self, request):
         return instance_mock(request, PackURI)
diff --git a/tests/parts/test_document.py b/tests/parts/test_document.py
@@ -11,7 +11,9 @@
 from mock import Mock
 
 from docx.enum.shape import WD_INLINE_SHAPE
-from docx.opc.constants import RELATIONSHIP_TYPE as RT
+from docx.opc.constants import CONTENT_TYPE as CT, RELATIONSHIP_TYPE as RT
+from docx.opc.package import PartFactory
+from docx.opc.packuri import PackURI
 from docx.oxml.parts.document import CT_Body, CT_Document
 from docx.oxml.shared import nsmap
 from docx.oxml.text import CT_R
@@ -39,7 +41,23 @@
 )
 
 
-class Describe_Document(object):
+class DescribeDocumentPart(object):
+
+    def it_is_used_by_PartFactory_to_construct_main_document_part(
+            self, part_load_fixture):
+        # fixture ----------------------
+        document_part_load_, partname_, blob_, package_, document_part_ = (
+            part_load_fixture
+        )
+        content_type = CT.WML_DOCUMENT_MAIN
+        reltype = RT.OFFICE_DOCUMENT
+        # exercise ---------------------
+        part = PartFactory(partname_, content_type, reltype, blob_, package_)
+        # verify -----------------------
+        document_part_load_.assert_called_once_with(
+            partname_, content_type, blob_, package_
+        )
+        assert part is document_part_
 
     def it_can_be_constructed_by_opc_part_factory(
             self, oxml_fromstring_, init):
@@ -99,6 +117,18 @@ def it_knows_the_next_available_xml_id(self, next_id_fixture):
 
     # fixtures -------------------------------------------------------
 
+    @pytest.fixture
+    def _Body_(self, request):
+        return class_mock(request, 'docx.parts.document._Body')
+
+    @pytest.fixture
+    def blob_(self, request):
+        return instance_mock(request, str)
+
+    @pytest.fixture
+    def content_type_(self, request):
+        return instance_mock(request, str)
+
     @pytest.fixture
     def document(self):
         return DocumentPart(None, None, None, None)
@@ -120,13 +150,18 @@ def document_body_fixture(self, request, _Body_):
         return document, _Body_, body_elm
 
     @pytest.fixture
-    def _Body_(self, request):
-        return class_mock(request, 'docx.parts.document._Body')
+    def document_part_(self, request):
+        return instance_mock(request, DocumentPart)
+
+    @pytest.fixture
+    def document_part_load_(self, request):
+        return method_mock(request, DocumentPart, 'load')
 
     @pytest.fixture
     def get_or_add_image_fixture(
             self, request, package_, image_descriptor_, image_parts_,
             relate_to_, image_part_, rId_):
+        package_.image_parts = image_parts_
         document = DocumentPart(None, None, None, package_)
         return (
             document, image_descriptor_, image_parts_, relate_to_,
@@ -185,9 +220,20 @@ def oxml_fromstring_(self, request):
 
     @pytest.fixture
     def package_(self, request, image_parts_):
-        package_ = instance_mock(request, Package)
-        package_.image_parts = image_parts_
-        return package_
+        return instance_mock(request, Package)
+
+    @pytest.fixture
+    def part_load_fixture(
+            self, document_part_load_, partname_, blob_, package_,
+            document_part_):
+        document_part_load_.return_value = document_part_
+        return (
+            document_part_load_, partname_, blob_, package_, document_part_
+        )
+
+    @pytest.fixture
+    def partname_(self, request):
+        return instance_mock(request, PackURI)
 
     @pytest.fixture
     def relate_to_(self, request, rId_):
