api: Document loads default template

Steve Canny · Steve Canny · commit 12eb4625e91a · 2013-12-20T20:10:43.000-08:00
diff --git a/docx/api.py b/docx/api.py
@@ -13,16 +13,24 @@
 opc.OpcPackage graph.
 """
 
+import os
+
 from opc import OpcPackage
 
 
+thisdir = os.path.split(__file__)[0]
+_default_docx_path = os.path.join(thisdir, 'templates', 'default.docx')
+
+
 def Document(docx=None):
     """
     Return a |_Document| instance loaded from *docx*, where *docx* can be
     either a path to a ``.docx`` file (a string) or a file-like object. If
     *docx* is missing or ``None``, the built-in default document "template"
     is loaded.
     """
+    if docx is None:
+        docx = _default_docx_path
     pkg = OpcPackage.open(docx)
     document_part = pkg.main_document
     return _Document(pkg, document_part)
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -15,7 +15,7 @@
 
 from docx.api import Document, _Document
 
-from .unitutil import class_mock
+from .unitutil import class_mock, var_mock
 
 
 class DescribeDocument(object):
@@ -24,6 +24,14 @@ class DescribeDocument(object):
     def _Document_(self, request):
         return class_mock('docx.api._Document', request)
 
+    @pytest.fixture
+    def default_docx(self, request):
+        return var_mock('docx.api._default_docx_path', request)
+
+    @pytest.fixture
+    def OpcPackage_(self, OpcPackage_mockery):
+        return OpcPackage_mockery[0]
+
     @pytest.fixture
     def OpcPackage_mockery(self, request):
         OpcPackage_ = class_mock('docx.api.OpcPackage', request)
@@ -45,3 +53,8 @@ def it_opens_a_docx_file_on_construction(self, OpcPackage_mockery,
         main_document.assert_called_once_with()
         _Document_.assert_called_once_with(pkg, document_part)
         assert isinstance(doc, _Document)
+
+    def it_uses_default_if_no_file_provided(self, OpcPackage_, _Document_,
+                                            default_docx):
+        Document()
+        OpcPackage_.open.assert_called_once_with(default_docx)
