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