mirror of
https://github.com/webrecorder/pywb.git
synced 2025-03-15 00:03:28 +01:00
1aac5a9f15
proxy_resolvers range cache: and buffering cache for serving range requests, intended for videos but not only. full response cached in temp file and range requests served from cache, still experimental need to add deletion. youtube_dl: wrap youtube-dl import due to youtube-dl HTMLParser regex bug tests: add test for vi_ handler
47 lines
1.9 KiB
Python
47 lines
1.9 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/http://example.com/', headers=headers)
|
|
assert resp.status_int == 200
|
|
|
|
def test_live_rewrite_redirect_2(self):
|
|
resp = self.testapp.get('/rewrite/http://httpbin.org/redirect-to?url=http://example.com/')
|
|
assert resp.status_int == 302
|
|
|
|
def test_live_rewrite_post(self):
|
|
resp = self.testapp.post('/rewrite/httpbin.org/post', {'foo': 'bar', 'test': 'abc'})
|
|
assert resp.status_int == 200
|
|
assert '"foo": "bar"' in resp.body
|
|
assert '"test": "abc"' in resp.body
|
|
assert resp.status_int == 200
|
|
|
|
def test_live_rewrite_frame(self):
|
|
resp = self.testapp.get('/rewrite/tf_/http://example.com/')
|
|
assert resp.status_int == 200
|
|
assert '<iframe ' in resp.body
|
|
assert 'src="/rewrite/http://example.com/"' in resp.body
|
|
|
|
def test_live_invalid(self):
|
|
resp = self.testapp.get('/rewrite/http://abcdef', status=400)
|
|
assert resp.status_int == 400
|
|
|
|
def test_live_invalid_2(self):
|
|
resp = self.testapp.get('/rewrite/@#$@#$', status=400)
|
|
assert resp.status_int == 400
|
|
|
|
def test_live_video_info(self):
|
|
resp = self.testapp.get('/rewrite/vi_/https://www.youtube.com/watch?v=DjFZyFWSt1M')
|
|
assert resp.status_int == 200
|
|
assert resp.content_type == 'application/vnd.youtube-dl_formats+json', resp.content_type
|
|
|
|
|