Index: third_party/mojo/src/mojo/public/tools/bindings/pylib/mojom_tests/fileutil_unittest.py |
diff --git a/third_party/mojo/src/mojo/public/tools/bindings/pylib/mojom_tests/fileutil_unittest.py b/third_party/mojo/src/mojo/public/tools/bindings/pylib/mojom_tests/fileutil_unittest.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..d56faadb196c95b8f7a860a0c51bf50cc87a7d91 |
--- /dev/null |
+++ b/third_party/mojo/src/mojo/public/tools/bindings/pylib/mojom_tests/fileutil_unittest.py |
@@ -0,0 +1,55 @@ |
+# Copyright 2015 The Chromium Authors. All rights reserved. |
+# Use of this source code is governed by a BSD-style license that can be |
+# found in the LICENSE file. |
+ |
+import imp |
+import os.path |
+import shutil |
+import sys |
+import tempfile |
+import unittest |
+ |
+def _GetDirAbove(dirname): |
+ """Returns the directory "above" this file containing |dirname| (which must |
+ also be "above" this file).""" |
+ path = os.path.abspath(__file__) |
+ while True: |
+ path, tail = os.path.split(path) |
+ assert tail |
+ if tail == dirname: |
+ return path |
+ |
+try: |
+ imp.find_module("mojom") |
+except ImportError: |
+ sys.path.append(os.path.join(_GetDirAbove("pylib"), "pylib")) |
+from mojom import fileutil |
+ |
+ |
+class FileUtilTest(unittest.TestCase): |
+ |
+ def testEnsureDirectoryExists(self): |
+ """Test that EnsureDirectoryExists fuctions correctly.""" |
+ |
+ temp_dir = tempfile.mkdtemp() |
+ try: |
+ self.assertTrue(os.path.exists(temp_dir)) |
+ |
+ # Directory does not exist, yet. |
+ full = os.path.join(temp_dir, "foo", "bar") |
+ self.assertFalse(os.path.exists(full)) |
+ |
+ # Create the directory. |
+ fileutil.EnsureDirectoryExists(full) |
+ self.assertTrue(os.path.exists(full)) |
+ |
+ # Trying to create it again does not cause an error. |
+ fileutil.EnsureDirectoryExists(full) |
+ self.assertTrue(os.path.exists(full)) |
+ |
+ # Bypass check for directory existence to tickle error handling that |
+ # occurs in response to a race. |
+ fileutil.EnsureDirectoryExists(full, always_try_to_create=True) |
+ self.assertTrue(os.path.exists(full)) |
+ finally: |
+ shutil.rmtree(temp_dir) |