forked from Ponysearch/Ponysearch
62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
from searx.testing import SearxTestCase
|
|
|
|
from searx.poolrequests import get_proxy_cycles, get_proxies
|
|
|
|
|
|
CONFIG = {'http': ['http://localhost:9090', 'http://localhost:9092'],
|
|
'https': ['http://localhost:9091', 'http://localhost:9093']}
|
|
|
|
|
|
class TestProxy(SearxTestCase):
|
|
|
|
def test_noconfig(self):
|
|
cycles = get_proxy_cycles(None)
|
|
self.assertIsNone(cycles)
|
|
|
|
cycles = get_proxy_cycles(False)
|
|
self.assertIsNone(cycles)
|
|
|
|
def test_oldconfig(self):
|
|
config = {
|
|
'http': 'http://localhost:9090',
|
|
'https': 'http://localhost:9091',
|
|
}
|
|
cycles = get_proxy_cycles(config)
|
|
self.assertEqual(next(cycles['http']), 'http://localhost:9090')
|
|
self.assertEqual(next(cycles['http']), 'http://localhost:9090')
|
|
self.assertEqual(next(cycles['https']), 'http://localhost:9091')
|
|
self.assertEqual(next(cycles['https']), 'http://localhost:9091')
|
|
|
|
def test_one_proxy(self):
|
|
config = {
|
|
'http': ['http://localhost:9090'],
|
|
'https': ['http://localhost:9091'],
|
|
}
|
|
cycles = get_proxy_cycles(config)
|
|
self.assertEqual(next(cycles['http']), 'http://localhost:9090')
|
|
self.assertEqual(next(cycles['http']), 'http://localhost:9090')
|
|
self.assertEqual(next(cycles['https']), 'http://localhost:9091')
|
|
self.assertEqual(next(cycles['https']), 'http://localhost:9091')
|
|
|
|
def test_multiple_proxies(self):
|
|
cycles = get_proxy_cycles(CONFIG)
|
|
self.assertEqual(next(cycles['http']), 'http://localhost:9090')
|
|
self.assertEqual(next(cycles['http']), 'http://localhost:9092')
|
|
self.assertEqual(next(cycles['http']), 'http://localhost:9090')
|
|
self.assertEqual(next(cycles['https']), 'http://localhost:9091')
|
|
self.assertEqual(next(cycles['https']), 'http://localhost:9093')
|
|
self.assertEqual(next(cycles['https']), 'http://localhost:9091')
|
|
|
|
def test_getproxies_none(self):
|
|
self.assertIsNone(get_proxies(None))
|
|
|
|
def test_getproxies_config(self):
|
|
cycles = get_proxy_cycles(CONFIG)
|
|
self.assertEqual(get_proxies(cycles), {
|
|
'http': 'http://localhost:9090',
|
|
'https': 'http://localhost:9091'
|
|
})
|
|
self.assertEqual(get_proxies(cycles), {
|
|
'http': 'http://localhost:9092',
|
|
'https': 'http://localhost:9093'
|
|
})
|