Mercurial > cpython-lru_cache-weakref

annotate Lib/test/test_pkgutil.py @ 78069:a8aa77458e12 3.2

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add c_size_t and c_ssize_t to the ctypes summary table.
author Antoine Pitrou <solipsis@pitrou.net>
date Thu, 12 Jul 2012 20:31:50 +0200
parents f86df559319d
children 606f5cc41d35
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
78039
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
1 from test.support import run_unittest, unload
46478
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
2 import unittest
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
3 import sys
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
4 import imp
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
5 import pkgutil
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
6 import os
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
7 import os.path
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
8 import tempfile
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
9 import shutil
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
10 import zipfile
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
11
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
12
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
13
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
14 class PkgutilTests(unittest.TestCase):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
15
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
16 def setUp(self):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
17 self.dirname = tempfile.mkdtemp()
72799
ff72f76dcf43 Issue #7367: Ensure test directory always gets removed.
Ned Deily <nad@acm.org>
parents: 72773
diff changeset
18 self.addCleanup(shutil.rmtree, self.dirname)
46478
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
19 sys.path.insert(0, self.dirname)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
20
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
21 def tearDown(self):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
22 del sys.path[0]
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
23
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
24 def test_getdata_filesys(self):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
25 pkg = 'test_getdata_filesys'
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
26
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
27 # Include a LF and a CRLF, to test that binary data is read back
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
28 RESOURCE_DATA = b'Hello, world!\nSecond line\r\nThird line'
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
29
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
30 # Make a package with some resources
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
31 package_dir = os.path.join(self.dirname, pkg)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
32 os.mkdir(package_dir)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
33 # Empty init.py
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
34 f = open(os.path.join(package_dir, '__init__.py'), "wb")
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
35 f.close()
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
36 # Resource files, res.txt, sub/res.txt
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
37 f = open(os.path.join(package_dir, 'res.txt'), "wb")
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
38 f.write(RESOURCE_DATA)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
39 f.close()
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
40 os.mkdir(os.path.join(package_dir, 'sub'))
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
41 f = open(os.path.join(package_dir, 'sub', 'res.txt'), "wb")
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
42 f.write(RESOURCE_DATA)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
43 f.close()
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
44
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
45 # Check we can read the resources
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
46 res1 = pkgutil.get_data(pkg, 'res.txt')
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
47 self.assertEqual(res1, RESOURCE_DATA)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
48 res2 = pkgutil.get_data(pkg, 'sub/res.txt')
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
49 self.assertEqual(res2, RESOURCE_DATA)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
50
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
51 del sys.modules[pkg]
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
52
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
53 def test_getdata_zipfile(self):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
54 zip = 'test_getdata_zipfile.zip'
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
55 pkg = 'test_getdata_zipfile'
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
56
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
57 # Include a LF and a CRLF, to test that binary data is read back
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
58 RESOURCE_DATA = b'Hello, world!\nSecond line\r\nThird line'
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
59
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
60 # Make a package with some resources
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
61 zip_file = os.path.join(self.dirname, zip)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
62 z = zipfile.ZipFile(zip_file, 'w')
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
63
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
64 # Empty init.py
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
65 z.writestr(pkg + '/__init__.py', "")
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
66 # Resource files, res.txt, sub/res.txt
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
67 z.writestr(pkg + '/res.txt', RESOURCE_DATA)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
68 z.writestr(pkg + '/sub/res.txt', RESOURCE_DATA)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
69 z.close()
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
70
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
71 # Check we can read the resources
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
72 sys.path.insert(0, zip_file)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
73 res1 = pkgutil.get_data(pkg, 'res.txt')
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
74 self.assertEqual(res1, RESOURCE_DATA)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
75 res2 = pkgutil.get_data(pkg, 'sub/res.txt')
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
76 self.assertEqual(res2, RESOURCE_DATA)
55602
680ad8eed33b Merged revisions 73833,73838,73850-73852,73856-73857 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
77
680ad8eed33b Merged revisions 73833,73838,73850-73852,73856-73857 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
78 names = []
680ad8eed33b Merged revisions 73833,73838,73850-73852,73856-73857 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
79 for loader, name, ispkg in pkgutil.iter_modules([zip_file]):
680ad8eed33b Merged revisions 73833,73838,73850-73852,73856-73857 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
80 names.append(name)
680ad8eed33b Merged revisions 73833,73838,73850-73852,73856-73857 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
81 self.assertEqual(names, ['test_getdata_zipfile'])
680ad8eed33b Merged revisions 73833,73838,73850-73852,73856-73857 via svnmerge from
Georg Brandl <georg@python.org>
parents: 47201
diff changeset
82
46478
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
83 del sys.path[0]
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
84
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
85 del sys.modules[pkg]
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
86
72773
a1e6633ef3f1 Issue #7367: Add test case to test_pkgutil for walking path with
Ned Deily <nad@acm.org>
parents: 55602
diff changeset
87 def test_unreadable_dir_on_syspath(self):
a1e6633ef3f1 Issue #7367: Add test case to test_pkgutil for walking path with
Ned Deily <nad@acm.org>
parents: 55602
diff changeset
88 # issue7367 - walk_packages failed if unreadable dir on sys.path
a1e6633ef3f1 Issue #7367: Add test case to test_pkgutil for walking path with
Ned Deily <nad@acm.org>
parents: 55602
diff changeset
89 package_name = "unreadable_package"
a1e6633ef3f1 Issue #7367: Add test case to test_pkgutil for walking path with
Ned Deily <nad@acm.org>
parents: 55602
diff changeset
90 d = os.path.join(self.dirname, package_name)
a1e6633ef3f1 Issue #7367: Add test case to test_pkgutil for walking path with
Ned Deily <nad@acm.org>
parents: 55602
diff changeset
91 # this does not appear to create an unreadable dir on Windows
a1e6633ef3f1 Issue #7367: Add test case to test_pkgutil for walking path with
Ned Deily <nad@acm.org>
parents: 55602
diff changeset
92 # but the test should not fail anyway
a1e6633ef3f1 Issue #7367: Add test case to test_pkgutil for walking path with
Ned Deily <nad@acm.org>
parents: 55602
diff changeset
93 os.mkdir(d, 0)
72799
ff72f76dcf43 Issue #7367: Ensure test directory always gets removed.
Ned Deily <nad@acm.org>
parents: 72773
diff changeset
94 self.addCleanup(os.rmdir, d)
72773
a1e6633ef3f1 Issue #7367: Add test case to test_pkgutil for walking path with
Ned Deily <nad@acm.org>
parents: 55602
diff changeset
95 for t in pkgutil.walk_packages(path=[self.dirname]):
a1e6633ef3f1 Issue #7367: Add test case to test_pkgutil for walking path with
Ned Deily <nad@acm.org>
parents: 55602
diff changeset
96 self.fail("unexpected package found")
a1e6633ef3f1 Issue #7367: Add test case to test_pkgutil for walking path with
Ned Deily <nad@acm.org>
parents: 55602
diff changeset
97
46478
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
98 class PkgutilPEP302Tests(unittest.TestCase):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
99
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
100 class MyTestLoader(object):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
101 def load_module(self, fullname):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
102 # Create an empty module
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
103 mod = sys.modules.setdefault(fullname, imp.new_module(fullname))
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
104 mod.__file__ = "<%s>" % self.__class__.__name__
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
105 mod.__loader__ = self
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
106 # Make it a package
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
107 mod.__path__ = []
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
108 # Count how many times the module is reloaded
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
109 mod.__dict__['loads'] = mod.__dict__.get('loads',0) + 1
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
110 return mod
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
111
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
112 def get_data(self, path):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
113 return "Hello, world!"
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
114
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
115 class MyTestImporter(object):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
116 def find_module(self, fullname, path=None):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
117 return PkgutilPEP302Tests.MyTestLoader()
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
118
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
119 def setUp(self):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
120 sys.meta_path.insert(0, self.MyTestImporter())
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
121
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
122 def tearDown(self):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
123 del sys.meta_path[0]
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
124
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
125 def test_getdata_pep302(self):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
126 # Use a dummy importer/loader
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
127 self.assertEqual(pkgutil.get_data('foo', 'dummy'), "Hello, world!")
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
128 del sys.modules['foo']
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
129
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
130 def test_alreadyloaded(self):
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
131 # Ensure that get_data works without reloading - the "loads" module
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
132 # variable in the example loader should count how many times a reload
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
133 # occurs.
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
134 import foo
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
135 self.assertEqual(foo.loads, 1)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
136 self.assertEqual(pkgutil.get_data('foo', 'dummy'), "Hello, world!")
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
137 self.assertEqual(foo.loads, 1)
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
138 del sys.modules['foo']
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
139
76972
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
140
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
141 class ExtendPathTests(unittest.TestCase):
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
142 def create_init(self, pkgname):
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
143 dirname = tempfile.mkdtemp()
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
144 self.addCleanup(shutil.rmtree, dirname)
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
145 sys.path.insert(0, dirname)
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
146
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
147 pkgdir = os.path.join(dirname, pkgname)
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
148 os.mkdir(pkgdir)
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
149 with open(os.path.join(pkgdir, '__init__.py'), 'w') as fl:
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
150 fl.write('from pkgutil import extend_path\n__path__ = extend_path(__path__, __name__)\n')
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
151
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
152 return dirname
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
153
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
154 def create_submodule(self, dirname, pkgname, submodule_name, value):
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
155 module_name = os.path.join(dirname, pkgname, submodule_name + '.py')
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
156 with open(module_name, 'w') as fl:
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
157 print('value={}'.format(value), file=fl)
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
158
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
159 def setUp(self):
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
160 # Create 2 directories on sys.path
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
161 self.pkgname = 'foo'
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
162 self.dirname_0 = self.create_init(self.pkgname)
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
163 self.dirname_1 = self.create_init(self.pkgname)
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
164
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
165 def tearDown(self):
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
166 del sys.path[0]
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
167 del sys.path[0]
76983
57936bd91bcf Cleanup so subsequent tests won't fail. Needs to be moved into a support routine (see 14715).
Eric V. Smith <eric@trueblade.com>
parents: 76972
diff changeset
168 del sys.modules['foo']
57936bd91bcf Cleanup so subsequent tests won't fail. Needs to be moved into a support routine (see 14715).
Eric V. Smith <eric@trueblade.com>
parents: 76972
diff changeset
169 del sys.modules['foo.bar']
57936bd91bcf Cleanup so subsequent tests won't fail. Needs to be moved into a support routine (see 14715).
Eric V. Smith <eric@trueblade.com>
parents: 76972
diff changeset
170 del sys.modules['foo.baz']
76972
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
171
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
172 def test_simple(self):
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
173 self.create_submodule(self.dirname_0, self.pkgname, 'bar', 0)
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
174 self.create_submodule(self.dirname_1, self.pkgname, 'baz', 1)
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
175 import foo.bar
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
176 import foo.baz
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
177 # Ensure we read the expected values
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
178 self.assertEqual(foo.bar.value, 0)
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
179 self.assertEqual(foo.baz.value, 1)
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
180
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
181 # Ensure the path is set up correctly
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
182 self.assertEqual(sorted(foo.__path__),
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
183 sorted([os.path.join(self.dirname_0, self.pkgname),
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
184 os.path.join(self.dirname_1, self.pkgname)]))
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
185
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
186 # XXX: test .pkg files
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
187
48cb6b67d306 Issue #14817: Add rudimentary tests for pkgutil.extend_path.
Eric V. Smith <eric@trueblade.com>
parents: 72799
diff changeset
188
78039
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
189 class NestedNamespacePackageTest(unittest.TestCase):
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
190
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
191 def setUp(self):
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
192 self.basedir = tempfile.mkdtemp()
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
193 self.old_path = sys.path[:]
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
194
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
195 def tearDown(self):
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
196 sys.path[:] = self.old_path
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
197 shutil.rmtree(self.basedir)
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
198
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
199 def create_module(self, name, contents):
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
200 base, final = name.rsplit('.', 1)
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
201 base_path = os.path.join(self.basedir, base.replace('.', os.path.sep))
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
202 os.makedirs(base_path, exist_ok=True)
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
203 with open(os.path.join(base_path, final + ".py"), 'w') as f:
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
204 f.write(contents)
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
205
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
206 def test_nested(self):
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
207 pkgutil_boilerplate = (
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
208 'import pkgutil; '
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
209 '__path__ = pkgutil.extend_path(__path__, __name__)')
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
210 self.create_module('a.pkg.__init__', pkgutil_boilerplate)
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
211 self.create_module('b.pkg.__init__', pkgutil_boilerplate)
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
212 self.create_module('a.pkg.subpkg.__init__', pkgutil_boilerplate)
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
213 self.create_module('b.pkg.subpkg.__init__', pkgutil_boilerplate)
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
214 self.create_module('a.pkg.subpkg.c', 'c = 1')
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
215 self.create_module('b.pkg.subpkg.d', 'd = 2')
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
216 sys.path.insert(0, os.path.join(self.basedir, 'a'))
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
217 sys.path.insert(0, os.path.join(self.basedir, 'b'))
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
218 import pkg
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
219 self.addCleanup(unload, 'pkg')
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
220 self.assertEqual(len(pkg.__path__), 2)
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
221 import pkg.subpkg
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
222 self.addCleanup(unload, 'pkg.subpkg')
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
223 self.assertEqual(len(pkg.subpkg.__path__), 2)
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
224 from pkg.subpkg.c import c
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
225 from pkg.subpkg.d import d
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
226 self.assertEqual(c, 1)
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
227 self.assertEqual(d, 2)
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
228
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
229
46478
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
230 def test_main():
78039
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
231 run_unittest(PkgutilTests, PkgutilPEP302Tests, ExtendPathTests,
f86df559319d Backport test for nested namespace packages.
Antoine Pitrou <solipsis@pitrou.net>
parents: 76983
diff changeset
232 NestedNamespacePackageTest)
46516
ec48490b969a Merged revisions 62420-62421,62423-62424 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 46478
diff changeset
233 # this is necessary if test is run repeated (like when finding leaks)
ec48490b969a Merged revisions 62420-62421,62423-62424 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 46478
diff changeset
234 import zipimport
ec48490b969a Merged revisions 62420-62421,62423-62424 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents: 46478
diff changeset
235 zipimport._zip_directory_cache.clear()
46478
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
236
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
237 if __name__ == '__main__':
beb7abcd6c01 Merged revisions 62350-62355,62358-62359,62364-62365,62370,62372-62375,62378-62379,62381 via svnmerge from
Christian Heimes <christian@cheimes.de>
parents:
diff changeset
238 test_main()