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
|
import os
import tempfile
import shutil
import stat
import unittest
from . import environment
class TestEggInfo(unittest.TestCase):
def setUp(self):
self.test_dir = tempfile.mkdtemp()
self.old_cwd = os.getcwd()
os.chdir(self.test_dir)
def tearDown(self):
os.chdir(self.old_cwd)
shutil.rmtree(self.test_dir)
def _create_project(self):
with open('setup.py', 'w') as f:
f.write('from setuptools import setup\n')
f.write('\n')
f.write('setup(\n')
f.write(" name='foo',\n")
f.write(" py_modules=['hello'],\n")
f.write(" entry_points={'console_scripts': ['hi = hello.run']},\n")
f.write(' zip_safe=False,\n')
f.write(' )\n')
with open('hello.py', 'w') as f:
f.write('def run():\n')
f.write(" print('hello')\n")
def test_egg_base_installed_egg_info(self):
self._create_project()
temp_dir = tempfile.mkdtemp(prefix='setuptools-test.')
os.chmod(temp_dir, stat.S_IRWXU)
try:
paths = {}
for dirname in ['home', 'lib', 'scripts', 'data', 'egg-base']:
paths[dirname] = os.path.join(temp_dir, dirname)
os.mkdir(paths[dirname])
config = os.path.join(paths['home'], '.pydistutils.cfg')
with open(config, 'w') as f:
f.write('[egg_info]\n')
f.write('egg-base = %s\n' % paths['egg-base'])
environ = os.environ.copy()
environ['HOME'] = paths['home']
code, data = environment.run_setup_py(
cmd=[
'install', '--home', paths['home'],
'--install-lib', paths['lib'],
'--install-scripts', paths['scripts'],
'--install-data', paths['data']],
pypath=os.pathsep.join([paths['lib'], self.old_cwd]),
data_stream=1,
env=environ)
if code:
raise AssertionError(data)
egg_info = None
for dirpath, dirnames, filenames in os.walk(paths['lib']):
if os.path.basename(dirpath) == 'EGG-INFO':
egg_info = sorted(filenames)
self.assertEqual(
egg_info,
['PKG-INFO',
'SOURCES.txt',
'dependency_links.txt',
'entry_points.txt',
'not-zip-safe',
'top_level.txt'])
finally:
shutil.rmtree(temp_dir)
def test_suite():
return unittest.defaultTestLoader.loadTestsFromName(__name__)
|
