From: Nick Coghlan <ncoghlan@gmail.com>
Date: Sat, 23 Jun 2012 10:07:39 +0000 (+1000)
Subject: Properly test the various builtins lookup cases in inspect.getclosurevars
X-Git-Tag: v3.3.0b1~117
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=6c6e2545cb6904f78754d799d4cfb180f7a626d9;p=thirdparty%2FPython%2Fcpython.git

Properly test the various builtins lookup cases in inspect.getclosurevars
---

diff --git a/Lib/test/test_inspect.py b/Lib/test/test_inspect.py
index 9d971e055e3f..7ed4ffef8c64 100644
--- a/Lib/test/test_inspect.py
+++ b/Lib/test/test_inspect.py
@@ -763,6 +763,30 @@ class TestGetClosureVars(unittest.TestCase):
         self.assertRaises(TypeError, inspect.getclosurevars, list)
         self.assertRaises(TypeError, inspect.getclosurevars, {})
 
+    def _private_globals(self):
+        code = """def f(): print(path)"""
+        ns = {}
+        exec(code, ns)
+        return ns["f"], ns
+
+    def test_builtins_fallback(self):
+        f, ns = self._private_globals()
+        ns.pop("__builtins__", None)
+        expected = inspect.ClosureVars({}, {}, {"print":print}, {"path"})
+        self.assertEqual(inspect.getclosurevars(f), expected)
+
+    def test_builtins_as_dict(self):
+        f, ns = self._private_globals()
+        ns["__builtins__"] = {"path":1}
+        expected = inspect.ClosureVars({}, {}, {"path":1}, {"print"})
+        self.assertEqual(inspect.getclosurevars(f), expected)
+
+    def test_builtins_as_module(self):
+        f, ns = self._private_globals()
+        ns["__builtins__"] = os
+        expected = inspect.ClosureVars({}, {}, {"path":os.path}, {"print"})
+        self.assertEqual(inspect.getclosurevars(f), expected)
+
 
 class TestGetcallargsFunctions(unittest.TestCase):