From 865911acee7968869210d7b6b94e65789c8ba07b Mon Sep 17 00:00:00 2001
From: Kos Korchak <kostia.korchak@squadformers.com>
Date: Fri, 13 Oct 2023 17:56:04 -0400
Subject: [PATCH] Download Lineage Results Cypress Test

---
 .../styled/search/DownloadAsCsvModal.tsx      |  2 +
 .../styled/search/SearchExtendedMenu.tsx      |  4 +-
 .../e2e/lineage/download_lineage_results.js   | 80 +++++++++++++++++++
 3 files changed, 84 insertions(+), 2 deletions(-)
 create mode 100644 smoke-test/tests/cypress/cypress/e2e/lineage/download_lineage_results.js

diff --git a/datahub-web-react/src/app/entity/shared/components/styled/search/DownloadAsCsvModal.tsx b/datahub-web-react/src/app/entity/shared/components/styled/search/DownloadAsCsvModal.tsx
index 452658583cf61..92e859ee1b329 100644
--- a/datahub-web-react/src/app/entity/shared/components/styled/search/DownloadAsCsvModal.tsx
+++ b/datahub-web-react/src/app/entity/shared/components/styled/search/DownloadAsCsvModal.tsx
@@ -130,6 +130,7 @@ export default function DownloadAsCsvModal({
                         Close
                     </Button>
                     <Button
+                        data-testid="csv-modal-download-button"
                         onClick={() => {
                             setShowDownloadAsCsvModal(false);
                             triggerCsvDownload(saveAsTitle);
@@ -142,6 +143,7 @@ export default function DownloadAsCsvModal({
             }
         >
             <Input
+                data-testid="download-as-csv-input"
                 placeholder="datahub.csv"
                 value={saveAsTitle}
                 onChange={(e) => {
diff --git a/datahub-web-react/src/app/entity/shared/components/styled/search/SearchExtendedMenu.tsx b/datahub-web-react/src/app/entity/shared/components/styled/search/SearchExtendedMenu.tsx
index d81ec673324bf..9f14a842dc6a6 100644
--- a/datahub-web-react/src/app/entity/shared/components/styled/search/SearchExtendedMenu.tsx
+++ b/datahub-web-react/src/app/entity/shared/components/styled/search/SearchExtendedMenu.tsx
@@ -45,7 +45,7 @@ export default function SearchExtendedMenu({
 
     const menu = (
         <Menu>
-            <MenuItem key="0">
+            <MenuItem key="0" data-testid="download-as-csv-menu-item">
                 <DownloadAsCsvButton
                     isDownloadingCsv={isDownloadingCsv}
                     setShowDownloadAsCsvModal={setShowDownloadAsCsvModal}
@@ -75,7 +75,7 @@ export default function SearchExtendedMenu({
                 totalResults={totalResults}
             />
             <Dropdown overlay={menu} trigger={['click']}>
-                <MenuIcon />
+                <MenuIcon data-testid="tree-dot-menu" />
             </Dropdown>
         </>
     );
diff --git a/smoke-test/tests/cypress/cypress/e2e/lineage/download_lineage_results.js b/smoke-test/tests/cypress/cypress/e2e/lineage/download_lineage_results.js
new file mode 100644
index 0000000000000..204ff72552195
--- /dev/null
+++ b/smoke-test/tests/cypress/cypress/e2e/lineage/download_lineage_results.js
@@ -0,0 +1,80 @@
+const test_dataset = "urn:li:dataset:(urn:li:dataPlatform:kafka,SampleCypressKafkaDataset,PROD)";
+const first_degree = [
+    "urn:li:chart:(looker,cypress_baz1)",
+    "urn:li:dataset:(urn:li:dataPlatform:hdfs,SampleCypressHdfsDataset,PROD)",
+    "urn:li:mlFeature:(cypress-test-2,some-cypress-feature-1)"
+];
+const second_degree = [
+    "urn:li:chart:(looker,cypress_baz2)",
+    "urn:li:dashboard:(looker,cypress_baz)",
+    "urn:li:dataset:(urn:li:dataPlatform:hive,SampleCypressHiveDataset,PROD)",
+    "urn:li:mlPrimaryKey:(cypress-test-2,some-cypress-feature-2)"
+];
+const third_degree_plus = [
+    "urn:li:dataJob:(urn:li:dataFlow:(airflow,cypress_dag_abc,PROD),cypress_task_123)",
+    "urn:li:dataJob:(urn:li:dataFlow:(airflow,cypress_dag_abc,PROD),cypress_task_456)",
+    "urn:li:dataset:(urn:li:dataPlatform:hive,cypress_logging_events,PROD)",
+    "urn:li:dataset:(urn:li:dataPlatform:hive,fct_cypress_users_created,PROD)",
+    "urn:li:dataset:(urn:li:dataPlatform:hive,fct_cypress_users_created_no_tag,PROD)",
+    "urn:li:dataset:(urn:li:dataPlatform:hive,fct_cypress_users_deleted,PROD)"
+];
+const downloadCsvFile = (filename) => {
+    cy.get('[data-testid="tree-dot-menu"]').click();
+    cy.get('[data-testid="download-as-csv-menu-item"]').click();
+    cy.get('[data-testid="download-as-csv-input"]').clear().type(filename);
+    cy.get('[data-testid="csv-modal-download-button"]').click().wait(5000);
+    cy.ensureTextNotPresent("Creating CSV to download");
+};
+
+describe("download lineage results to .csv file", () => {
+
+    it("download and verify lineage results for 1st, 2nd and 3+ degree of dependencies", () => {
+      cy.loginWithCredentials();
+      cy.goToDataset(test_dataset,"SampleCypressKafkaDataset");
+      cy.openEntityTab("Lineage");
+
+      // Verify 1st degree of dependencies
+      cy.contains(/1 - 3 of 3/);
+      downloadCsvFile("first_degree_results.csv");
+      let first_degree_csv = cy.readFile('cypress/downloads/first_degree_results.csv');
+      first_degree.forEach(function (urn) {
+        first_degree_csv.should('contain', urn) 
+      });
+      second_degree.forEach(function (urn) {
+        first_degree_csv.should('not.contain', urn)
+      });
+      third_degree_plus.forEach(function (urn) {
+        first_degree_csv.should('not.contain', urn);
+      });
+
+      // Verify 1st and 2nd degree of dependencies
+      cy.get('[data-testid="facet-degree-2"]').click().wait(5000);
+      cy.contains(/1 - 7 of 7/);
+      downloadCsvFile("second_degree_results.csv");
+      let second_degree_csv = cy.readFile('cypress/downloads/second_degree_results.csv');
+      first_degree.forEach(function (urn) {
+        second_degree_csv.should('contain', urn) 
+      });
+      second_degree.forEach(function (urn) {
+        second_degree_csv.should('contain', urn)
+      });
+      third_degree_plus.forEach(function (urn) {
+        second_degree_csv.should('not.contain', urn);
+      });
+
+      // Verify 1st 2nd and 3+ degree of dependencies(Verify multi page download)
+      cy.get('[data-testid="facet-degree-3+"]').click().wait(5000);
+      cy.contains(/1 - 10 of 13/);
+      downloadCsvFile("third_plus_degree_results.csv");
+      let third_degree_csv = cy.readFile('cypress/downloads/third_plus_degree_results.csv');
+      first_degree.forEach(function (urn) {
+        third_degree_csv.should('contain', urn) 
+      });
+      second_degree.forEach(function (urn) {
+        third_degree_csv.should('contain', urn)
+      });
+      third_degree_plus.forEach(function (urn) {
+        third_degree_csv.should('contain', urn);
+      });
+    });
+}); 
\ No newline at end of file