From 990af5ee793a6383a49055794e855ed61997e91f Mon Sep 17 00:00:00 2001
From: Ilya Kreymer <ikreymer@gmail.com>
Date: Tue, 31 Mar 2015 23:30:56 -0700
Subject: [PATCH] rewrite: add extra test for rewriting html with <script> tag
 that's never closed

---
 pywb/rewrite/test/test_rewrite_live.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)
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 '<head><script src="/static/__pywb/wombat.js"> </script>' in buff, buff
+
+    # JS location and JS link rewritten
+    assert 'window.WB_wombat_location = "/pywb/20131226101010/http:\/\/example.com/dynamic_page.html";\n}\n</script>' in buff, buff
+
+
 def test_example_1():
     status_headers, buff = get_rewritten('http://example.com/', urlrewriter, req_headers={'Connection': 'close'})
 
