1
0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-15 08:04:49 +01:00
pywb/tests/test_socks.py

36 lines
1.1 KiB
Python
Raw Permalink Normal View History

from .base_config_test import BaseConfigTest, fmod_sl
import os
import pytest
# ============================================================================
class TestSOCKSProxy(BaseConfigTest):
@classmethod
def setup_class(cls):
pytest.importorskip('socks')
os.environ['SOCKS_HOST'] = 'localhost'
os.environ['SOCKS_PORT'] = '0'
super(TestSOCKSProxy, cls).setup_class('config_test.yaml')
@classmethod
def teardown_class(cls):
super(TestSOCKSProxy, cls).teardown_class()
def test_socks_attempt_connect(self, fmod_sl):
# no proxy is set, expect to fail if socks is being used
resp = self.get('/live/{0}http://httpbin.org/get', fmod_sl, status=400)
assert resp.status_int == 400
def test_socks_disable_enable(self, fmod_sl):
os.environ['SOCKS_DISABLE'] = '1'
resp = self.get('/live/{0}http://httpbin.org/get', fmod_sl, status=200)
assert resp.status_int == 200
os.environ['SOCKS_DISABLE'] = ''
resp = self.get('/live/{0}http://httpbin.org/get', fmod_sl, status=400)
assert resp.status_int == 400