reuse identical logic in 3 places

Author: archmoj

tasks/test_mock.js

@@ -1,17 +1,9 @@
 var minimist = require('minimist');
 var path = require('path');
 var fs = require('fs');
-var JSDOM = require('jsdom').JSDOM;
 
-var window = new JSDOM('', {runScripts: 'dangerously'}).window;
-
-// Mock a few things that jsdom doesn't support out-of-the-box
-window.URL.createObjectURL = function() {};
-
-var scriptEl = window.document.createElement('script');
-scriptEl.textContent = fs.readFileSync('build/plotly.js', { encoding: 'utf-8' });
-window.document.body.appendChild(scriptEl);
-var Plotly = window.Plotly;
+var plotlyNode = require('./util/plotly_node');
+var Plotly = plotlyNode('build/plotly.js');
 
 var pathToRoot = path.join(__dirname, '..');
 var pathToMocks = path.join(pathToRoot, 'test', 'image', 'mocks');

tasks/test_plain_obj.js

@@ -1,15 +1,5 @@
-var fs = require('fs');
-var JSDOM = require('jsdom').JSDOM;
-
-var window = new JSDOM('', {runScripts: 'dangerously'}).window;
-
-// Mock a few things that jsdom doesn't support out-of-the-box
-window.URL.createObjectURL = function() {};
-
-var scriptEl = window.document.createElement('script');
-scriptEl.textContent = fs.readFileSync('dist/plotly.js', { encoding: 'utf-8' });
-window.document.body.appendChild(scriptEl);
-var Plotly = window.Plotly;
+var plotlyNode = require('./util/plotly_node');
+var Plotly = plotlyNode('build/plotly.js');
 
 var assertValidate = function(fig, exp) {
     console.log(fig);

tasks/util/make_schema.js

@@ -1,18 +1,10 @@
 var fs = require('fs');
 var path = require('path');
-var JSDOM = require('jsdom').JSDOM;
-
-var window = new JSDOM('', {runScripts: 'dangerously'}).window;
-
-// Mock a few things that jsdom doesn't support out-of-the-box
-window.URL.createObjectURL = function() {};
+var plotlyNode = require('./plotly_node');
 
 module.exports = function makeSchema(plotlyPath, schemaPath) {
     return function() {
-        var scriptEl = window.document.createElement('script');
-        scriptEl.textContent = fs.readFileSync(plotlyPath, { encoding: 'utf-8' });
-        window.document.body.appendChild(scriptEl);
-        var Plotly = window.Plotly;
+        var Plotly = plotlyNode(plotlyPath);
 
         var plotSchema = Plotly.PlotSchema.get();
         var plotSchemaStr = JSON.stringify(plotSchema, null, 1);

tasks/util/plotly_node.js

@@ -0,0 +1,18 @@
+var fs = require('fs');
+var JSDOM = require('jsdom').JSDOM;
+
+var window = new JSDOM('', {
+    runScripts: 'dangerously'
+}).window;
+
+// Mock things that jsdom doesn't support out-of-the-box
+window.URL.createObjectURL = function() {};
+
+module.exports = function plotlyNode(plotlyPath) {
+    // Execute source code by inserting a <script> tag containing it
+    var scriptEl = window.document.createElement('script');
+    scriptEl.textContent = fs.readFileSync(plotlyPath, { encoding: 'utf-8' });
+    window.document.body.appendChild(scriptEl);
+
+    return window.Plotly;
+};