Add FT for TSVB support with MDS

Signed-off-by: Emma Jin <[email protected]>

Author: amznyja

cypress/fixtures/dashboard/opensearch_dashboards/vis_type_tsvb/metrics.data.txt

@@ -0,0 +1,99 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { CURRENT_TENANT } from '../../../../../utils/commands';
+import { DS_BASIC_AUTH_LABEL } from '../../../../../utils/dashboards/datasource-management-dashboards-plugin/constants';
+import {
+  TSVB_INDEX_ID,
+  TSVB_PATH_INDEX_DATA,
+  TSVB_CREATE_URL,
+  VIS_APP_PATH,
+  TSVB_INDEX_START_TIME,
+  TSVB_INDEX_END_TIME,
+  TSVB_INDEX_PATTERN,
+  TSVB_VIS_TYPE,
+} from '../../../../../utils/dashboards/vis_type_tsvb/constants';
+
+describe('TSVB Visualization', () => {
+  before(() => {
+    CURRENT_TENANT.newTenant = 'global';
+    cy.fleshTenantSettings();
+    cy.deleteIndex(TSVB_INDEX_ID);
+    cy.bulkUploadDocs(TSVB_PATH_INDEX_DATA);
+
+    // Dashboards requires an index pattern to continue to the Create Visualization stage
+    cy.deleteIndexPattern(TSVB_INDEX_PATTERN);
+    cy.createIndexPattern(TSVB_INDEX_PATTERN, {
+      title: TSVB_INDEX_ID,
+      timeFieldName: 'timestamp',
+    });
+
+    cy.deleteSavedObjectByType(TSVB_VIS_TYPE, TSVB_INDEX_ID);
+
+    // Visit the page
+    cy.log('create a new tsvb visualization: ', TSVB_CREATE_URL);
+    cy.visit(TSVB_CREATE_URL);
+    cy.url().should('contain', VIS_APP_PATH);
+    cy.setTopNavDate(TSVB_INDEX_START_TIME, TSVB_INDEX_END_TIME);
+
+    // Wait for page to load
+    cy.waitForLoader();
+  });
+
+  if (Cypress.env('DATASOURCE_MANAGEMENT_ENABLED')) {
+    before(() => {
+      cy.deleteDataSourceIndexBasicAuth(TSVB_INDEX_ID);
+      cy.createDataSourceBasicAuth();
+      cy.bulkUploadDocsToDataSourceBasicAuth(TSVB_PATH_INDEX_DATA);
+    });
+
+    describe('When MDS is enabled', () => {
+      [
+        {
+          dataSourceName: DS_BASIC_AUTH_LABEL,
+          canvasExists: 'exist',
+        },
+      ].forEach(({ dataSourceName, canvasExists }) => {
+        it(`should query data from ${dataSourceName} and resulting visualization should ${canvasExists}`, () => {
+          cy.contains('button', 'Panel options').click();
+          cy.contains('label', 'Data source');
+          // Click on the dropdown to open it
+          cy.get('[data-test-subj="dataSourceSelectorComboBox"]').click();
+
+          // Find the option you want to select by its text and click on it
+          cy.contains('[role="option"]', 'RemoteDataSourceBasicAuth').click();
+          cy.get('input[data-test-subj="metricsIndexPatternInput"]').type(
+            TSVB_INDEX_ID
+          );
+          cy.tsvbSaveVisualization();
+
+          cy.get('canvas').should(canvasExists);
+        });
+      });
+    });
+  } else {
+    describe('When MDS is disabled', () => {
+      it('should query from local cluster', () => {
+        cy.contains('button', 'Panel options').click();
+        cy.get('input[data-test-subj="metricsIndexPatternInput"]').type(
+          TSVB_INDEX_ID
+        );
+        cy.tsvbSaveVisualization();
+        // Visualization should be drawn; correct visualizations do not have warning messages
+        cy.get('canvas').should('exist');
+      });
+    });
+  }
+
+  after(() => {
+    cy.deleteIndex(TSVB_INDEX_ID);
+    cy.deleteIndexPattern(TSVB_INDEX_PATTERN);
+
+    if (Cypress.env('DATASOURCE_MANAGEMENT_ENABLED')) {
+      cy.deleteDataSourceIndexBasicAuth(TSVB_INDEX_ID);
+      cy.deleteAllDataSources();
+    }
+  });
+});

cypress/integration/core-opensearch-dashboards/opensearch-dashboards/apps/vis_type_tsvb/tsvb.spec.js

@@ -8,6 +8,7 @@ import './vis_type_timeline/commands';
 import './vis_type_table/commands';
 import './vis_type_vega/commands';
 import './vis-augmenter/commands';
+import './vis_type_tsvb/commands';
 import './data_explorer/commands';
 
 Cypress.Commands.add('waitForLoader', () => {

cypress/utils/dashboards/commands.js

@@ -23,4 +23,8 @@ export * from './vis_type_table/constants';
 export * from './vis_type_timeline/constants';
 export * from './vis-augmenter/constants';
 export * from './data_explorer/constants';
+<<<<<<< HEAD
 export * from './vis_type_vega/constants';
+=======
+export * from './vis_type_tsvb/constants';
+>>>>>>> Add FT for TSVB support with MDS

cypress/utils/dashboards/constants.js

@@ -24,6 +24,9 @@ export const DS_NO_AUTH_LABEL = 'RemoteDataSourceNoAuth';
 
 export const DEFAULT_DS_TITLE = 'DefaultDataSource';
 
+export const DS_BASIC_AUTH_HEADER = `Basic ${btoa(`${USERNAME}:${PASSWORD}`)}`;
+export const DS_BASIC_AUTH_LABEL = 'RemoteDataSourceBasicAuth';
+
 export const TIMEOUT_OPTS = { timeout: 60000 };
 export const FORCE_CLICK_OPTS = { force: true };
 export const DATASOURCE_DELAY = 1000;

cypress/utils/dashboards/datasource-management-dashboards-plugin/constants.js

@@ -0,0 +1,13 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { TSVB_INDEX_PATTERN } from './constants';
+
+Cypress.Commands.add('tsvbSaveVisualization', () => {
+  cy.contains('button', 'Save').click();
+  // Type in a random name into the input field
+  cy.get('[data-test-subj="savedObjectTitle"]').type(TSVB_INDEX_PATTERN);
+  // Click the button with data-test-subj="confirmSaveSavedObjectButton"
+  cy.get('[data-test-subj="confirmSaveSavedObjectButton"]').click();
+});

cypress/utils/dashboards/vis_type_tsvb/commands.js

@@ -0,0 +1,23 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { BASE_PATH } from '../../base_constants';
+
+// Data
+export const TSVB_INDEX_DATA = 'metrics.data.txt';
+export const TSVB_PATH_FIXTURE =
+  'dashboard/opensearch_dashboards/vis_type_tsvb/';
+export const TSVB_PATH_INDEX_DATA = TSVB_PATH_FIXTURE + TSVB_INDEX_DATA;
+
+// Update the constants
+export const TSVB_INDEX_START_TIME = 'May 16, 2010 @ 00:00:00.000';
+export const TSVB_INDEX_END_TIME = 'May 16, 2024 @ 00:00:00.000';
+export const TSVB_INDEX_ID = 'metrics';
+export const TSVB_INDEX_PATTERN = 'index-pattern-vis-metrics';
+
+export const TSVB_VIS_TYPE = 'visualization';
+
+// App URL Paths
+export const VIS_APP_PATH = '/app/visualize';
+export const TSVB_CREATE_URL = `${BASE_PATH}${VIS_APP_PATH}#/create?type=metrics`;