From 676f9717496772cb020e3c19ccb6af4ed6135e03 Mon Sep 17 00:00:00 2001
From: Devon A Thomson <devon.thomson@hotmail.com>
Date: Thu, 14 Jan 2021 18:34:09 -0500
Subject: [PATCH] Visualize Save with Dashboard Picker

---
 .../saved_object_save_modal_dashboard.tsx     |  8 ++---
 .../functional/page_objects/visualize_page.ts | 34 +++++++++++++++++--
 2 files changed, 35 insertions(+), 7 deletions(-)

diff --git a/src/plugins/presentation_util/public/components/saved_object_save_modal_dashboard.tsx b/src/plugins/presentation_util/public/components/saved_object_save_modal_dashboard.tsx
index aafa500002ae2..04700e9379840 100644
--- a/src/plugins/presentation_util/public/components/saved_object_save_modal_dashboard.tsx
+++ b/src/plugins/presentation_util/public/components/saved_object_save_modal_dashboard.tsx
@@ -99,11 +99,11 @@ export function SavedObjectSaveModalDashboard(props: DashboardSaveModalProps) {
           }
           hasChildLabel={false}
         >
-          <EuiPanel color="subdued" hasShadow={false}>
+          <EuiPanel color="subdued" hasShadow={false} data-test-subj="add-to-dashboard-options">
             <div>
               <EuiRadio
                 checked={dashboardOption === 'existing'}
-                id="existing"
+                id="existing-dashboard-option"
                 name="dashboard-option"
                 label={i18n.translate(
                   'presentationUtil.saveModalDashboard.existingDashboardOptionLabel',
@@ -129,7 +129,7 @@ export function SavedObjectSaveModalDashboard(props: DashboardSaveModalProps) {
 
               <EuiRadio
                 checked={dashboardOption === 'new'}
-                id="new"
+                id="new-dashboard-option"
                 name="dashboard-option"
                 label={i18n.translate(
                   'presentationUtil.saveModalDashboard.newDashboardOptionLabel',
@@ -145,7 +145,7 @@ export function SavedObjectSaveModalDashboard(props: DashboardSaveModalProps) {
 
               <EuiRadio
                 checked={dashboardOption === null}
-                id="library"
+                id="add-to-library-option"
                 name="dashboard-option"
                 label={i18n.translate('presentationUtil.saveModalDashboard.libraryOptionLabel', {
                   defaultMessage: 'No dashboard, but add to library',
diff --git a/test/functional/page_objects/visualize_page.ts b/test/functional/page_objects/visualize_page.ts
index d8329f492fe80..14160cc106261 100644
--- a/test/functional/page_objects/visualize_page.ts
+++ b/test/functional/page_objects/visualize_page.ts
@@ -20,6 +20,13 @@
 import { FtrProviderContext } from '../ftr_provider_context';
 import { VisualizeConstants } from '../../../src/plugins/visualize/public/application/visualize_constants';
 
+interface VisualizeSaveModalArgs {
+  saveAsNew?: boolean;
+  redirectToOrigin?: boolean;
+  addToDashboard?: boolean;
+  dashboardId?: string;
+}
+
 export function VisualizePageProvider({ getService, getPageObjects }: FtrProviderContext) {
   const testSubjects = getService('testSubjects');
   const retry = getService('retry');
@@ -348,7 +355,7 @@ export function VisualizePageProvider({ getService, getPageObjects }: FtrProvide
 
     public async saveVisualization(
       vizName: string,
-      { saveAsNew = false, redirectToOrigin = false } = {}
+      { saveAsNew, redirectToOrigin, addToDashboard, dashboardId }: VisualizeSaveModalArgs = {}
     ) {
       await this.ensureSavePanelOpen();
       await testSubjects.setValue('savedObjectTitle', vizName);
@@ -366,6 +373,22 @@ export function VisualizePageProvider({ getService, getPageObjects }: FtrProvide
         log.debug('redirect to origin checkbox exists. Setting its state to', state);
         await testSubjects.setEuiSwitch('returnToOriginModeSwitch', state);
       }
+
+      const dashboardSelectorExists = await testSubjects.exists('add-to-dashboard-options');
+      if (dashboardSelectorExists) {
+        const dashboardSelector = await testSubjects.find('add-to-dashboard-options');
+        let optionSelector = 'add-to-library-option';
+        if (addToDashboard) {
+          optionSelector = dashboardId ? 'existing-dashboard-option' : 'new-dashboard-option';
+        }
+        log.debug('dashboard selector exists, choosing option:', optionSelector);
+        const label = await dashboardSelector.findByCssSelector(`label[for="${optionSelector}"]`);
+        await label.click();
+
+        if (dashboardId) {
+          // TODO - selecting an existing dashboard
+        }
+      }
       log.debug('Click Save Visualization button');
 
       await testSubjects.click('confirmSaveSavedObjectButton');
@@ -381,9 +404,14 @@ export function VisualizePageProvider({ getService, getPageObjects }: FtrProvide
 
     public async saveVisualizationExpectSuccess(
       vizName: string,
-      { saveAsNew = false, redirectToOrigin = false } = {}
+      { saveAsNew, redirectToOrigin, addToDashboard, dashboardId }: VisualizeSaveModalArgs = {}
     ) {
-      const saveMessage = await this.saveVisualization(vizName, { saveAsNew, redirectToOrigin });
+      const saveMessage = await this.saveVisualization(vizName, {
+        saveAsNew,
+        redirectToOrigin,
+        addToDashboard,
+        dashboardId,
+      });
       if (!saveMessage) {
         throw new Error(
           `Expected saveVisualization to respond with the saveMessage from the toast, got ${saveMessage}`