diff --git a/pywb/rewrite/test/test_rewrite_live.py b/pywb/rewrite/test/test_rewrite_live.py index 18b56807..6fc47b63 100644 --- a/pywb/rewrite/test/test_rewrite_live.py +++ b/pywb/rewrite/test/test_rewrite_live.py @@ -174,6 +174,19 @@ def test_local_2_no_rewrite(): # still link rewrite in HTML assert '"/pywb/20131226101010/http://example.com/some/path/another.html"' in buff +def test_local_unclosed_script(): + status_headers, buff = get_rewritten(get_test_dir() + 'text_content/sample_unclosed_script.html', + urlrewriter, + head_insert_func, + 'com,example,test)/') + + # wombat insert added + assert '' in buff, buff + + # JS location and JS link rewritten + assert 'window.WB_wombat_location = "/pywb/20131226101010/http:\/\/example.com/dynamic_page.html";\n}\n' in buff, buff + + def test_example_1(): status_headers, buff = get_rewritten('http://example.com/', urlrewriter, req_headers={'Connection': 'close'})