-

cool-RR · cool-RR · commit 06d10f05203f · 2016-06-07T14:15:04.000+03:00
diff --git a/source_py2/test_python_toolbox/test_temp_file_tools/test_create_temp_folder.py b/source_py2/test_python_toolbox/test_temp_file_tools/test_create_temp_folder.py
@@ -51,7 +51,25 @@ def test_basic():
     assert not file_path.exists()
     assert not file_path.is_file()
     
-
+def test_exception():
+    try:
+        with create_temp_folder() as tf1:
+            assert isinstance(tf1, pathlib.Path)
+            assert tf1.exists()
+            assert tf1.is_dir()
+            file_path = (tf1 / 'my_file')
+            with file_path.open('w') as my_file:
+                my_file.write('Woo hoo!')
+            
+            assert file_path.exists()
+            assert file_path.is_file()
+            raise MyException
+    except MyException:
+        assert not tf1.exists()
+        assert not tf1.is_dir()
+        assert not file_path.exists()
+        assert not file_path.is_file()
+            
 def test_without_pathlib():
     with create_temp_folder() as tf1:
         assert os.path.exists(str(tf1))
diff --git a/source_py3/test_python_toolbox/test_temp_file_tools/test_create_temp_folder.py b/source_py3/test_python_toolbox/test_temp_file_tools/test_create_temp_folder.py
@@ -17,6 +17,9 @@
 
 from python_toolbox.temp_file_tools import create_temp_folder
 
+class MyException(Exception):
+    pass
+        
 
 def test_basic():
     with create_temp_folder() as tf1:
@@ -51,7 +54,25 @@ def test_basic():
     assert not file_path.exists()
     assert not file_path.is_file()
     
-
+def test_exception():
+    try:
+        with create_temp_folder() as tf1:
+            assert isinstance(tf1, pathlib.Path)
+            assert tf1.exists()
+            assert tf1.is_dir()
+            file_path = (tf1 / 'my_file')
+            with file_path.open('w') as my_file:
+                my_file.write('Woo hoo!')
+            
+            assert file_path.exists()
+            assert file_path.is_file()
+            raise MyException
+    except MyException:
+        assert not tf1.exists()
+        assert not tf1.is_dir()
+        assert not file_path.exists()
+        assert not file_path.is_file()
+            
 def test_without_pathlib():
     with create_temp_folder() as tf1:
         assert os.path.exists(str(tf1))
