Update internal Rollup plugin removing from output with Rollup 1.0 ch…

…anges (ampproject#225)
Zako77xx · Jan 10, 2019 · 160e319 · 160e319
1 parent 24976fe
commit 160e319
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 14 deletions.
diff --git a/config/rollup.plugins.js b/config/rollup.plugins.js
@@ -69,23 +69,20 @@ export function removeTestingDocument() {
     buildStart() {
       context = this;
     },
-    transformChunk: async (code) => {
+    renderChunk: async (code) => {
       const source = new MagicString(code);
+      const program = context.parse(code, { ranges: true });
 
-      if (context) {
-        const program = context.parse(code, { ranges: true });
-
-        walk.simple(program, {
-          VariableDeclarator(node) {
-            if (node.id && node.id.type === 'Identifier' && node.id.name && node.id.name === 'documentForTesting') {
-              const range = node.range;
-              if (range) {
-                source.overwrite(node.range[0], node.range[1], 'documentForTesting = undefined');
-              }
+      walk.simple(program, {
+        VariableDeclarator(node) {
+          if (node.id && node.id.type === 'Identifier' && node.id.name && node.id.name === 'documentForTesting') {
+            const range = node.range;
+            if (range) {
+              source.overwrite(node.range[0], node.range[1], 'documentForTesting = undefined');
             }
-          },
-        });
-      }
+          }
+        },
+      });
 
       return {
         code: source.toString(),

diff --git a/src/worker-thread/dom/Document.ts b/src/worker-thread/dom/Document.ts
@@ -134,3 +134,6 @@ export function createDocument(postMessageMethod?: Function): Document {
 
   return doc;
 }
+
+/** Should only be used for testing. */
+export const documentForTesting = createDocument();