- [feature] Added support for context managers,

  i.e. "% with x as e:/ % endwith" support.
  Courtesy Ben Trofatter [ticket:147]

1 files changed, 16 insertions, 0 deletions

diff --git a/test/test_ast.py b/test/test_ast.py
index adea08a..60ad6ec 100644
--- a/test/test_ast.py
+++ b/test/test_ast.py
@@ -189,6 +189,22 @@ def x(q):
         eq_(parsed.declared_identifiers, set(['x']))
         eq_(parsed.undeclared_identifiers, set())
 
+    def test_locate_identifiers_12(self):
+        code = """
+class ContextManager(object):
+    def __enter__(self):
+        return 1
+    def __exit__(self, exc_type, exc_value, traceback):
+        pass
+
+with ContextManager() as x, ContextManager():
+    print x
+"""
+        parsed = ast.PythonCode(code, **exception_kwargs)
+        eq_(parsed.declared_identifiers, set(['ContextManager', 'x']))
+        eq_(parsed.undeclared_identifiers, set())
+        
+
     def test_no_global_imports(self):
         code = """
 from foo import *