From 127e8e67dd54c8abc5975d09a57b58754cf53ca8 Mon Sep 17 00:00:00 2001
From: LeeDr <lee.drengenberg@elastic.co>
Date: Tue, 19 Jan 2016 12:20:09 -0600
Subject: [PATCH] Fix screenshots Data Table and Tile Map.

---
 .../apps/visualize/_vertical_bar_chart.js     | 20 ++++++++-----------
 1 file changed, 8 insertions(+), 12 deletions(-)

diff --git a/test/functional/apps/visualize/_vertical_bar_chart.js b/test/functional/apps/visualize/_vertical_bar_chart.js
index f21fc8cc7cdb..88ee497427da 100644
--- a/test/functional/apps/visualize/_vertical_bar_chart.js
+++ b/test/functional/apps/visualize/_vertical_bar_chart.js
@@ -86,15 +86,11 @@ define(function (require) {
       });
 
       bdd.describe('vertical bar chart', function indexPatternCreation() {
+        var testSubName = 'VerticalBarChart';
+        var vizName1 = 'Visualization ' + testSubName;
 
 
-        bdd.it('should save and load, take screenshot', function pageHeader() {
-
-          var testSubName = 'VerticalBarChart';
-          common.debug('Start of test' + testSubName + 'Visualization');
-          var vizName1 = 'Visualization ' + testSubName;
-          // var remote = this.remote;
-
+        bdd.it('should save and load', function pageHeader() {
           return visualizePage.saveVisualization(vizName1)
           .then(function (message) {
             common.debug('Saved viz message = ' + message);
@@ -112,15 +108,11 @@ define(function (require) {
           .then(function waitForVisualization() {
             return visualizePage.waitForVisualization();
           })
-          .then(function takeScreenshot() {
-            common.debug('Take screenshot');
-            common.saveScreenshot('./screenshot-' + testSubName + '.png');
-          })
           .catch(common.handleError(this));
         });
 
 
-        bdd.it('should show correct chart', function pageHeader() {
+        bdd.it('should show correct chart, take screenshot', function pageHeader() {
 
           var expectedChartValues = [37, 202, 740, 1437, 1371, 751, 188, 31, 42, 202, 683,
             1361, 1415, 707, 177, 27, 32, 175, 707, 1408, 1355, 726, 201, 29
@@ -138,6 +130,10 @@ define(function (require) {
             common.debug('data.length=' + data.length);
             expect(data).to.eql(expectedChartValues);
           })
+          .then(function takeScreenshot() {
+            common.debug('Take screenshot');
+            common.saveScreenshot('./screenshot-' + testSubName + '.png');
+          })
           .catch(common.handleError(this));
         });