mirror of
https://github.com/webrecorder/pywb.git
synced 2025-03-24 06:59:52 +01:00
35 lines
1.3 KiB
Python
35 lines
1.3 KiB
Python
from pywb.webapp.live_rewrite_handler import create_live_rewriter_app
|
|
from pywb.framework.wsgi_wrappers import init_app
|
|
import webtest
|
|
|
|
class TestLiveRewriter:
|
|
def setup(self):
|
|
self.app = init_app(create_live_rewriter_app, load_yaml=False,
|
|
config=dict(framed_replay=True))
|
|
self.testapp = webtest.TestApp(self.app)
|
|
|
|
def test_live_rewrite_1(self):
|
|
headers = [('User-Agent', 'python'), ('Referer', 'http://localhost:80/rewrite/other.example.com')]
|
|
resp = self.testapp.get('/rewrite/mp_/http://example.com/', headers=headers)
|
|
assert resp.status_int == 200
|
|
|
|
def test_live_rewrite_redirect_2(self):
|
|
resp = self.testapp.get('/rewrite/mp_/http://facebook.com/')
|
|
assert resp.status_int == 301
|
|
|
|
def test_live_rewrite_frame(self):
|
|
resp = self.testapp.get('/rewrite/http://example.com/')
|
|
assert resp.status_int == 200
|
|
assert '<iframe ' in resp.body
|
|
assert 'src="/rewrite/mp_/http://example.com/"' in resp.body
|
|
|
|
def test_live_invalid(self):
|
|
resp = self.testapp.get('/rewrite/mp_/http://abcdef', status=400)
|
|
assert resp.status_int == 400
|
|
|
|
def test_live_invalid_2(self):
|
|
resp = self.testapp.get('/rewrite/mp_/@#$@#$', status=400)
|
|
assert resp.status_int == 400
|
|
|
|
|