[Python-checkins] r79020 - python/branches/py3k/Lib/test/test_import.py

Wed Mar 17 04:14:31 CET 2010

Author: collin.winter
Date: Wed Mar 17 04:14:31 2010
New Revision: 79020
Log:
Add tests for overriding and shadowing __import__; these are a useful tripwire for an incoming JIT optimization.
Modified:
 python/branches/py3k/Lib/test/test_import.py
Modified: python/branches/py3k/Lib/test/test_import.py
==============================================================================

--- python/branches/py3k/Lib/test/test_import.py	(original)
+++ python/branches/py3k/Lib/test/test_import.py	Wed Mar 17 04:14:31 2010
@@ -1,3 +1,4 @@
+import builtins
 import imp
 import marshal
 import os
@@ -9,7 +10,7 @@
 import unittest
 import warnings
 from test.support import (unlink, TESTFN, unload, run_unittest,
- TestFailed, EnvironmentVarGuard)
+ TestFailed, EnvironmentVarGuard, swap_attr, swap_item)
 
 
 def remove_files(name):
@@ -446,8 +447,29 @@
 self.assertRaises(ValueError, check_relative)
 
 
+class OverridingImportBuiltinTests(unittest.TestCase):
+ def test_override_builtin(self):
+ # Test that overriding builtins.__import__ can bypass sys.modules.
+ import os
+
+ def foo():
+ import os
+ return os
+ self.assertEqual(foo(), os) # Quick sanity check.
+
+ with swap_attr(builtins, "__import__", lambda *x: 5):
+ self.assertEqual(foo(), 5)
+
+ # Test what happens when we shadow __import__ in globals(); this
+ # currently does not impact the import process, but if this changes,
+ # other code will need to change, so keep this test as a tripwire.
+ with swap_item(globals(), "__import__", lambda *x: 5):
+ self.assertEqual(foo(), os)
+
+
 def test_main(verbose=None):
- run_unittest(ImportTests, PycRewritingTests, PathsTests, RelativeImportTests)
+ run_unittest(ImportTests, PycRewritingTests, PathsTests, RelativeImportTests,
+ OverridingImportBuiltinTests)
 
 if __name__ == '__main__':
 # Test needs to be a package, so we can do relative imports.
