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
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
|
# Copyright 2014 Pants project contributors (see CONTRIBUTORS.md).
# Licensed under the Apache License, Version 2.0 (see LICENSE).
import os
import zipimport
import pkg_resources
import pytest
from pex.compatibility import to_bytes
from pex.finders import ChainedFinder
from pex.finders import _add_finder as add_finder
from pex.finders import _remove_finder as remove_finder
from pex.finders import (
find_eggs_in_zip,
find_wheels_in_zip,
get_entry_point_from_console_script,
get_script_from_egg,
get_script_from_whl
)
try:
import mock
except ImportError:
from unittest import mock
def test_chained_finder():
def finder1(importer, path_item, only=False):
for foo in ('foo', 'bar'):
yield foo
def finder2(importer, path_item, only=False):
yield 'baz'
cf = ChainedFinder([finder1])
assert list(cf(None, None)) == ['foo', 'bar']
cf = ChainedFinder([finder1, finder2])
assert list(cf(None, None)) == ['foo', 'bar', 'baz']
GET_FINDER = 'pex.finders._get_finder'
REGISTER_FINDER = 'pex.finders.pkg_resources.register_finder'
def test_add_new_finder():
with mock.patch(GET_FINDER) as mock_get_finder:
with mock.patch(REGISTER_FINDER) as mock_register_finder:
mock_get_finder.return_value = None
add_finder('foo', 'bar')
mock_register_finder.assert_called_with('foo', 'bar')
def test_append_finder():
with mock.patch(GET_FINDER) as mock_get_finder:
with mock.patch(REGISTER_FINDER) as mock_register_finder:
mock_get_finder.return_value = 'bar'
add_finder('foo', 'baz')
mock_register_finder.assert_called_with('foo', ChainedFinder(['bar', 'baz']))
with mock.patch(GET_FINDER) as mock_get_finder:
with mock.patch(REGISTER_FINDER) as mock_register_finder:
mock_get_finder.return_value = ChainedFinder(['bar'])
add_finder('foo', 'baz')
mock_register_finder.assert_called_with('foo', ChainedFinder(['bar', 'baz']))
def test_remove_finder():
# wasn't registered
with mock.patch(GET_FINDER) as mock_get_finder:
with mock.patch(REGISTER_FINDER) as mock_register_finder:
mock_get_finder.return_value = None
remove_finder('foo', 'baz')
assert not mock_register_finder.called
# was registered but we're asking for the wrong one
with mock.patch(GET_FINDER) as mock_get_finder:
with mock.patch(REGISTER_FINDER) as mock_register_finder:
mock_get_finder.return_value = ChainedFinder(['bar'])
remove_finder('foo', 'baz')
assert not mock_register_finder.called
# was registered but we're asking for the wrong one
with mock.patch(GET_FINDER) as mock_get_finder:
with mock.patch(REGISTER_FINDER) as mock_register_finder:
cf = ChainedFinder(['bar', 'baz', 'bak'])
mock_get_finder.return_value = cf
remove_finder('foo', 'baz')
assert cf.finders == ['bar', 'bak']
assert not mock_register_finder.called
# was registered but we're asking for the wrong one
with mock.patch(GET_FINDER) as mock_get_finder:
with mock.patch(REGISTER_FINDER) as mock_register_finder:
cf = ChainedFinder(['bar', 'baz'])
mock_get_finder.return_value = cf
remove_finder('foo', 'baz')
mock_register_finder.assert_called_with('foo', 'bar')
# was registered but we're asking for the wrong one
with mock.patch(GET_FINDER) as mock_get_finder:
with mock.patch(REGISTER_FINDER) as mock_register_finder:
mock_get_finder.return_value = 'bar'
remove_finder('foo', 'bar')
mock_register_finder.assert_called_with('foo', pkg_resources.find_nothing)
# was registered but we're asking for the wrong one
with mock.patch(GET_FINDER) as mock_get_finder:
with mock.patch(REGISTER_FINDER) as mock_register_finder:
mock_get_finder.return_value = ChainedFinder(['bar'])
remove_finder('foo', 'bar')
mock_register_finder.assert_called_with('foo', pkg_resources.find_nothing)
def test_get_script_from_egg_with_no_scripts():
# Make sure eggs without scripts don't cause errors.
egg_path = './tests/example_packages/Flask_Cache-0.13.1-py2.7.egg'
dists = list(find_eggs_in_zip(zipimport.zipimporter(egg_path), egg_path, only=True))
assert len(dists) == 1
dist = dists[0]
assert (None, None) == get_script_from_egg('non_existent_script', dist)
def test_get_script_from_egg():
egg_path = './tests/example_packages/eno-0.0.17-py2.7.egg'
dists = list(find_eggs_in_zip(zipimport.zipimporter(egg_path), egg_path, only=True))
assert len(dists) == 1
dist = dists[0]
location, content = get_script_from_egg('run_eno_server', dist)
assert os.path.join(egg_path, 'EGG-INFO/scripts/run_eno_server') == location
assert content.startswith('#!'), 'Expected a `scripts` style script with shebang.'
assert (None, None) == get_script_from_egg('non_existent_script', dist)
# In-part, tests a bug where the wheel distribution name has dashes as reported in:
# https://github.com/pantsbuild/pex/issues/443
# https://github.com/pantsbuild/pex/issues/551
def test_get_script_from_whl():
whl_path = './tests/example_packages/aws_cfn_bootstrap-1.4-py2-none-any.whl'
dists = list(find_wheels_in_zip(zipimport.zipimporter(whl_path), whl_path))
assert len(dists) == 1
dist = dists[0]
assert 'aws-cfn-bootstrap' == dist.project_name
script_path, script_content = get_script_from_whl('cfn-signal', dist)
assert os.path.join(whl_path, 'aws_cfn_bootstrap-1.4.data/scripts/cfn-signal') == script_path
assert script_content.startswith(to_bytes('#!')), 'Expected a `scripts`-style script w/shebang.'
assert (None, None) == get_script_from_whl('non_existent_script', dist)
class FakeDist(object):
def __init__(self, key, console_script_entry):
self.key = key
script = console_script_entry.split('=')[0].strip()
self._entry_map = {'console_scripts': {script: console_script_entry}}
def get_entry_map(self):
return self._entry_map
def test_get_entry_point_from_console_script():
dists = [FakeDist(key='fake', console_script_entry='bob= bob.main:run'),
FakeDist(key='fake', console_script_entry='bob =bob.main:run')]
dist, entrypoint = get_entry_point_from_console_script('bob', dists)
assert 'bob.main:run' == entrypoint
assert dist in dists
def test_get_entry_point_from_console_script_conflict():
dists = [FakeDist(key='bob', console_script_entry='bob= bob.main:run'),
FakeDist(key='fake', console_script_entry='bob =bob.main:run')]
with pytest.raises(RuntimeError):
get_entry_point_from_console_script('bob', dists)
def test_get_entry_point_from_console_script_dne():
dists = [FakeDist(key='bob', console_script_entry='bob= bob.main:run'),
FakeDist(key='fake', console_script_entry='bob =bob.main:run')]
assert (None, None) == get_entry_point_from_console_script('jane', dists)
|
