|
|
@ -77,17 +77,28 @@ class TestMultipleSocks(unittest.TestCase):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TestSocks(unittest.TestCase):
|
|
|
|
class TestSocks(unittest.TestCase):
|
|
|
|
|
|
|
|
_SKIP_SOCKS_TEST = True
|
|
|
|
|
|
|
|
|
|
|
|
def setUp(self):
|
|
|
|
def setUp(self):
|
|
|
|
|
|
|
|
if self._SKIP_SOCKS_TEST:
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
|
|
self.port = random.randint(20000, 30000)
|
|
|
|
self.port = random.randint(20000, 30000)
|
|
|
|
self.server_process = subprocess.Popen([
|
|
|
|
self.server_process = subprocess.Popen([
|
|
|
|
'srelay', '-f', '-i', '127.0.0.1:%d' % self.port],
|
|
|
|
'srelay', '-f', '-i', '127.0.0.1:%d' % self.port],
|
|
|
|
stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
|
|
stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
|
|
|
|
|
|
|
|
|
|
def tearDown(self):
|
|
|
|
def tearDown(self):
|
|
|
|
|
|
|
|
if self._SKIP_SOCKS_TEST:
|
|
|
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
|
|
self.server_process.terminate()
|
|
|
|
self.server_process.terminate()
|
|
|
|
self.server_process.communicate()
|
|
|
|
self.server_process.communicate()
|
|
|
|
|
|
|
|
|
|
|
|
def _get_ip(self, protocol):
|
|
|
|
def _get_ip(self, protocol):
|
|
|
|
|
|
|
|
if self._SKIP_SOCKS_TEST:
|
|
|
|
|
|
|
|
return '127.0.0.1'
|
|
|
|
|
|
|
|
|
|
|
|
ydl = FakeYDL({
|
|
|
|
ydl = FakeYDL({
|
|
|
|
'proxy': '%s://127.0.0.1:%d' % (protocol, self.port),
|
|
|
|
'proxy': '%s://127.0.0.1:%d' % (protocol, self.port),
|
|
|
|
})
|
|
|
|
})
|
|
|
|