def test_get_proxy_has_proxy(self):
with mock.patch.object(Proxies, 'check_live') as check_alive_mock:
Registry().set('config', {'main': {'proxies_died_limit': 10}})
proxies = Proxies()
test_set = ['a', 'b', 'c']
proxies._proxies = test_set
check_alive_mock.return_value = True
for _ in range(0, 3):
assert proxies.get_proxy() in test_set
def test_count(self):
Registry().set('config', {'main': {'proxies_died_limit': 10}})
proxies = Proxies()
proxies._proxies = ['a', 'b', 'c']
assert 3 == proxies.count()