1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
import zipimport
import py
import os
import py_compile
import shutil
import time
import zipfile
TESTFN = '@test'
created_paths = dict.fromkeys([u'_top_level',
os.path.join(u'_pkg', '__init__'),
os.path.join(u'_pkg', 'submodule'),
os.path.join(u'_pkg', '_subpkg', '__init__'),
os.path.join(u'_pkg', '_subpkg', 'submodule')
])
class AppTestZipImport:
spaceconfig = {
"usemodules": ['zipimport', 'time', 'struct', 'itertools', 'binascii']
}
def setup_class(cls):
cls.w_created_paths = cls.space.wrap(created_paths)
def w_temp_zipfile(self, created_paths, source=True, bytecode=True):
"""Create a temporary zip file for testing.
Clears zipimport._zip_directory_cache.
"""
import zipimport, os, shutil, zipfile, py_compile
example_code = 'attr = None'
TESTFN = '@test'
zipimport._zip_directory_cache.clear()
zip_path = TESTFN + '.zip'
bytecode_suffix = 'c'# if __debug__ else 'o'
zip_file = zipfile.ZipFile(zip_path, 'w')
for path in created_paths:
if os.sep in path:
directory = os.path.split(path)[0]
if not os.path.exists(directory):
os.makedirs(directory)
code_path = path + '.py'
try:
temp_file = open(code_path, 'w')
temp_file.write(example_code)
finally:
temp_file.close()
if source:
zip_file.write(code_path)
if bytecode:
py_compile.compile(code_path, code_path + bytecode_suffix,
doraise=True)
zip_file.write(code_path + bytecode_suffix)
zip_file.close()
return os.path.abspath(zip_path)
def w_cleanup_zipfile(self, created_paths):
import os, shutil
bytecode_suffix = 'c'# if __debug__ else 'o'
zip_path = '@test.zip'
for path in created_paths:
if os.sep in path:
directory = os.path.split(path)[0]
if os.path.exists(directory):
shutil.rmtree(directory)
else:
for suffix in ('.py', '.py' + bytecode_suffix):
if os.path.exists(path + suffix):
os.unlink(path + suffix)
os.unlink(zip_path)
def test_inheritance(self):
# Should inherit from ImportError.
import zipimport
assert issubclass(zipimport.ZipImportError, ImportError)
def test_nonzip(self):
import os
import zipimport
# ZipImportError should be raised if a non-zip file is specified.
TESTFN = '@test'
test_file = open(TESTFN, 'w')
try:
test_file.write("# Test file for zipimport.")
raises(zipimport.ZipImportError,
zipimport.zipimporter, TESTFN)
finally:
test_file.close()
os.unlink(TESTFN)
def test_root(self):
import zipimport, os
raises(zipimport.ZipImportError, zipimport.zipimporter,
os.sep)
def test_direct_path(self):
# A zipfile should return an instance of zipimporter.
import zipimport
zip_path = self.temp_zipfile(self.created_paths)
try:
zip_importer = zipimport.zipimporter(zip_path)
assert isinstance(zip_importer, zipimport.zipimporter)
assert zip_importer.archive == zip_path
assert zip_importer.prefix == ''
assert zip_path in zipimport._zip_directory_cache
finally:
self.cleanup_zipfile(self.created_paths)
def test_pkg_path(self):
# Thanks to __path__, need to be able to work off of a path with a zip
# file at the front and a path for the rest.
import zipimport, os
zip_path = self.temp_zipfile(self.created_paths)
try:
prefix = '_pkg/'
path = os.path.join(zip_path, prefix)
zip_importer = zipimport.zipimporter(path)
assert isinstance(zip_importer, zipimport.zipimporter)
assert zip_importer.archive == zip_path
assert zip_importer.prefix == prefix.replace('/', os.path.sep)
assert zip_path in zipimport._zip_directory_cache
finally:
self.cleanup_zipfile(self.created_paths)
def test_zip_directory_cache(self):
# Test that _zip_directory_cache is set properly.
# Using a package entry to test using a hard example.
import zipimport, os
zip_path = self.temp_zipfile(self.created_paths, bytecode=False)
try:
importer = zipimport.zipimporter(os.path.join(zip_path, '_pkg'))
assert zip_path in zipimport._zip_directory_cache
file_set = set(zipimport._zip_directory_cache[zip_path].keys())
compare_set = set(path + '.py' for path in self.created_paths)
assert file_set == compare_set
finally:
self.cleanup_zipfile(self.created_paths)
|
