first pass js jupyter tests

etpinard · theengineear · commit eba8bd7590d3 · 2017-01-05T10:08:59.000-08:00
diff --git a/plotly/tests/test_core/test_jupyter/.gitignore b/plotly/tests/test_core/test_jupyter/.gitignore
@@ -0,0 +1,2 @@
+node_modules
+fixtures/*.html
diff --git a/plotly/tests/test_core/test_jupyter/lib/server.js b/plotly/tests/test_core/test_jupyter/lib/server.js
@@ -0,0 +1,93 @@
+var http = require('http');
+var url = require('url');
+var fs = require('fs');
+var path = require('path');
+var ecstatic = require('ecstatic');
+var browserify = require('browserify');
+var cheerio = require('cheerio');
+var chrome = require('chrome-launch');
+var tapParser = require('tap-parser');
+
+var PORT = 8080;
+var PATH_ROOT = path.join(__dirname, '..');
+var PATH_FIXTURES = path.join(PATH_ROOT, 'fixtures');
+var PATH_INDEX = path.join(PATH_FIXTURES, 'test.html');
+var PATH_INDEX_STUB = path.join(PATH_FIXTURES, 'test.tmp.html');
+var PATH_TEST_FILE = path.join(PATH_ROOT, 'test.js');
+var PATH_TEST_BUNDLE = path.join(PATH_ROOT, 'test.tmp.js');
+var URL = 'http://localhost:' + PORT + '/fixtures/test.tmp.html';
+var EXIT_CODE = 0;
+
+// main
+stubIndex()
+    .then(bundleTests)
+    .then(startServer)
+    .then(launch)
+
+function stubIndex() {
+    return new Promise(function(resolve, reject) {
+        var html = fs.readFileSync(PATH_INDEX, 'utf-8');
+        var $ = cheerio.load(html);
+
+        $('body').append('<script type="text/javascript" src="../test.tmp.js"></script>');
+
+        fs.writeFile(PATH_INDEX_STUB, $.html(), resolve);
+    });
+}
+
+function bundleTests() {
+    return new Promise(function(resolve, reject) {
+        var wsBundle = fs.createWriteStream(PATH_TEST_BUNDLE);
+
+        browserify(PATH_TEST_FILE, { debug: true })
+            .bundle()
+            .pipe(wsBundle);
+
+        wsBundle.on('close', resolve);
+    });
+}
+
+function startServer() {
+    return new Promise(function(resolve, reject) {
+        var server = http.createServer(ecstatic({ root: PATH_ROOT }));
+
+        server.on('request', handle);
+
+        server.listen(PORT, resolve);
+    });
+}
+
+function handle(req, res) {
+    var query = url.parse(req.url).query || '';
+    var parser = tapParser();
+
+    function is(query, root) {
+        return query.indexOf(root) !== -1;
+    }
+
+    if(is(query, 'data')) handleData(req, res);
+    if(is(query, 'done')) handleDone();
+
+    function handleData(req, res) {
+        req.pipe(parser);
+        req.pipe(process.stdout);
+    }
+
+    parser.on('assert', function(assert) {
+        if(EXIT_CODE === 0 && assert.ok === false) EXIT_CODE = 1;
+    })
+
+    function handleDone() {
+        removeBuildFiles();
+        process.exit(EXIT_CODE);
+    }
+}
+
+function launch() {
+    chrome(URL);
+}
+
+function removeBuildFiles() {
+    fs.unlinkSync(PATH_INDEX_STUB);
+    fs.unlinkSync(PATH_TEST_BUNDLE);
+}
diff --git a/plotly/tests/test_core/test_jupyter/lib/tape-wrapper.js b/plotly/tests/test_core/test_jupyter/lib/tape-wrapper.js
@@ -0,0 +1,22 @@
+var test = require('tape');
+var xhr = require('xhr');
+
+var cnt = 0;
+var noop = function() {};
+
+var post = function(query, data) {
+    var opts = data ? { body: data } : {};
+    xhr.post('/?' + query + '&' + (cnt++), opts, noop);
+};
+
+var ws = test.createStream();
+
+ws.on('data', function(data) {
+    post('data', data)
+});
+
+test.onFinish(function() {
+    post('done');
+});
+
+module.exports = test;
diff --git a/plotly/tests/test_core/test_jupyter/package.json b/plotly/tests/test_core/test_jupyter/package.json
@@ -0,0 +1,19 @@
+{
+  "name": "plotly.py-jupyter-tests",
+  "version": "1.0.0",
+  "description": "js deps for plotly.py jupyter tests",
+  "scripts": {
+    "test": "node lib/server.js"
+  },
+  "author": "Plotly Inc.",
+  "license": "MIT",
+  "dependencies": {
+    "browserify": "^13.1.0",
+    "cheerio": "^0.20.0",
+    "chrome-launch": "^1.1.4",
+    "ecstatic": "^2.1.0",
+    "tap-parser": "^2.0.0",
+    "tape": "^4.6.0",
+    "xhr": "^2.2.2"
+  }
+}
diff --git a/plotly/tests/test_core/test_jupyter/test.js b/plotly/tests/test_core/test_jupyter/test.js
@@ -0,0 +1,15 @@
+var test = require('./lib/tape-wrapper');
+
+test('should have the correct number of script tags', function(t) {
+    t.plan(1);
+
+    var nodes = document.querySelectorAll('script');
+    t.equal(nodes.length, 8);
+});
+
+test('should have one plotly.js graph', function(t) {
+    t.plan(1);
+
+    var nodes = document.querySelectorAll('.js-plotly-plot');
+    t.equal(nodes.length, 1);
+});
diff --git a/plotly/tests/test_core/test_jupyter/test_jupyter.py b/plotly/tests/test_core/test_jupyter/test_jupyter.py
@@ -0,0 +1,46 @@
+"""
+test__jupyter
+
+"""
+from __future__ import absolute_import
+
+import nbformat
+from nbconvert import HTMLExporter
+from nbconvert.preprocessors import ExecutePreprocessor
+
+from unittest import TestCase
+from os import path
+import subprocess
+
+import plotly
+
+PATH_ROOT = path.dirname(__file__)
+PATH_FIXTURES = path.join(PATH_ROOT, 'fixtures')
+PATH_TEST_NB = path.join(PATH_FIXTURES, 'test.ipynb')
+PATH_TEST_HTML = path.join(PATH_FIXTURES, 'test.html')
+
+
+class PlotlyJupyterTestCase(TestCase):
+    def setUp(self):
+        with open(PATH_TEST_NB, 'r') as f:
+            self.nb = nbformat.read(f, as_version=4)
+
+        self.ep = ExecutePreprocessor(timeout=600)
+        self.html_exporter = HTMLExporter()
+
+        self.ep.preprocess(self.nb, {'metadata': {'path': '.'}})
+        (self.body, _) = self.html_exporter.from_notebook_node(self.nb)
+
+        with open(PATH_TEST_HTML, 'w') as f:
+            f.write(self.body)
+
+    def test_one(self):
+        proc = subprocess.Popen(['npm', 'test'],
+                                cwd=PATH_ROOT,
+                                stdout=subprocess.PIPE,
+                                stderr=subprocess.PIPE)
+
+        (_, stderr) = proc.communicate()
+
+        if stderr:
+            self.fail('One or more javascript test failed')
