From 41cba71db2249b74a077b4285febcc2200fa7fea Mon Sep 17 00:00:00 2001
From: Levko Kravets <levko.ne@gmail.com>
Date: Thu, 17 Oct 2019 11:16:40 +0300
Subject: [PATCH] Update tests

---
 client/app/pages/queries/query.html                |  7 ++++++-
 .../visualizations/sankey_sunburst_spec.js         | 14 ++++++++++++++
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/client/app/pages/queries/query.html b/client/app/pages/queries/query.html
index 97148729b9..15ecb22e33 100644
--- a/client/app/pages/queries/query.html
+++ b/client/app/pages/queries/query.html
@@ -42,7 +42,12 @@ <h3>
               <a ng-show="!sourceMode" ng-href="{{query.getUrl(true, selectedTab)}}" class="btn btn-default btn--showhide">
                 <i class="fa fa-pencil-square-o" aria-hidden="true"></i> Edit Source</i>
               </a>
-              <a ng-show="sourceMode" ng-href="{{query.getUrl(false, selectedTab)}}" class="btn btn-default btn--showhide">
+              <a
+                ng-show="sourceMode"
+                ng-href="{{query.getUrl(false, selectedTab)}}"
+                class="btn btn-default btn--showhide"
+                data-test="QueryPageShowDataOnly"
+              >
                 <i class="fa fa-table" aria-hidden="true"></i> Show Data Only</i>
               </a>
             </span>
diff --git a/client/cypress/integration/visualizations/sankey_sunburst_spec.js b/client/cypress/integration/visualizations/sankey_sunburst_spec.js
index 2d3a70db42..aa536ff19c 100644
--- a/client/cypress/integration/visualizations/sankey_sunburst_spec.js
+++ b/client/cypress/integration/visualizations/sankey_sunburst_spec.js
@@ -17,6 +17,8 @@ const SQL = `
 `;
 
 describe('Sankey and Sunburst', () => {
+  const viewportWidth = Cypress.config('viewportWidth');
+
   beforeEach(() => {
     cy.login();
     createQuery({ query: SQL }).then(({ id }) => {
@@ -35,6 +37,12 @@ describe('Sankey and Sunburst', () => {
     cy.getByTestId('VisualizationPreview').find('svg').should('exist');
     cy.getByTestId('EditVisualizationDialog').contains('button', 'Save').click();
     cy.getByTestId('QueryPageVisualizationTabs').contains('li', visualizationName).should('exist');
+
+    cy.getByTestId('QueryPageShowDataOnly').click();
+
+    // wait a bit before taking snapshot
+    cy.wait(1000); // eslint-disable-line cypress/no-unnecessary-waiting
+    cy.percySnapshot('Visualizations - Sunburst', { widths: [viewportWidth] });
   });
 
   it('creates Sankey', () => {
@@ -47,5 +55,11 @@ describe('Sankey and Sunburst', () => {
     cy.getByTestId('VisualizationPreview').find('svg').should('exist');
     cy.getByTestId('EditVisualizationDialog').contains('button', 'Save').click();
     cy.getByTestId('QueryPageVisualizationTabs').contains('li', visualizationName).should('exist');
+
+    cy.getByTestId('QueryPageShowDataOnly').click();
+
+    // wait a bit before taking snapshot
+    cy.wait(1000); // eslint-disable-line cypress/no-unnecessary-waiting
+    cy.percySnapshot('Visualizations - Sankey', { widths: [viewportWidth] });
   });
 });