mirror of
https://github.com/webrecorder/pywb.git
synced 2025-03-15 00:03:28 +01:00
eaaefbfd24
will run (this was cumbersome to maintain and not really useful) ReferRedirect just checks that the current request host header, if present, matches that of the referrer and checks that the coll and script name match. * removed proxy_pac as it was also unneeded/unused and required use of the hostpaths * added test for invalid CONNECT usage (405 response)
88 lines
3.5 KiB
Python
88 lines
3.5 KiB
Python
from pytest import raises
|
|
import webtest
|
|
import base64
|
|
|
|
from pywb.webapp.pywb_init import create_wb_router
|
|
from pywb.framework.wsgi_wrappers import init_app
|
|
from pywb.cdx.cdxobject import CDXObject
|
|
|
|
|
|
class TestProxyWb:
|
|
TEST_CONFIG = 'tests/test_config.yaml'
|
|
|
|
def setup(self):
|
|
self.app = init_app(create_wb_router,
|
|
load_yaml=True,
|
|
config_file=self.TEST_CONFIG)
|
|
|
|
self.testapp = webtest.TestApp(self.app)
|
|
|
|
def _assert_basic_html(self, resp):
|
|
assert resp.status_int == 200
|
|
assert resp.content_type == 'text/html'
|
|
assert resp.content_length > 0
|
|
|
|
def _assert_basic_text(self, resp):
|
|
assert resp.status_int == 200
|
|
assert resp.content_type == 'text/plain'
|
|
assert resp.content_length > 0
|
|
|
|
# 'Simulating' proxy by settings REQUEST_URI explicitly to http:// url and no SCRIPT_NAME
|
|
# would be nice to be able to test proxy more
|
|
def test_proxy_replay(self):
|
|
resp = self.testapp.get('/x-ignore-this-x', extra_environ = dict(REQUEST_URI = 'http://www.iana.org/domains/idn-tables', SCRIPT_NAME = ''))
|
|
self._assert_basic_html(resp)
|
|
|
|
assert '"20140126201127"' in resp.body
|
|
assert 'wb.js' in resp.body
|
|
|
|
def test_proxy_replay_auth_filtered(self):
|
|
headers = [('Proxy-Authorization', 'Basic ' + base64.b64encode('pywb-filt-2:'))]
|
|
resp = self.testapp.get('/x-ignore-this-x', headers = headers,
|
|
extra_environ = dict(REQUEST_URI = 'http://www.iana.org/', SCRIPT_NAME = ''))
|
|
|
|
self._assert_basic_html(resp)
|
|
|
|
assert '"20140126200624"' in resp.body
|
|
assert 'wb.js' in resp.body
|
|
|
|
def test_proxy_replay_auth(self):
|
|
headers = [('Proxy-Authorization', 'Basic ' + base64.b64encode('pywb'))]
|
|
resp = self.testapp.get('/x-ignore-this-x', headers = headers,
|
|
extra_environ = dict(REQUEST_URI = 'http://www.iana.org/', SCRIPT_NAME = ''))
|
|
|
|
self._assert_basic_html(resp)
|
|
|
|
assert '"20140127171238"' in resp.body
|
|
assert 'wb.js' in resp.body
|
|
|
|
def test_proxy_replay_auth_no_coll(self):
|
|
headers = [('Proxy-Authorization', 'Basic ' + base64.b64encode('no-such-coll'))]
|
|
resp = self.testapp.get('/x-ignore-this-x', headers = headers,
|
|
extra_environ = dict(REQUEST_URI = 'http://www.iana.org/', SCRIPT_NAME = ''),
|
|
status=407)
|
|
|
|
assert resp.status_int == 407
|
|
|
|
def test_proxy_replay_auth_invalid_1(self):
|
|
headers = [('Proxy-Authorization', 'abc' + base64.b64encode('no-such-coll'))]
|
|
resp = self.testapp.get('/x-ignore-this-x', headers = headers,
|
|
extra_environ = dict(REQUEST_URI = 'http://www.iana.org/', SCRIPT_NAME = ''),
|
|
status=407)
|
|
|
|
assert resp.status_int == 407
|
|
|
|
def test_proxy_replay_auth_invalid_2(self):
|
|
headers = [('Proxy-Authorization', 'basic')]
|
|
resp = self.testapp.get('/x-ignore-this-x', headers = headers,
|
|
extra_environ = dict(REQUEST_URI = 'http://www.iana.org/', SCRIPT_NAME = ''),
|
|
status=407)
|
|
|
|
|
|
def test_proxy_connect_unsupported(self):
|
|
resp = self.testapp.request('/x-ignore-this-x', method='CONNECT',
|
|
environ=dict(REQUEST_URI='example:443', SCRIPT_NAME=''),
|
|
status=405)
|
|
|
|
assert resp.status_int == 405
|