treemap jasmine tests

Author: archmoj

tasks/test_syntax.js

@@ -140,12 +140,15 @@ function assertSrcContents() {
                     } else if(IE_BLACK_LIST.indexOf(lastPart) !== -1) {
                         logs.push(file + ' : contains .' + lastPart + ' (IE failure)');
                     } else if(IE_SVG_BLACK_LIST.indexOf(lastPart) !== -1) {
-                        // add special case for sunburst where we use 'children'
+                        // add special case for sunburst and treemap where we use 'children'
                         // off the d3-hierarchy output
                         var dirParts = path.dirname(file).split(path.sep);
-                        var isSunburstFile = dirParts[dirParts.length - 1] === 'sunburst';
-                        var isLinkedToObject = ['pt', 'd', 'parent'].indexOf(parts[parts.length - 2]) !== -1;
-                        if(!(isSunburstFile && isLinkedToObject)) {
+                        var filename = dirParts[dirParts.length - 1];
+                        var isSunburstOrTreemap =
+                            filename === 'sunburst' ||
+                            filename === 'treemap';
+                        var isLinkedToObject = ['pt', 'd', 'parent', 'node'].indexOf(parts[parts.length - 2]) !== -1;
+                        if(!(isSunburstOrTreemap && isLinkedToObject)) {
                             logs.push(file + ' : contains .' + lastPart + ' (IE failure in SVG)');
                         }
                     } else if(FF_BLACK_LIST.indexOf(lastPart) !== -1) {

test/jasmine/assets/mock_lists.js

@@ -33,6 +33,7 @@ var svgMockList = [
     ['range_slider_multiple', require('@mocks/range_slider_multiple.json')],
     ['sankey_energy', require('@mocks/sankey_energy.json')],
     ['sunburst_coffee', require('@mocks/sunburst_coffee.json')],
+    ['treemap_coffee', require('@mocks/treemap_coffee.json')],
     ['parcats_bad-displayindex', require('@mocks/parcats_bad-displayindex.json')],
     ['scattercarpet', require('@mocks/scattercarpet.json')],
     ['shapes', require('@mocks/shapes.json')],

test/jasmine/tests/plot_api_react_test.js

@@ -1901,6 +1901,42 @@ describe('Plotly.react and uirevision attributes', function() {
         .catch(failTest)
         .then(done);
     });
+
+    it('preserves treemap level changes', function(done) {
+        function assertLevel(msg, exp) {
+            expect(gd._fullData[0].level).toBe(exp, msg);
+        }
+
+        Plotly.react(gd, [{
+            type: 'treemap',
+            labels: ['Eve', 'Cain', 'Seth', 'Enos', 'Noam', 'Abel', 'Awan', 'Enoch', 'Azura'],
+            parents: ['', 'Eve', 'Eve', 'Seth', 'Seth', 'Eve', 'Eve', 'Awan', 'Eve'],
+            uirevision: 1
+        }])
+        .then(function() {
+            assertLevel('no set level at start', undefined);
+        })
+        .then(function() {
+            var nodeSeth = d3.select('.slice:nth-child(2)').node();
+            mouseEvent('click', 0, 0, {element: nodeSeth});
+        })
+        .then(function() {
+            assertLevel('after clicking on Seth sector', 'Seth');
+        })
+        .then(function() {
+            return Plotly.react(gd, [{
+                type: 'treemap',
+                labels: ['Eve', 'Cain', 'Seth', 'Enos', 'Noam', 'Abel', 'Awan', 'Enoch', 'Azura', 'Joe'],
+                parents: ['', 'Eve', 'Eve', 'Seth', 'Seth', 'Eve', 'Eve', 'Awan', 'Eve', 'Seth'],
+                uirevision: 1
+            }]);
+        })
+        .then(function() {
+            assertLevel('after reacting with new data, but with same uirevision', 'Seth');
+        })
+        .catch(failTest)
+        .then(done);
+    });
 });
 
 describe('Test Plotly.react + interactions under uirevision:', function() {