From b8b8c30573f2175d2280f80893b1f4043f6bd728 Mon Sep 17 00:00:00 2001 From: Ilya Kreymer Date: Thu, 13 Nov 2014 09:43:50 -0800 Subject: [PATCH] cookie_rewriter: add tests for exact cookie rewriter --- pywb/rewrite/test/test_cookie_rewriter.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/pywb/rewrite/test/test_cookie_rewriter.py b/pywb/rewrite/test/test_cookie_rewriter.py index c20f56f9..4f57464f 100644 --- a/pywb/rewrite/test/test_cookie_rewriter.py +++ b/pywb/rewrite/test/test_cookie_rewriter.py @@ -1,4 +1,5 @@ r""" +# Default -- MinimalScopeRewriter # No rewriting >>> rewrite_cookie('a=b; c=d;') [('Set-Cookie', 'a=b'), ('Set-Cookie', 'c=d')] @@ -23,10 +24,17 @@ r""" >>> 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 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/') -def rewrite_cookie(cookie_str, rewriter=urlrewriter): - return MinimalScopeCookieRewriter(rewriter).rewrite(cookie_str) +def rewrite_cookie(cookie_str, rewriter=urlrewriter, cookie_rewriter=MinimalScopeCookieRewriter): + return cookie_rewriter(rewriter).rewrite(cookie_str)