+++ /dev/null
-from test.test_support import run_unittest\r
-import unittest\r
-import sys\r
-import imp\r
-import pkgutil\r
-import os\r
-import os.path\r
-import tempfile\r
-import shutil\r
-import zipfile\r
-\r
-\r
-\r
-class PkgutilTests(unittest.TestCase):\r
-\r
- def setUp(self):\r
- self.dirname = tempfile.mkdtemp()\r
- sys.path.insert(0, self.dirname)\r
-\r
- def tearDown(self):\r
- del sys.path[0]\r
- shutil.rmtree(self.dirname)\r
-\r
- def test_getdata_filesys(self):\r
- pkg = 'test_getdata_filesys'\r
-\r
- # Include a LF and a CRLF, to test that binary data is read back\r
- RESOURCE_DATA = 'Hello, world!\nSecond line\r\nThird line'\r
-\r
- # Make a package with some resources\r
- package_dir = os.path.join(self.dirname, pkg)\r
- os.mkdir(package_dir)\r
- # Empty init.py\r
- f = open(os.path.join(package_dir, '__init__.py'), "wb")\r
- f.close()\r
- # Resource files, res.txt, sub/res.txt\r
- f = open(os.path.join(package_dir, 'res.txt'), "wb")\r
- f.write(RESOURCE_DATA)\r
- f.close()\r
- os.mkdir(os.path.join(package_dir, 'sub'))\r
- f = open(os.path.join(package_dir, 'sub', 'res.txt'), "wb")\r
- f.write(RESOURCE_DATA)\r
- f.close()\r
-\r
- # Check we can read the resources\r
- res1 = pkgutil.get_data(pkg, 'res.txt')\r
- self.assertEqual(res1, RESOURCE_DATA)\r
- res2 = pkgutil.get_data(pkg, 'sub/res.txt')\r
- self.assertEqual(res2, RESOURCE_DATA)\r
-\r
- del sys.modules[pkg]\r
-\r
- def test_getdata_zipfile(self):\r
- zip = 'test_getdata_zipfile.zip'\r
- pkg = 'test_getdata_zipfile'\r
-\r
- # Include a LF and a CRLF, to test that binary data is read back\r
- RESOURCE_DATA = 'Hello, world!\nSecond line\r\nThird line'\r
-\r
- # Make a package with some resources\r
- zip_file = os.path.join(self.dirname, zip)\r
- z = zipfile.ZipFile(zip_file, 'w')\r
-\r
- # Empty init.py\r
- z.writestr(pkg + '/__init__.py', "")\r
- # Resource files, res.txt, sub/res.txt\r
- z.writestr(pkg + '/res.txt', RESOURCE_DATA)\r
- z.writestr(pkg + '/sub/res.txt', RESOURCE_DATA)\r
- z.close()\r
-\r
- # Check we can read the resources\r
- sys.path.insert(0, zip_file)\r
- res1 = pkgutil.get_data(pkg, 'res.txt')\r
- self.assertEqual(res1, RESOURCE_DATA)\r
- res2 = pkgutil.get_data(pkg, 'sub/res.txt')\r
- self.assertEqual(res2, RESOURCE_DATA)\r
- del sys.path[0]\r
-\r
- del sys.modules[pkg]\r
-\r
-class PkgutilPEP302Tests(unittest.TestCase):\r
-\r
- class MyTestLoader(object):\r
- def load_module(self, fullname):\r
- # Create an empty module\r
- mod = sys.modules.setdefault(fullname, imp.new_module(fullname))\r
- mod.__file__ = "<%s>" % self.__class__.__name__\r
- mod.__loader__ = self\r
- # Make it a package\r
- mod.__path__ = []\r
- # Count how many times the module is reloaded\r
- mod.__dict__['loads'] = mod.__dict__.get('loads',0) + 1\r
- return mod\r
-\r
- def get_data(self, path):\r
- return "Hello, world!"\r
-\r
- class MyTestImporter(object):\r
- def find_module(self, fullname, path=None):\r
- return PkgutilPEP302Tests.MyTestLoader()\r
-\r
- def setUp(self):\r
- sys.meta_path.insert(0, self.MyTestImporter())\r
-\r
- def tearDown(self):\r
- del sys.meta_path[0]\r
-\r
- def test_getdata_pep302(self):\r
- # Use a dummy importer/loader\r
- self.assertEqual(pkgutil.get_data('foo', 'dummy'), "Hello, world!")\r
- del sys.modules['foo']\r
-\r
- def test_alreadyloaded(self):\r
- # Ensure that get_data works without reloading - the "loads" module\r
- # variable in the example loader should count how many times a reload\r
- # occurs.\r
- import foo\r
- self.assertEqual(foo.loads, 1)\r
- self.assertEqual(pkgutil.get_data('foo', 'dummy'), "Hello, world!")\r
- self.assertEqual(foo.loads, 1)\r
- del sys.modules['foo']\r
-\r
-def test_main():\r
- run_unittest(PkgutilTests, PkgutilPEP302Tests)\r
- # this is necessary if test is run repeated (like when finding leaks)\r
- import zipimport\r
- zipimport._zip_directory_cache.clear()\r
-\r
-if __name__ == '__main__':\r
- test_main()\r