-

cool-RR · cool-RR · commit 3ce93b2857c4 · 2015-12-17T20:54:33.000+02:00
diff --git a/source_py3/test_python_toolbox/test_context_management/test_as_reentrant.py b/source_py3/test_python_toolbox/test_context_management/test_as_reentrant.py
@@ -147,4 +147,32 @@ def __exit__(self, exc_type, exc_value, exc_traceback):
                                       
     # `.__wrapped__.__enter__` saw a depth of 1, because the depth decrement
     # happens *after* `.__wrapped__.__enter__` is called:
-    assert depth_log.get(block=False) == 1
+    assert depth_log.get(block=False) == 1
+    
+def test_decorator():
+    class Meow(ContextManager):
+        n = 0
+            
+        @as_reentrant
+        def manage_context(self):
+            self.n += 1
+            try:
+                yield
+            finally:
+                self.n -= 1
+                
+            
+    meow = Meow()
+    assert meow.n == 0
+    with meow:
+        assert meow.n == 1
+        with meow:
+            assert meow.n == 1
+            with meow:
+                assert meow.n == 1
+            assert meow.n == 1
+        assert meow.n == 1
+    assert meow.n == 0
+        
+        
+        
