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_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']
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)