api: add _Document.save()

Steve Canny · Steve Canny · commit 554129256e76 · 2013-12-20T20:10:44.000-08:00
diff --git a/docx/api.py b/docx/api.py
@@ -47,10 +47,18 @@ class _Document(object):
     def __init__(self, pkg, document_part):
         super(_Document, self).__init__()
         self._document = document_part
+        self._pkg = pkg
 
     @property
     def body(self):
         """
         Return a reference to the |Body| instance for this document.
         """
         return self._document.body
+
+    def save(self, file_):
+        """
+        Save this document to *file_*, where *file_* can be either a path to
+        a file (a string) or a file-like object.
+        """
+        return self._pkg.save(file_)
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -81,3 +81,8 @@ def it_provides_access_to_the_document_body(self):
         body = doc.body
         # verify -----------------------
         assert body is document_part.body
+
+    def it_can_save_the_package(self):
+        pkg, file_ = (Mock(name='pkg'), Mock(name='file_'))
+        _Document(pkg, None).save(file_)
+        pkg.save.assert_called_once_with(file_)
