doc: add DocumentPart.save()

Steve Canny · Steve Canny · commit 6c9d5def1262 · 2015-02-15T02:16:58.000-08:00
diff --git a/docx/parts/document.py b/docx/parts/document.py
@@ -118,7 +118,7 @@ def save(self, path_or_stream):
         Save this document to *path_or_stream*, which can be either a path to
         a filesystem location (a string) or a file-like object.
         """
-        raise NotImplementedError
+        self.package.save(path_or_stream)
 
     @lazyproperty
     def sections(self):
diff --git a/tests/parts/test_document.py b/tests/parts/test_document.py
@@ -35,6 +35,11 @@
 
 class DescribeDocumentPart(object):
 
+    def it_can_save_the_package_to_a_file(self, save_fixture):
+        document, file_ = save_fixture
+        document.save(file_)
+        document._package.save.assert_called_once_with(file_)
+
     def it_has_a_body(self, body_fixture):
         document_part, _Body_, body_elm = body_fixture
         _body = document_part.body
@@ -206,6 +211,12 @@ def nmprt_get_fixture(self, part_related_by_, numbering_part_):
         part_related_by_.return_value = numbering_part_
         return document_part, numbering_part_
 
+    @pytest.fixture
+    def save_fixture(self, package_):
+        document_part = DocumentPart(None, None, None, package_)
+        file_ = 'foobar.docx'
+        return document_part, file_
+
     @pytest.fixture
     def sections_fixture(self, Sections_):
         document_elm = a_document().with_nsdecls().element
