sankey: test to 🔒 down modebar button that resets view

Author: antoinerg

test/jasmine/tests/sankey_test.js

@@ -23,6 +23,7 @@ var defaultColors = require('@src/components/color/attributes').defaults;
 var drag = require('../assets/drag');
 var checkOverlap = require('../assets/check_overlap');
 var delay = require('../assets/delay');
+var selectButton = require('../assets/modebar_button');
 
 describe('sankey tests', function() {
     'use strict';
@@ -560,6 +561,52 @@ describe('sankey tests', function() {
             .catch(failTest)
             .then(done);
         });
+
+        it('resets each subplot to its initial view (ie. x, y groups) via modebar button', function(done) {
+            var mockCopy = Lib.extendDeep({}, require('@mocks/sankey_subplots_circular'));
+
+            // Set initial view
+            mockCopy.data[0].node.x = [0.25];
+            mockCopy.data[0].node.y = [0.25];
+
+            mockCopy.data[0].node.groups = [];
+            mockCopy.data[1].node.groups = [[2, 3]];
+
+            Plotly.plot(gd, mockCopy)
+            .then(function() {
+                expect(gd._fullData[0].node.groups).toEqual([]);
+                expect(gd._fullData[1].node.groups).toEqual([[2, 3]]);
+
+                // Change groups
+                return Plotly.restyle(gd, {
+                    'node.groups': [[[1, 2]], [[]]],
+                    'node.x': [[0.1]],
+                    'node.y': [[0.1]]
+                });
+            })
+            .then(function() {
+                // Check current state
+                expect(gd._fullData[0].node.x).toEqual([0.1]);
+                expect(gd._fullData[0].node.y).toEqual([0.1]);
+
+                expect(gd._fullData[0].node.groups).toEqual([[1, 2]]);
+                expect(gd._fullData[1].node.groups).toEqual([[]]);
+
+                // Click reset
+                var resetButton = selectButton(gd._fullLayout._modeBar, 'resetViewSankey');
+                resetButton.click();
+            })
+            .then(function() {
+                // Check we are back to initial view
+                expect(gd._fullData[0].node.x).toEqual([0.25]);
+                expect(gd._fullData[0].node.y).toEqual([0.25]);
+
+                expect(gd._fullData[0].node.groups).toEqual([]);
+                expect(gd._fullData[1].node.groups).toEqual([[2, 3]]);
+            })
+            .catch(failTest)
+            .then(done);
+        });
     });
 
     describe('Test hover/click interactions:', function() {