1 files changed, 18 insertions, 0 deletions
diff --git a/js/src/tests/lib/tempfile.py b/js/src/tests/lib/tempfile.py
new file mode 100644
index 0000000000..604864dce1
--- /dev/null
+++ b/js/src/tests/lib/tempfile.py
@@ -0,0 +1,18 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+try:
+    # Python 3.2
+    from tempfile import TemporaryDirectory
+except ImportError:
+    import shutil
+    import tempfile
+    from contextlib import contextmanager
+
+    @contextmanager
+    def TemporaryDirectory(*args, **kwds):
+        d = tempfile.mkdtemp(*args, **kwds)
+        try:
+            yield d
+        finally:
+            shutil.rmtree(d)