test the tempdir cm

1 files changed, 25 insertions, 0 deletions

diff --git a/test/unittest_context.py b/test/unittest_context.py
new file mode 100644
index 0000000..418c5e4
--- /dev/null
+++ b/test/unittest_context.py
@@ -0,0 +1,25 @@
+from __future__ import with_statement
+
+import unittest
+from os.path import isdir, exists
+
+from logilab.common.testlib import TestCase
+from logilab.common.context import tempdir
+
+class ContextTC(TestCase):
+
+    def test_withtempdir(self):
+        with tempdir() as tmpdir:
+            assert exists(tmpdir)
+            assert isdir(tmpdir)
+        assert not exists(tmpdir)
+        try:
+            with tempdir() as tmpdir:
+                assert exists(tmpdir)
+                raise Exception
+        except:
+            pass
+        assert not exists(tmpdir)
+
+if __name__ == '__main__':
+    unittest_main()