Auto dump cypress test data for workbench

Signed-off-by: Joshua Li <[email protected]>
opensearch-project · Sep 1, 2021 · b517bee · b517bee
1 parent 9ada743
commit b517bee
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 6 deletions.
diff --git a/workbench/.cypress/integration/ui.spec.js b/workbench/.cypress/integration/ui.spec.js
@@ -26,9 +26,32 @@
 
 /// <reference types="cypress" />
 
-import { edit } from "brace";
-import { delay, testQueries, verifyDownloadData, files } from "../utils/constants";
+import { edit } from 'brace';
+import { delay, files, testDataSet, testQueries, verifyDownloadData } from '../utils/constants';
 
+describe('Dump test data', () => {
+  it('Indexes test data for SQL and PPL', () => {
+    const dumpDataSet = (url, index) =>
+      cy.request(url).then((response) => {
+        cy.request({
+          method: 'POST',
+          form: true,
+          url: 'api/console/proxy',
+          headers: {
+            'content-type': 'application/json;charset=UTF-8',
+            'osd-xsrf': true,
+          },
+          qs: {
+            path: `${index}/_bulk`,
+            method: 'POST',
+          },
+          body: response.body,
+        });
+      });
+
+    testDataSet.forEach(({url, index}) => dumpDataSet(url, index));
+  });
+});
 
 describe('Test PPL UI', () => {
   beforeEach(() => {
@@ -183,13 +206,12 @@ describe('Test and verify SQL downloads', () => {
           'osd-xsrf': true,
         },
         body: {
-          'query': 'select * from accounts where balance > 49500'
-        }
+          query: 'select * from accounts where balance > 49500',
+        },
       }).then((response) => {
         if (title === 'Download and verify CSV' || title === 'Download and verify Text') {
           expect(response.body.data.body).to.have.string(files[file]);
-        }
-        else {
+        } else {
           expect(response.body.data.resp).to.have.string(files[file]);
         }
       });

diff --git a/workbench/.cypress/utils/constants.js b/workbench/.cypress/utils/constants.js
@@ -26,6 +26,17 @@
 
 export const delay = 1000;
 
+export const testDataSet = [
+  {
+    url: 'https://raw.githubusercontent.com/opensearch-project/sql/main/integ-test/src/test/resources/accounts.json',
+    index: 'accounts',
+  },
+  {
+    url: 'https://raw.githubusercontent.com/opensearch-project/sql/main/integ-test/src/test/resources/employee_nested.json',
+    index: 'employee_nested'
+  }
+]
+
 export const verifyDownloadData = [
   {
     title: 'Download and verify JSON',