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')