Override document.write too

googlearchive · Apr 26, 2013 · 733b350 · 733b350
1 parent 07be1f3
commit 733b350
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/wrappers/Document.js b/src/wrappers/Document.js
@@ -109,6 +109,7 @@
     'createTextNode',
     'elementFromPoint',
     'getElementById',
+    'write',
   ]);
 
   mixin(Document.prototype, GetElementsByInterface);

diff --git a/test/document-write.html b/test/document-write.html
@@ -4,11 +4,14 @@
 <script src="../shadowdom.js"></script>
 <a></a><script>
 var assert = chai.assert;
+var wrap = ShadowDOMPolyfill.wrap;
 
 var a = document.querySelector('a');
-document.write('<b></b>');
+document.write('<script><' + '/script>');
+wrap(document).write('<b></b>');
 assert.equal(a.nextSibling.localName, 'script');
-assert.equal(a.nextSibling.nextSibling.localName, 'b');
+assert.equal(a.nextSibling.nextSibling.localName, 'script');
+assert.equal(a.nextSibling.nextSibling.nextSibling.localName, 'b');
 
 var all = document.querySelectorAll('*');
 var last = all[all.length - 1];