mirror of
https://github.com/python/cpython.git
synced 2025-11-04 11:49:12 +00:00
238 lines
8 KiB
Python
238 lines
8 KiB
Python
from test.support import run_unittest, unload
|
|
import unittest
|
|
import sys
|
|
import imp
|
|
import pkgutil
|
|
import os
|
|
import os.path
|
|
import tempfile
|
|
import shutil
|
|
import zipfile
|
|
|
|
|
|
|
|
class PkgutilTests(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.dirname = tempfile.mkdtemp()
|
|
self.addCleanup(shutil.rmtree, self.dirname)
|
|
sys.path.insert(0, self.dirname)
|
|
|
|
def tearDown(self):
|
|
del sys.path[0]
|
|
|
|
def test_getdata_filesys(self):
|
|
pkg = 'test_getdata_filesys'
|
|
|
|
# Include a LF and a CRLF, to test that binary data is read back
|
|
RESOURCE_DATA = b'Hello, world!\nSecond line\r\nThird line'
|
|
|
|
# Make a package with some resources
|
|
package_dir = os.path.join(self.dirname, pkg)
|
|
os.mkdir(package_dir)
|
|
# Empty init.py
|
|
f = open(os.path.join(package_dir, '__init__.py'), "wb")
|
|
f.close()
|
|
# Resource files, res.txt, sub/res.txt
|
|
f = open(os.path.join(package_dir, 'res.txt'), "wb")
|
|
f.write(RESOURCE_DATA)
|
|
f.close()
|
|
os.mkdir(os.path.join(package_dir, 'sub'))
|
|
f = open(os.path.join(package_dir, 'sub', 'res.txt'), "wb")
|
|
f.write(RESOURCE_DATA)
|
|
f.close()
|
|
|
|
# Check we can read the resources
|
|
res1 = pkgutil.get_data(pkg, 'res.txt')
|
|
self.assertEqual(res1, RESOURCE_DATA)
|
|
res2 = pkgutil.get_data(pkg, 'sub/res.txt')
|
|
self.assertEqual(res2, RESOURCE_DATA)
|
|
|
|
del sys.modules[pkg]
|
|
|
|
def test_getdata_zipfile(self):
|
|
zip = 'test_getdata_zipfile.zip'
|
|
pkg = 'test_getdata_zipfile'
|
|
|
|
# Include a LF and a CRLF, to test that binary data is read back
|
|
RESOURCE_DATA = b'Hello, world!\nSecond line\r\nThird line'
|
|
|
|
# Make a package with some resources
|
|
zip_file = os.path.join(self.dirname, zip)
|
|
z = zipfile.ZipFile(zip_file, 'w')
|
|
|
|
# Empty init.py
|
|
z.writestr(pkg + '/__init__.py', "")
|
|
# Resource files, res.txt, sub/res.txt
|
|
z.writestr(pkg + '/res.txt', RESOURCE_DATA)
|
|
z.writestr(pkg + '/sub/res.txt', RESOURCE_DATA)
|
|
z.close()
|
|
|
|
# Check we can read the resources
|
|
sys.path.insert(0, zip_file)
|
|
res1 = pkgutil.get_data(pkg, 'res.txt')
|
|
self.assertEqual(res1, RESOURCE_DATA)
|
|
res2 = pkgutil.get_data(pkg, 'sub/res.txt')
|
|
self.assertEqual(res2, RESOURCE_DATA)
|
|
|
|
names = []
|
|
for loader, name, ispkg in pkgutil.iter_modules([zip_file]):
|
|
names.append(name)
|
|
self.assertEqual(names, ['test_getdata_zipfile'])
|
|
|
|
del sys.path[0]
|
|
|
|
del sys.modules[pkg]
|
|
|
|
def test_unreadable_dir_on_syspath(self):
|
|
# issue7367 - walk_packages failed if unreadable dir on sys.path
|
|
package_name = "unreadable_package"
|
|
d = os.path.join(self.dirname, package_name)
|
|
# this does not appear to create an unreadable dir on Windows
|
|
# but the test should not fail anyway
|
|
os.mkdir(d, 0)
|
|
self.addCleanup(os.rmdir, d)
|
|
for t in pkgutil.walk_packages(path=[self.dirname]):
|
|
self.fail("unexpected package found")
|
|
|
|
class PkgutilPEP302Tests(unittest.TestCase):
|
|
|
|
class MyTestLoader(object):
|
|
def load_module(self, fullname):
|
|
# Create an empty module
|
|
mod = sys.modules.setdefault(fullname, imp.new_module(fullname))
|
|
mod.__file__ = "<%s>" % self.__class__.__name__
|
|
mod.__loader__ = self
|
|
# Make it a package
|
|
mod.__path__ = []
|
|
# Count how many times the module is reloaded
|
|
mod.__dict__['loads'] = mod.__dict__.get('loads',0) + 1
|
|
return mod
|
|
|
|
def get_data(self, path):
|
|
return "Hello, world!"
|
|
|
|
class MyTestImporter(object):
|
|
def find_module(self, fullname, path=None):
|
|
return PkgutilPEP302Tests.MyTestLoader()
|
|
|
|
def setUp(self):
|
|
sys.meta_path.insert(0, self.MyTestImporter())
|
|
|
|
def tearDown(self):
|
|
del sys.meta_path[0]
|
|
|
|
def test_getdata_pep302(self):
|
|
# Use a dummy importer/loader
|
|
self.assertEqual(pkgutil.get_data('foo', 'dummy'), "Hello, world!")
|
|
del sys.modules['foo']
|
|
|
|
def test_alreadyloaded(self):
|
|
# Ensure that get_data works without reloading - the "loads" module
|
|
# variable in the example loader should count how many times a reload
|
|
# occurs.
|
|
import foo
|
|
self.assertEqual(foo.loads, 1)
|
|
self.assertEqual(pkgutil.get_data('foo', 'dummy'), "Hello, world!")
|
|
self.assertEqual(foo.loads, 1)
|
|
del sys.modules['foo']
|
|
|
|
|
|
class ExtendPathTests(unittest.TestCase):
|
|
def create_init(self, pkgname):
|
|
dirname = tempfile.mkdtemp()
|
|
self.addCleanup(shutil.rmtree, dirname)
|
|
sys.path.insert(0, dirname)
|
|
|
|
pkgdir = os.path.join(dirname, pkgname)
|
|
os.mkdir(pkgdir)
|
|
with open(os.path.join(pkgdir, '__init__.py'), 'w') as fl:
|
|
fl.write('from pkgutil import extend_path\n__path__ = extend_path(__path__, __name__)\n')
|
|
|
|
return dirname
|
|
|
|
def create_submodule(self, dirname, pkgname, submodule_name, value):
|
|
module_name = os.path.join(dirname, pkgname, submodule_name + '.py')
|
|
with open(module_name, 'w') as fl:
|
|
print('value={}'.format(value), file=fl)
|
|
|
|
def setUp(self):
|
|
# Create 2 directories on sys.path
|
|
self.pkgname = 'foo'
|
|
self.dirname_0 = self.create_init(self.pkgname)
|
|
self.dirname_1 = self.create_init(self.pkgname)
|
|
|
|
def tearDown(self):
|
|
del sys.path[0]
|
|
del sys.path[0]
|
|
del sys.modules['foo']
|
|
del sys.modules['foo.bar']
|
|
del sys.modules['foo.baz']
|
|
|
|
def test_simple(self):
|
|
self.create_submodule(self.dirname_0, self.pkgname, 'bar', 0)
|
|
self.create_submodule(self.dirname_1, self.pkgname, 'baz', 1)
|
|
import foo.bar
|
|
import foo.baz
|
|
# Ensure we read the expected values
|
|
self.assertEqual(foo.bar.value, 0)
|
|
self.assertEqual(foo.baz.value, 1)
|
|
|
|
# Ensure the path is set up correctly
|
|
self.assertEqual(sorted(foo.__path__),
|
|
sorted([os.path.join(self.dirname_0, self.pkgname),
|
|
os.path.join(self.dirname_1, self.pkgname)]))
|
|
|
|
# XXX: test .pkg files
|
|
|
|
|
|
class NestedNamespacePackageTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.basedir = tempfile.mkdtemp()
|
|
self.old_path = sys.path[:]
|
|
|
|
def tearDown(self):
|
|
sys.path[:] = self.old_path
|
|
shutil.rmtree(self.basedir)
|
|
|
|
def create_module(self, name, contents):
|
|
base, final = name.rsplit('.', 1)
|
|
base_path = os.path.join(self.basedir, base.replace('.', os.path.sep))
|
|
os.makedirs(base_path, exist_ok=True)
|
|
with open(os.path.join(base_path, final + ".py"), 'w') as f:
|
|
f.write(contents)
|
|
|
|
def test_nested(self):
|
|
pkgutil_boilerplate = (
|
|
'import pkgutil; '
|
|
'__path__ = pkgutil.extend_path(__path__, __name__)')
|
|
self.create_module('a.pkg.__init__', pkgutil_boilerplate)
|
|
self.create_module('b.pkg.__init__', pkgutil_boilerplate)
|
|
self.create_module('a.pkg.subpkg.__init__', pkgutil_boilerplate)
|
|
self.create_module('b.pkg.subpkg.__init__', pkgutil_boilerplate)
|
|
self.create_module('a.pkg.subpkg.c', 'c = 1')
|
|
self.create_module('b.pkg.subpkg.d', 'd = 2')
|
|
sys.path.insert(0, os.path.join(self.basedir, 'a'))
|
|
sys.path.insert(0, os.path.join(self.basedir, 'b'))
|
|
import pkg
|
|
self.addCleanup(unload, 'pkg')
|
|
self.assertEqual(len(pkg.__path__), 2)
|
|
import pkg.subpkg
|
|
self.addCleanup(unload, 'pkg.subpkg')
|
|
self.assertEqual(len(pkg.subpkg.__path__), 2)
|
|
from pkg.subpkg.c import c
|
|
from pkg.subpkg.d import d
|
|
self.assertEqual(c, 1)
|
|
self.assertEqual(d, 2)
|
|
|
|
|
|
def test_main():
|
|
run_unittest(PkgutilTests, PkgutilPEP302Tests, ExtendPathTests,
|
|
NestedNamespacePackageTest)
|
|
# this is necessary if test is run repeated (like when finding leaks)
|
|
import zipimport
|
|
zipimport._zip_directory_cache.clear()
|
|
|
|
if __name__ == '__main__':
|
|
test_main()
|