From 43a50c6eb5426d50297212e134d6848c3011baa1 Mon Sep 17 00:00:00 2001
From: Suchit Sahoo <suchsah@amazon.com>
Date: Mon, 18 Nov 2024 07:15:54 -0800
Subject: [PATCH] Add Legacy tests for Query Enhanced Workflow

Signed-off-by: Suchit Sahoo <suchsah@amazon.com>
---
 .../datasource_selector.spec.js               | 88 +++++++++++++++++++
 .../dashboards/query_enhancement/commands.js  |  5 ++
 2 files changed, 93 insertions(+)
 create mode 100644 cypress/integration/core-opensearch-dashboards/opensearch-dashboards/apps/query_enhancement/datasource_selector.spec.js

diff --git a/cypress/integration/core-opensearch-dashboards/opensearch-dashboards/apps/query_enhancement/datasource_selector.spec.js b/cypress/integration/core-opensearch-dashboards/opensearch-dashboards/apps/query_enhancement/datasource_selector.spec.js
new file mode 100644
index 000000000..e916dbdf8
--- /dev/null
+++ b/cypress/integration/core-opensearch-dashboards/opensearch-dashboards/apps/query_enhancement/datasource_selector.spec.js
@@ -0,0 +1,88 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import {
+  MiscUtils,
+  TestFixtureHandler,
+} from '@opensearch-dashboards-test/opensearch-dashboards-test-library';
+import { CURRENT_TENANT } from '../../../../../utils/commands';
+
+const miscUtils = new MiscUtils(cy);
+const testFixtureHandler = new TestFixtureHandler(
+  cy,
+  Cypress.env('openSearchUrl')
+);
+
+describe('Data source selector', () => {
+  before(() => {
+    CURRENT_TENANT.newTenant = 'global';
+    testFixtureHandler.importJSONMapping(
+      'cypress/fixtures/dashboard/opensearch_dashboards/data_explorer/index_pattern_without_timefield/mappings.json.txt'
+    );
+    testFixtureHandler.importJSONDoc(
+      'cypress/fixtures/dashboard/opensearch_dashboards/data_explorer/index_pattern_without_timefield/data.json.txt'
+    );
+
+    miscUtils.visitPage('app/data-explorer/discover#/');
+  });
+
+  after(() => {
+    testFixtureHandler.clearJSONMapping(
+      'cypress/fixtures/dashboard/opensearch_dashboards/data_explorer/index_pattern_without_timefield/mappings.json.txt'
+    );
+    cy.deleteSavedObjectByType('index-pattern');
+  });
+
+  it('displays all data sources by default', () => {
+    // Needs validation
+    cy.get('[data-test-subj="datasetSelectorButton"]').click({ force: true });
+    cy.get('.euiSelectableListItem__text').should('have.length', 2);
+  });
+
+  it('filters options based on user input', () => {
+    cy.get('input[aria-label="Filter options"]').click().type('without', {
+      force: true,
+    });
+    cy.get('.euiSelectableListItem__text').should('have.length', 1);
+    cy.get('.euiSelectableListItem__text')
+      .first()
+      .should('contain', 'without-timefield');
+  });
+
+  it('updates the visual length of the dropdown based on filtered results', () => {
+    // cy.get('[data-test-subj="datasetSelectorButton"]').click({ force: true });
+    cy.get('input[aria-label="Filter options"]').click().clear({
+      force: true,
+    });
+    cy.get('input[aria-label="Filter options"]')
+      .click()
+      .type('without-timefield', {
+        force: true,
+      });
+    cy.get('.euiSelectableList').then(($listAfterFilter) => {
+      const heightAfterFilter = $listAfterFilter.height();
+      cy.get('input[aria-label="Filter options"]').click().clear({
+        force: true,
+      });
+      cy.get('.euiSelectableList').should(($listAll) => {
+        expect($listAll.height()).to.be.greaterThan(heightAfterFilter);
+      });
+    });
+  });
+
+  it('selects the correct option when clicked', () => {
+    // cy.get('[data-test-subj="datasetSelectorButton"]').click({ force: true });
+    cy.get('input[aria-label="Filter options"]')
+      .click()
+      .type('with-timefield', {
+        force: true,
+      });
+
+    cy.contains('.euiSelectableListItem__text', 'with-timefield').click();
+    cy.get(
+      '[data-test-subj="datasetSelectorButton"] .euiButton__content'
+    ).should('contain', 'with-timefield');
+  });
+});
diff --git a/cypress/utils/dashboards/query_enhancement/commands.js b/cypress/utils/dashboards/query_enhancement/commands.js
index 88f852c9b..93f39177a 100644
--- a/cypress/utils/dashboards/query_enhancement/commands.js
+++ b/cypress/utils/dashboards/query_enhancement/commands.js
@@ -45,3 +45,8 @@ Cypress.Commands.add('setQueryLanguage', (value, submit = true) => {
     force: true,
   });
 });
+
+Cypress.Commands.add('setDatasource', (value) => {
+  cy.get(`[class~="datasetSelector__button"]`).click();
+  cy.get(`[data-test-subj="datasetOption-${value}"]`).click();
+});