From effd618bb322d0f3a32c64fafc179ac687465358 Mon Sep 17 00:00:00 2001 From: Ilya Kreymer Date: Thu, 10 Mar 2016 10:10:51 -0800 Subject: [PATCH] tests: add parse_comment test for html_rewriter --- pywb/rewrite/test/test_html_rewriter.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/pywb/rewrite/test/test_html_rewriter.py b/pywb/rewrite/test/test_html_rewriter.py index 002529e6..5f279d87 100644 --- a/pywb/rewrite/test/test_html_rewriter.py +++ b/pywb/rewrite/test/test_html_rewriter.py @@ -159,6 +159,14 @@ ur""" >>> parse('Some Text without any tags ', head_insert = '') Some Text without any tags +# no parse comments +>>> parse('') + + +# with parse comments +>>> parse('', parse_comments=True) + + # rel=canonical: rewrite (default) >>> parse('') @@ -236,8 +244,10 @@ urlrewriter_pencode = new_rewriter(rewrite_opts=dict(punycode_links=True)) no_base_canon_rewriter = new_rewriter(rewrite_opts=dict(rewrite_rel_canon=False, rewrite_base=False)) -def parse(data, head_insert=None, urlrewriter=urlrewriter): - parser = HTMLRewriter(urlrewriter, head_insert = head_insert, url = ORIGINAL_URL) +def parse(data, head_insert=None, urlrewriter=urlrewriter, parse_comments=False): + parser = HTMLRewriter(urlrewriter, head_insert=head_insert, + url=ORIGINAL_URL, + parse_comments=parse_comments) if isinstance(data, unicode): data = data.encode('utf-8')