220 lines
7.8 KiB
Python
220 lines
7.8 KiB
Python
import os
|
|
import sys
|
|
import unittest
|
|
from nose.importer import Importer
|
|
from nose.plugins.skip import SkipTest
|
|
|
|
|
|
class TestImporter(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.dir = os.path.normpath(os.path.join(os.path.dirname(__file__),
|
|
'support'))
|
|
self.imp = Importer()
|
|
self._mods = sys.modules.copy()
|
|
self._path = sys.path[:]
|
|
sys.modules.pop('mod', None)
|
|
sys.modules.pop('pak', None)
|
|
sys.modules.pop('pak.mod', None)
|
|
sys.modules.pop('pak.sub', None)
|
|
try:
|
|
os.symlink(
|
|
os.path.abspath(os.path.join(self.dir, 'dir1', 'pak')),
|
|
os.path.join(self.dir, 'dir3', 'pak'))
|
|
except (AttributeError, NotImplementedError):
|
|
self.has_symlinks = False
|
|
else:
|
|
self.has_symlinks = True
|
|
|
|
def tearDown(self):
|
|
to_del = [ m for m in sys.modules.keys() if
|
|
m not in self._mods ]
|
|
if to_del:
|
|
for mod in to_del:
|
|
del sys.modules[mod]
|
|
sys.modules.update(self._mods)
|
|
sys.path = self._path[:]
|
|
if self.has_symlinks:
|
|
os.unlink(os.path.join(self.dir, 'dir3', 'pak'))
|
|
|
|
def test_import_from_dir(self):
|
|
imp = self.imp
|
|
|
|
d1 = os.path.join(self.dir, 'dir1')
|
|
d2 = os.path.join(self.dir, 'dir2')
|
|
|
|
# simple name
|
|
m1 = imp.importFromDir(d1, 'mod')
|
|
m2 = imp.importFromDir(d2, 'mod')
|
|
self.assertNotEqual(m1, m2)
|
|
self.assertNotEqual(m1.__file__, m2.__file__)
|
|
|
|
# dotted name
|
|
p1 = imp.importFromDir(d1, 'pak.mod')
|
|
p2 = imp.importFromDir(d2, 'pak.mod')
|
|
self.assertNotEqual(p1, p2)
|
|
self.assertNotEqual(p1.__file__, p2.__file__)
|
|
|
|
def test_import_from_dirlink(self):
|
|
if not self.has_symlinks:
|
|
raise SkipTest("symlinks not available")
|
|
imp = self.imp
|
|
|
|
d1 = os.path.join(self.dir, 'dir3')
|
|
d2 = os.path.join(self.dir, 'dir2')
|
|
|
|
# simple name
|
|
m1 = imp.importFromDir(d1, 'pak')
|
|
m2 = imp.importFromDir(d2, 'pak')
|
|
self.assertNotEqual(m1, m2)
|
|
self.assertNotEqual(m1.__file__, m2.__file__)
|
|
|
|
# dotted name
|
|
p1 = imp.importFromDir(d1, 'pak.mod')
|
|
p2 = imp.importFromDir(d2, 'pak.mod')
|
|
self.assertNotEqual(p1, p2)
|
|
self.assertNotEqual(p1.__file__, p2.__file__)
|
|
|
|
def test_import_from_path(self):
|
|
imp = self.imp
|
|
|
|
jn = os.path.join
|
|
d1 = jn(self.dir, 'dir1')
|
|
d2 = jn(self.dir, 'dir2')
|
|
|
|
# simple name
|
|
m1 = imp.importFromPath(jn(d1, 'mod.py'), 'mod')
|
|
m2 = imp.importFromPath(jn(d2, 'mod.py'), 'mod')
|
|
self.assertNotEqual(m1, m2)
|
|
self.assertNotEqual(m1.__file__, m2.__file__)
|
|
|
|
# dotted name
|
|
p1 = imp.importFromPath(jn(d1, 'pak', 'mod.py'), 'pak.mod')
|
|
p2 = imp.importFromPath(jn(d2, 'pak', 'mod.py'), 'pak.mod')
|
|
self.assertNotEqual(p1, p2)
|
|
self.assertNotEqual(p1.__file__, p2.__file__)
|
|
|
|
# simple name -- package
|
|
sp1 = imp.importFromPath(jn(d1, 'pak'), 'pak')
|
|
sp2 = imp.importFromPath(jn(d2, 'pak'), 'pak')
|
|
self.assertNotEqual(sp1, sp2)
|
|
assert sp1.__path__
|
|
assert sp2.__path__
|
|
self.assertNotEqual(sp1.__path__, sp2.__path__)
|
|
|
|
# dotted name -- package
|
|
dp1 = imp.importFromPath(jn(d1, 'pak', 'sub'), 'pak.sub')
|
|
dp2 = imp.importFromPath(jn(d2, 'pak', 'sub'), 'pak.sub')
|
|
self.assertNotEqual(dp1, dp2)
|
|
assert dp1.__path__
|
|
assert dp2.__path__
|
|
self.assertNotEqual(dp1.__path__, dp2.__path__)
|
|
|
|
def test_import_sets_intermediate_modules(self):
|
|
imp = self.imp
|
|
path = os.path.join(self.dir,
|
|
'package2', 'test_pak', 'test_sub', 'test_mod.py')
|
|
mod = imp.importFromPath(path, 'test_pak.test_sub.test_mod')
|
|
print mod, dir(mod)
|
|
assert 'test_pak' in sys.modules, 'test_pak was not imported?'
|
|
test_pak = sys.modules['test_pak']
|
|
assert hasattr(test_pak, 'test_sub'), "test_pak.test_sub was not set"
|
|
|
|
def test_cached_no_reload(self):
|
|
imp = self.imp
|
|
d1 = os.path.join(self.dir, 'dir1')
|
|
m1 = imp.importFromDir(d1, 'mod')
|
|
m2 = imp.importFromDir(d1, 'mod')
|
|
assert m1 is m2, "%s is not %s" % (m1, m2)
|
|
|
|
def test_cached_no_reload_dotted(self):
|
|
imp = self.imp
|
|
d1 = os.path.join(self.dir, 'dir1')
|
|
p1 = imp.importFromDir(d1, 'pak.mod')
|
|
p2 = imp.importFromDir(d1, 'pak.mod')
|
|
assert p1 is p2, "%s is not %s" % (p1, p2)
|
|
|
|
def test_import_sets_sys_modules(self):
|
|
imp = self.imp
|
|
d1 = os.path.join(self.dir, 'dir1')
|
|
p1 = imp.importFromDir(d1, 'pak.mod')
|
|
assert sys.modules['pak.mod'] is p1, "pak.mod not in sys.modules"
|
|
assert sys.modules['pak'], "pak not in sys.modules"
|
|
assert sys.modules['pak'].mod is p1, \
|
|
"sys.modules['pak'].mod is not the module we loaded"
|
|
|
|
def test_failed_import_raises_import_error(self):
|
|
imp = self.imp
|
|
def bad_import():
|
|
imp.importFromPath(self.dir, 'no.such.module')
|
|
self.assertRaises(ImportError, bad_import)
|
|
|
|
def test_sys_modules_same_path_no_reload(self):
|
|
imp = self.imp
|
|
|
|
d1 = os.path.join(self.dir, 'dir1')
|
|
d2 = os.path.join(self.dir, 'dir2')
|
|
sys.path.insert(0, d1)
|
|
mod_sys_imported = __import__('mod')
|
|
mod_nose_imported = imp.importFromDir(d1, 'mod')
|
|
assert mod_nose_imported is mod_sys_imported, \
|
|
"nose reimported a module in sys.modules from the same path"
|
|
|
|
mod_nose_imported2 = imp.importFromDir(d2, 'mod')
|
|
assert mod_nose_imported2 != mod_sys_imported, \
|
|
"nose failed to reimport same name, different dir"
|
|
|
|
def test_sys_modules_symlinked_package_no_reload(self):
|
|
if not self.has_symlinks:
|
|
raise SkipTest("symlinks not available")
|
|
imp = self.imp
|
|
|
|
d1 = os.path.join(self.dir, 'dir1')
|
|
d2 = os.path.join(self.dir, 'dir3')
|
|
sys.path.insert(0, d1)
|
|
# Symlinked package
|
|
mod_sys_imported = __import__('pak')
|
|
mod_nose_imported = imp.importFromDir(d2, 'pak')
|
|
assert mod_nose_imported is mod_sys_imported, \
|
|
"nose reimported a module in sys.modules from the same file"
|
|
|
|
# Module inside symlinked package
|
|
mod_sys_imported = __import__('pak.mod', fromlist=['mod'])
|
|
mod_nose_imported = imp.importFromDir(d2, 'pak.mod')
|
|
assert mod_nose_imported is mod_sys_imported, \
|
|
("nose reimported a module in sys.modules from the same file",
|
|
mod_sys_imported.__file__, mod_nose_imported.__file__)
|
|
|
|
def test_import_pkg_from_path_fpw(self):
|
|
imp = self.imp
|
|
imp.config.firstPackageWins = True
|
|
jn = os.path.join
|
|
d1 = jn(self.dir, 'dir1')
|
|
d2 = jn(self.dir, 'dir2')
|
|
|
|
# dotted name
|
|
p1 = imp.importFromPath(jn(d1, 'pak', 'mod.py'), 'pak.mod')
|
|
p2 = imp.importFromPath(jn(d2, 'pak', 'mod.py'), 'pak.mod')
|
|
self.assertEqual(p1, p2)
|
|
self.assertEqual(p1.__file__, p2.__file__)
|
|
|
|
# simple name -- package
|
|
sp1 = imp.importFromPath(jn(d1, 'pak'), 'pak')
|
|
sp2 = imp.importFromPath(jn(d2, 'pak'), 'pak')
|
|
self.assertEqual(sp1, sp2)
|
|
assert sp1.__path__
|
|
assert sp2.__path__
|
|
self.assertEqual(sp1.__path__, sp2.__path__)
|
|
|
|
# dotted name -- package
|
|
dp1 = imp.importFromPath(jn(d1, 'pak', 'sub'), 'pak.sub')
|
|
dp2 = imp.importFromPath(jn(d2, 'pak', 'sub'), 'pak.sub')
|
|
self.assertEqual(dp1, dp2)
|
|
assert dp1.__path__
|
|
assert dp2.__path__
|
|
self.assertEqual(dp1.__path__, dp2.__path__)
|
|
|
|
if __name__ == '__main__':
|
|
import logging
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
unittest.main()
|