From 20483ffe59432446486163349a61897858f3136e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Etienne=20T=C3=A9treault-Pinard?= <etienne@plot.ly>
Date: Fri, 4 Aug 2017 10:58:10 -0400
Subject: [PATCH] fix toSVG for marker gradient legend items

---
 src/snapshot/tosvg.js               |  2 +-
 test/jasmine/tests/snapshot_test.js | 11 ++++++++---
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/snapshot/tosvg.js b/src/snapshot/tosvg.js
index 85b4cd280b1..70308d112e1 100644
--- a/src/snapshot/tosvg.js
+++ b/src/snapshot/tosvg.js
@@ -113,7 +113,7 @@ module.exports = function toSVG(gd, format) {
             }
         });
 
-    svg.selectAll('.point').each(function() {
+    svg.selectAll('.point,.scatterpts').each(function() {
         var pt = d3.select(this);
         var fill = pt.style('fill');
 
diff --git a/test/jasmine/tests/snapshot_test.js b/test/jasmine/tests/snapshot_test.js
index 12cc6ac3b64..caf86aca991 100644
--- a/test/jasmine/tests/snapshot_test.js
+++ b/test/jasmine/tests/snapshot_test.js
@@ -260,7 +260,8 @@ describe('Plotly.Snapshot', function() {
                     color: ['red', 'blue', 'green']
                 }
             }], {
-                font: { family: 'Times New Roman' }
+                font: { family: 'Times New Roman' },
+                showlegend: true
             })
             .then(function() {
                 d3.selectAll('text').each(function() {
@@ -268,7 +269,7 @@ describe('Plotly.Snapshot', function() {
                     expect(tx.style('font-family')).toEqual('\"Times New Roman\"');
                 });
 
-                d3.selectAll('.point').each(function() {
+                d3.selectAll('.point,.scatterpts').each(function() {
                     var pt = d3.select(this);
                     expect(pt.style('fill').substr(0, 6)).toEqual('url(\"#');
                 });
@@ -280,7 +281,7 @@ describe('Plotly.Snapshot', function() {
                 var i;
 
                 var textElements = svgDOM.getElementsByTagName('text');
-                expect(textElements.length).toEqual(11);
+                expect(textElements.length).toEqual(12);
 
                 for(i = 0; i < textElements.length; i++) {
                     expect(textElements[i].style['font-family']).toEqual('\"Times New Roman\"');
@@ -292,6 +293,10 @@ describe('Plotly.Snapshot', function() {
                 for(i = 0; i < pointElements.length; i++) {
                     expect(pointElements[i].style.fill.substr(0, 6)).toEqual('url(\"#');
                 }
+
+                var legendPointElements = svgDOM.getElementsByClassName('scatterpts');
+                expect(legendPointElements.length).toEqual(1);
+                expect(legendPointElements[0].style.fill.substr(0, 6)).toEqual('url(\"#');
             })
             .catch(fail)
             .then(done);