fixed flakey tests

Signed-off-by: sumukhswamy <[email protected]>
opensearch-project · Jun 11, 2024 · 3dbe681 · 3dbe681
1 parent e22d6b4
commit 3dbe681
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 8 deletions.
diff --git a/public/components/Main/__snapshots__/main.test.tsx.snap b/public/components/Main/__snapshots__/main.test.tsx.snap
@@ -971,6 +971,8 @@ exports[`<Main /> spec click run button, and response is not ok 1`] = `
                 >
                   <span
                     class="euiButton__text euiButtonGroupButton__textShift"
+                    data-text="SQL"
+                    title="SQL"
                   >
                     <input
                       checked=""
@@ -995,6 +997,8 @@ exports[`<Main /> spec click run button, and response is not ok 1`] = `
                 >
                   <span
                     class="euiButton__text euiButtonGroupButton__textShift"
+                    data-text="PPL"
+                    title="PPL"
                   >
                     <input
                       class="euiScreenReaderOnly"
@@ -1407,6 +1411,8 @@ exports[`<Main /> spec click run button, and response is ok 1`] = `
                 >
                   <span
                     class="euiButton__text euiButtonGroupButton__textShift"
+                    data-text="SQL"
+                    title="SQL"
                   >
                     <input
                       checked=""
@@ -1431,6 +1437,8 @@ exports[`<Main /> spec click run button, and response is ok 1`] = `
                 >
                   <span
                     class="euiButton__text euiButtonGroupButton__textShift"
+                    data-text="PPL"
+                    title="PPL"
                   >
                     <input
                       class="euiScreenReaderOnly"
@@ -1843,6 +1851,8 @@ exports[`<Main /> spec click run button, response fills null and missing values
                 >
                   <span
                     class="euiButton__text euiButtonGroupButton__textShift"
+                    data-text="SQL"
+                    title="SQL"
                   >
                     <input
                       checked=""
@@ -1867,6 +1877,8 @@ exports[`<Main /> spec click run button, response fills null and missing values
                 >
                   <span
                     class="euiButton__text euiButtonGroupButton__textShift"
+                    data-text="PPL"
+                    title="PPL"
                   >
                     <input
                       class="euiScreenReaderOnly"

diff --git a/public/components/Main/main.test.tsx b/public/components/Main/main.test.tsx
@@ -119,9 +119,10 @@ describe('<Main /> spec', () => {
       <Main httpClient={client} setBreadcrumbs={setBreadcrumbsMock} />
     );
     const onRunButton = getByText('Run');
-    const pplbutton =getByTestId('PPL');
+    const sqlbutton = getByText('SQL');
     waitFor(() => {
-     expect(pplbutton).toBeInTheDocument();
+     expect(sqlbutton).toBeInTheDocument();
+     fireEvent.click(sqlbutton);
     });
     const asyncTest = () => {
       fireEvent.click(onRunButton);
@@ -138,9 +139,10 @@ describe('<Main /> spec', () => {
     const { getByText ,getByTestId } = await render(
       <Main httpClient={client} setBreadcrumbs={setBreadcrumbsMock} />
     );
-    const pplbutton = getByTestId('PPL');
+    const sqlbutton = getByText('SQL');
     waitFor(() => {
-     expect(pplbutton).toBeInTheDocument();
+     expect(sqlbutton).toBeInTheDocument();
+     fireEvent.click(sqlbutton);
     });
     const onRunButton = getByText('Run');
     const asyncTest = () => {
@@ -182,9 +184,10 @@ describe('<Main /> spec', () => {
     const { getByText, getByTestId } = await render(
       <Main httpClient={client} setBreadcrumbs={setBreadcrumbsMock} />
     );
-    const pplbutton = getByTestId("PPL");
+    const sqlbutton = getByText('SQL');
     waitFor(() => {
-     expect(pplbutton).toBeInTheDocument();
+     expect(sqlbutton).toBeInTheDocument();
+     fireEvent.click(sqlbutton);
     });
     const onRunButton = getByText('Run');
     const asyncTest = () => {
@@ -209,9 +212,10 @@ describe('<Main /> spec', () => {
     const { getByText,getByTestId } = await render(
       <Main httpClient={client} setBreadcrumbs={setBreadcrumbsMock} />
     );
-    const pplbutton = getByTestId("PPL");
+    const sqlbutton = getByText('SQL');
     waitFor(() => {
-     expect(pplbutton).toBeInTheDocument();
+     expect(sqlbutton).toBeInTheDocument();
+     fireEvent.click(sqlbutton);
     });
     const onTranslateButton = getByText('Explain');
     const asyncTest = () => {