backup/pywb
1
0
Fork 0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-24 06:59:52 +01:00
Code Issues Projects Releases Activity

cookie_rewriter: add tests for exact cookie rewriter

Browse Source
This commit is contained in:
Ilya Kreymer 2014-11-13 09:43:50 -08:00
parent 20070e95b6
commit b8b8c30573
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pywb/rewrite/test/test_cookie_rewriter.py
View File

@ -1,4 +1,5 @@
r""" r"""
# Default -- MinimalScopeRewriter
# No rewriting # No rewriting
>>> rewrite_cookie('a=b; c=d;') >>> rewrite_cookie('a=b; c=d;')
[('Set-Cookie', 'a=b'), ('Set-Cookie', 'c=d')] [('Set-Cookie', 'a=b'), ('Set-Cookie', 'c=d')]
@ -23,10 +24,17 @@ r"""
>>> rewrite_cookie('abc@def=123') >>> rewrite_cookie('abc@def=123')
[] []
# ExactCookieRewriter
>>> rewrite_cookie('some=value; Path=/diff/path/;', urlrewriter, ExactPathCookieRewriter)
[('Set-Cookie', 'some=value')]
>>> rewrite_cookie('some=value; Domain=.example.com; Path=/diff/path/; Max-Age=1500', urlrewriter, ExactPathCookieRewriter)
[('Set-Cookie', 'some=value')]
""" """
from pywb.rewrite.cookie_rewriter import MinimalScopeCookieRewriter from pywb.rewrite.cookie_rewriter import MinimalScopeCookieRewriter, ExactPathCookieRewriter
from pywb.rewrite.url_rewriter import UrlRewriter from pywb.rewrite.url_rewriter import UrlRewriter
urlrewriter = UrlRewriter('20131226101010/http://example.com/some/path/index.html', '/pywb/') urlrewriter = UrlRewriter('20131226101010/http://example.com/some/path/index.html', '/pywb/')
@ -34,6 +42,6 @@ urlrewriter = UrlRewriter('20131226101010/http://example.com/some/path/index.htm
urlrewriter2 = UrlRewriter('em_/http://example.com/', '/preview/') urlrewriter2 = UrlRewriter('em_/http://example.com/', '/preview/')
def rewrite_cookie(cookie_str, rewriter=urlrewriter): def rewrite_cookie(cookie_str, rewriter=urlrewriter, cookie_rewriter=MinimalScopeCookieRewriter):
return MinimalScopeCookieRewriter(rewriter).rewrite(cookie_str) return cookie_rewriter(rewriter).rewrite(cookie_str)