Merge "Introduce fileutils ensure_dir creation mode parameter"

1 files changed, 7 insertions, 2 deletions

diff --git a/tests/unit/test_fileutils.py b/tests/unit/test_fileutils.py
index f9e30ffe..2934776b 100644
--- a/tests/unit/test_fileutils.py
+++ b/tests/unit/test_fileutils.py
@@ -16,6 +16,7 @@
 import errno
 import os
 import shutil
+import stat
 import tempfile
 
 import mock
@@ -26,14 +27,18 @@ import six
 from openstack.common import fileutils
 
 
+TEST_PERMISSIONS = stat.S_IRWXU
+
+
 class EnsureTree(test_base.BaseTestCase):
     def test_ensure_tree(self):
         tmpdir = tempfile.mkdtemp()
         try:
             testdir = '%s/foo/bar/baz' % (tmpdir,)
-            fileutils.ensure_tree(testdir)
+            fileutils.ensure_tree(testdir, TEST_PERMISSIONS)
             self.assertTrue(os.path.isdir(testdir))
-
+            self.assertEqual(os.stat(testdir).st_mode,
+                             TEST_PERMISSIONS | stat.S_IFDIR)
         finally:
             if os.path.exists(tmpdir):
                 shutil.rmtree(tmpdir)