def test_forwarded_scheme(): rewriter = LiveRewriter() env = {'HTTP_X_FORWARDED_PROTO': 'https', 'Other': 'Value'} req_headers = rewriter.translate_headers('http://example.com/', 'com,example)/', env) assert req_headers == {'X-Forwarded-Proto': 'http'}
def test_csrf_token_headers(): rewriter = LiveRewriter() env = {'HTTP_X_CSRFTOKEN': 'wrong', 'HTTP_COOKIE': 'csrftoken=foobar'} req_headers = rewriter.translate_headers('http://example.com/', 'com,example)/', env) assert req_headers == {'X-CSRFToken': 'foobar', 'Cookie': 'csrftoken=foobar'}
def test_req_cookie_rewrite_3(): rewriter = LiveRewriter() env = {} urlkey = 'example,example,test)/' url = 'test.example.example/' req_headers = rewriter.translate_headers(url, urlkey, env) assert req_headers == {'Cookie': '; FOO=&bar=1'}
def __init__(self, config): super(RewriteHandler, self).__init__(config) proxyhostport = config.get('proxyhostport') self.rewriter = LiveRewriter(is_framed_replay=self.is_frame_mode, proxies=proxyhostport) self.proxies = self.rewriter.proxies self.head_insert_view = HeadInsertView.init_from_config(config) self.live_cookie = config.get('live-cookie', self.LIVE_COOKIE) self.verify = config.get('verify_ssl', True) self.ydl = None self._cache = None
def get_rewritten(*args, **kwargs): status_headers, buff = LiveRewriter().get_rewritten(remote_only=False, *args, **kwargs) return status_headers, to_native_str(buff)
def get_rewritten(*args, **kwargs): return LiveRewriter().get_rewritten(remote_only=False, *args, **kwargs)
def get_rewritten(*args, **kwargs): return LiveRewriter().get_rewritten(*args, **kwargs)