diff --git a/src/plugins/data_source_management/public/components/direct_query_data_sources_components/acceleration_creation/selectors/__snapshots__/preview_sql_defintion.test.tsx.snap b/src/plugins/data_source_management/public/components/direct_query_data_sources_components/acceleration_creation/selectors/__snapshots__/preview_sql_defintion.test.tsx.snap
new file mode 100644
index 000000000000..4ac998ac3ac2
--- /dev/null
+++ b/src/plugins/data_source_management/public/components/direct_query_data_sources_components/acceleration_creation/selectors/__snapshots__/preview_sql_defintion.test.tsx.snap
@@ -0,0 +1,386 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`PreviewSQLDefinition renders component correctly 1`] = `
+<PreviewSQLDefinition
+  accelerationFormData={
+    Object {
+      "dataSource": "test_source",
+      "formErrors": Object {},
+    }
+  }
+  application={
+    Object {
+      "applications$": BehaviorSubject {
+        "_isScalar": false,
+        "_value": Map {},
+        "closed": false,
+        "hasError": false,
+        "isStopped": false,
+        "observers": Array [],
+        "thrownError": null,
+      },
+      "capabilities": Object {
+        "catalogue": Object {},
+        "management": Object {},
+        "navLinks": Object {},
+        "workspaces": Object {},
+      },
+      "currentAppId$": Observable {
+        "_isScalar": false,
+        "source": Subject {
+          "_isScalar": false,
+          "closed": false,
+          "hasError": false,
+          "isStopped": false,
+          "observers": Array [],
+          "thrownError": null,
+        },
+      },
+      "getUrlForApp": [MockFunction],
+      "navigateToApp": [MockFunction],
+      "navigateToUrl": [MockFunction],
+      "registerMountContext": [MockFunction],
+      "setAppBadgeControls": [MockFunction],
+      "setAppBottomControls": [MockFunction],
+      "setAppCenterControls": [MockFunction],
+      "setAppDescriptionControls": [MockFunction],
+      "setAppLeftControls": [MockFunction],
+      "setAppRightControls": [MockFunction],
+    }
+  }
+  http={
+    Object {
+      "addLoadingCountSource": [MockFunction],
+      "anonymousPaths": Object {
+        "isAnonymous": [MockFunction],
+        "register": [MockFunction],
+      },
+      "basePath": BasePath {
+        "basePath": "",
+        "clientBasePath": "",
+        "get": [Function],
+        "getBasePath": [Function],
+        "prepend": [Function],
+        "remove": [Function],
+        "serverBasePath": "",
+      },
+      "delete": [MockFunction],
+      "fetch": [MockFunction],
+      "get": [MockFunction] {
+        "calls": Array [
+          Array [
+            "/api/status",
+          ],
+        ],
+        "results": Array [
+          Object {
+            "type": "return",
+            "value": Promise {},
+          },
+        ],
+      },
+      "getLoadingCount$": [MockFunction],
+      "head": [MockFunction],
+      "intercept": [MockFunction],
+      "options": [MockFunction],
+      "patch": [MockFunction],
+      "post": [MockFunction],
+      "put": [MockFunction],
+    }
+  }
+  notifications={
+    Object {
+      "toasts": Object {
+        "add": [MockFunction],
+        "addDanger": [MockFunction],
+        "addError": [MockFunction],
+        "addInfo": [MockFunction],
+        "addSuccess": [MockFunction],
+        "addWarning": [MockFunction],
+        "get$": [MockFunction],
+        "remove": [MockFunction],
+      },
+    }
+  }
+  resetFlyout={[MockFunction]}
+  setAccelerationFormData={[MockFunction]}
+>
+  <EuiAccordion
+    arrowDisplay="left"
+    buttonContent={
+      <EuiText
+        data-test-subj="preview-sql-header"
+      >
+        <h3>
+          Preview SQL definition
+        </h3>
+      </EuiText>
+    }
+    id="accordion1"
+    initialIsOpen={false}
+    isLoading={false}
+    isLoadingMessage={false}
+    paddingSize="l"
+  >
+    <div
+      className="euiAccordion"
+    >
+      <div
+        className="euiAccordion__triggerWrapper"
+      >
+        <button
+          aria-controls="accordion1"
+          aria-expanded={false}
+          className="euiAccordion__button"
+          id="generated-id"
+          onClick={[Function]}
+          type="button"
+        >
+          <span
+            className="euiAccordion__iconWrapper"
+          >
+            <EuiIcon
+              className="euiAccordion__icon"
+              size="m"
+              type="arrowRight"
+            >
+              <span
+                className="euiAccordion__icon"
+                data-euiicon-type="arrowRight"
+                size="m"
+              />
+            </EuiIcon>
+          </span>
+          <span
+            className="euiIEFlexWrapFix"
+          >
+            <EuiText
+              data-test-subj="preview-sql-header"
+            >
+              <div
+                className="euiText euiText--medium"
+                data-test-subj="preview-sql-header"
+              >
+                <h3>
+                  Preview SQL definition
+                </h3>
+              </div>
+            </EuiText>
+          </span>
+        </button>
+      </div>
+      <div
+        aria-labelledby="generated-id"
+        className="euiAccordion__childWrapper"
+        id="accordion1"
+        role="region"
+        tabIndex={-1}
+      >
+        <EuiResizeObserver
+          onResize={[Function]}
+        >
+          <div>
+            <div
+              className="euiAccordion__padding--l"
+            >
+              <EuiFlexGroup>
+                <div
+                  className="euiFlexGroup euiFlexGroup--gutterLarge euiFlexGroup--directionRow euiFlexGroup--responsive"
+                >
+                  <EuiFlexItem
+                    grow={false}
+                  >
+                    <div
+                      className="euiFlexItem euiFlexItem--flexGrowZero"
+                    >
+                      <EuiButton
+                        color="success"
+                        onClick={[Function]}
+                      >
+                        <EuiButtonDisplay
+                          baseClassName="euiButton"
+                          color="success"
+                          disabled={false}
+                          element="button"
+                          isDisabled={false}
+                          onClick={[Function]}
+                          type="button"
+                        >
+                          <button
+                            className="euiButton euiButton--success"
+                            disabled={false}
+                            onClick={[Function]}
+                            style={
+                              Object {
+                                "minWidth": undefined,
+                              }
+                            }
+                            type="button"
+                          >
+                            <EuiButtonContent
+                              className="euiButton__content"
+                              iconGap="m"
+                              iconSide="left"
+                              textProps={
+                                Object {
+                                  "className": "euiButton__text",
+                                }
+                              }
+                            >
+                              <span
+                                className="euiButtonContent euiButton__content"
+                              >
+                                <span
+                                  className="euiButton__text"
+                                >
+                                  Generate preview
+                                </span>
+                              </span>
+                            </EuiButtonContent>
+                          </button>
+                        </EuiButtonDisplay>
+                      </EuiButton>
+                    </div>
+                  </EuiFlexItem>
+                  <EuiFlexItem
+                    grow={false}
+                  >
+                    <div
+                      className="euiFlexItem euiFlexItem--flexGrowZero"
+                    >
+                      <EuiButton
+                        data-test-subj="workbenchButton"
+                        iconSide="right"
+                        onClick={[Function]}
+                      >
+                        <EuiButtonDisplay
+                          baseClassName="euiButton"
+                          data-test-subj="workbenchButton"
+                          disabled={false}
+                          element="button"
+                          iconSide="right"
+                          isDisabled={false}
+                          onClick={[Function]}
+                          type="button"
+                        >
+                          <button
+                            className="euiButton euiButton--primary"
+                            data-test-subj="workbenchButton"
+                            disabled={false}
+                            onClick={[Function]}
+                            style={
+                              Object {
+                                "minWidth": undefined,
+                              }
+                            }
+                            type="button"
+                          >
+                            <EuiButtonContent
+                              className="euiButton__content"
+                              iconGap="m"
+                              iconSide="right"
+                              textProps={
+                                Object {
+                                  "className": "euiButton__text",
+                                }
+                              }
+                            >
+                              <span
+                                className="euiButtonContent euiButtonContent--iconRight euiButton__content"
+                              >
+                                <span
+                                  className="euiButton__text"
+                                >
+                                  Edit in Query Workbench
+                                </span>
+                              </span>
+                            </EuiButtonContent>
+                          </button>
+                        </EuiButtonDisplay>
+                      </EuiButton>
+                    </div>
+                  </EuiFlexItem>
+                </div>
+              </EuiFlexGroup>
+              <EuiSpacer
+                size="l"
+              >
+                <div
+                  className="euiSpacer euiSpacer--l"
+                />
+              </EuiSpacer>
+              <EuiCodeBlock
+                fontSize="m"
+                isCopyable={true}
+                language="sql"
+                paddingSize="m"
+              >
+                <EuiCodeBlockImpl
+                  fontSize="m"
+                  inline={false}
+                  isCopyable={true}
+                  language="sql"
+                  paddingSize="m"
+                >
+                  <div>
+                    <pre>
+                      <code />
+                    </pre>
+                  </div>
+                </EuiCodeBlockImpl>
+              </EuiCodeBlock>
+            </div>
+          </div>
+        </EuiResizeObserver>
+      </div>
+    </div>
+  </EuiAccordion>
+</PreviewSQLDefinition>
+`;
+
+exports[`PreviewSQLDefinition renders correctly 1`] = `
+<Fragment>
+  <EuiAccordion
+    arrowDisplay="left"
+    buttonContent={
+      <EuiText
+        data-test-subj="preview-sql-header"
+      >
+        <h3>
+          Preview SQL definition
+        </h3>
+      </EuiText>
+    }
+    id="accordion1"
+    initialIsOpen={false}
+    isLoading={false}
+    isLoadingMessage={false}
+    paddingSize="l"
+  >
+    <EuiFlexGroup>
+      <EuiFlexItem
+        grow={false}
+      >
+        <EuiButton
+          color="success"
+          onClick={[Function]}
+        >
+          Generate preview
+        </EuiButton>
+      </EuiFlexItem>
+      <EuiFlexItem
+        grow={false}
+      />
+    </EuiFlexGroup>
+    <EuiSpacer
+      size="l"
+    />
+    <EuiCodeBlock
+      fontSize="m"
+      isCopyable={true}
+      language="sql"
+      paddingSize="m"
+    />
+  </EuiAccordion>
+</Fragment>
+`;
diff --git a/src/plugins/data_source_management/public/components/direct_query_data_sources_components/acceleration_creation/selectors/preview_sql_defintion.test.tsx b/src/plugins/data_source_management/public/components/direct_query_data_sources_components/acceleration_creation/selectors/preview_sql_defintion.test.tsx
new file mode 100644
index 000000000000..8d4c9d159554
--- /dev/null
+++ b/src/plugins/data_source_management/public/components/direct_query_data_sources_components/acceleration_creation/selectors/preview_sql_defintion.test.tsx
@@ -0,0 +1,167 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import React from 'react';
+import { mount } from 'enzyme';
+import { act } from 'react-dom/test-utils';
+import { PreviewSQLDefinition } from './preview_sql_defintion';
+import { EuiButton } from '@elastic/eui';
+import { coreMock } from './../../../../../../../core/public/mocks';
+import { formValidator, hasError } from '../create/utils';
+
+jest.mock('../visual_editors/query_builder', () => ({
+  accelerationQueryBuilder: jest.fn(() => 'mocked SQL query'),
+}));
+
+jest.mock('../create/utils', () => ({
+  formValidator: jest.fn(() => ({})),
+  hasError: jest.fn(() => false),
+}));
+
+describe('PreviewSQLDefinition', () => {
+  const mockCoreStart = coreMock.createStart();
+  const mockAccelerationFormData = {
+    dataSource: 'test_source',
+    formErrors: {},
+  };
+
+  const mockSetAccelerationFormData = jest.fn();
+  const mockResetFlyout = jest.fn();
+
+  beforeEach(() => {
+    jest.clearAllMocks();
+    mockCoreStart.http.get.mockResolvedValue({
+      status: {
+        statuses: [{ id: 'plugin:queryWorkbenchDashboards' }],
+      },
+    });
+  });
+
+  const mountComponent = async () => {
+    const wrapper = mount(
+      <PreviewSQLDefinition
+        accelerationFormData={mockAccelerationFormData}
+        setAccelerationFormData={mockSetAccelerationFormData}
+        resetFlyout={mockResetFlyout}
+        notifications={mockCoreStart.notifications}
+        application={mockCoreStart.application}
+        http={mockCoreStart.http}
+      />
+    );
+
+    await act(async () => {
+      await Promise.resolve();
+    });
+    wrapper.update();
+
+    return wrapper;
+  };
+
+  test('renders component correctly', async () => {
+    const wrapper = await mountComponent();
+    expect(wrapper).toMatchSnapshot();
+  });
+
+  test('opens SQL Workbench when button is clicked', async () => {
+    const wrapper = await mountComponent();
+
+    const button = wrapper.findWhere(
+      (node) => node.type() === EuiButton && node.prop('data-test-subj') === 'workbenchButton'
+    );
+
+    expect(button.exists()).toBe(true);
+    expect(button.length).toBe(1);
+
+    await act(async () => {
+      button.prop('onClick')();
+      await Promise.resolve();
+    });
+
+    expect(mockCoreStart.application.navigateToApp).toHaveBeenCalledWith(expect.any(String), {
+      path: '#/test_source',
+      state: {
+        language: 'sql',
+        queryToRun: 'mocked SQL query',
+      },
+    });
+    expect(mockResetFlyout).toHaveBeenCalled();
+  });
+
+  test('displays error toast if SQL Workbench plugin check fails', async () => {
+    mockCoreStart.http.get.mockRejectedValueOnce(new Error('Error checking plugin'));
+
+    await mountComponent();
+
+    expect(mockCoreStart.notifications.toasts.addDanger).toHaveBeenCalledWith(
+      'Error checking Query Workbench Plugin Installation status.'
+    );
+  });
+
+  test('shows update preview button when form data changes', async () => {
+    const wrapper = await mountComponent();
+
+    const previewButton = wrapper.findWhere(
+      (node) => node.type() === EuiButton && node.text().includes('Generate preview')
+    );
+
+    await act(async () => {
+      previewButton.prop('onClick')();
+      await Promise.resolve();
+    });
+
+    await act(async () => {
+      wrapper.setProps({
+        accelerationFormData: { ...mockAccelerationFormData, dataSource: 'new_source' },
+      });
+      await Promise.resolve();
+    });
+    wrapper.update();
+
+    const updateButton = wrapper.findWhere(
+      (node) => node.type() === EuiButton && node.text().includes('Update preview')
+    );
+
+    expect(updateButton.exists()).toBe(true);
+  });
+
+  test('validates form before opening SQL Workbench', async () => {
+    // Mock form validation to return errors
+    (formValidator as jest.Mock).mockReturnValueOnce({ someField: 'error' });
+    (hasError as jest.Mock).mockReturnValueOnce(true);
+
+    const wrapper = await mountComponent();
+
+    const button = wrapper.findWhere(
+      (node) => node.type() === EuiButton && node.prop('data-test-subj') === 'workbenchButton'
+    );
+
+    await act(async () => {
+      button.prop('onClick')();
+      await Promise.resolve();
+    });
+
+    expect(mockCoreStart.application.navigateToApp).not.toHaveBeenCalled();
+    expect(mockSetAccelerationFormData).toHaveBeenCalledWith({
+      ...mockAccelerationFormData,
+      formErrors: { someField: 'error' },
+    });
+  });
+
+  test('hides SQL Workbench button when plugin is not installed', async () => {
+    mockCoreStart.http.get.mockResolvedValueOnce({
+      status: {
+        statuses: [{ id: 'some-other-plugin' }],
+      },
+    });
+
+    const wrapper = await mountComponent();
+
+    const button = wrapper.findWhere(
+      (node) => node.type() === EuiButton && node.prop('data-test-subj') === 'workbenchButton'
+    );
+
+    expect(button.exists()).toBe(false);
+  });
+});
diff --git a/src/plugins/data_source_management/public/components/direct_query_data_sources_components/acceleration_creation/selectors/preview_sql_defintion.tsx b/src/plugins/data_source_management/public/components/direct_query_data_sources_components/acceleration_creation/selectors/preview_sql_defintion.tsx
index 3914cc4d9b44..20e9af8e8412 100644
--- a/src/plugins/data_source_management/public/components/direct_query_data_sources_components/acceleration_creation/selectors/preview_sql_defintion.tsx
+++ b/src/plugins/data_source_management/public/components/direct_query_data_sources_components/acceleration_creation/selectors/preview_sql_defintion.tsx
@@ -94,7 +94,7 @@ export const PreviewSQLDefinition = ({
   };
 
   const queryWorkbenchButton = sqlWorkbenchPLuginExists ? (
-    <EuiButton iconSide="right" onClick={openInWorkbench}>
+    <EuiButton iconSide="right" onClick={openInWorkbench} data-test-subj="workbenchButton">
       Edit in Query Workbench
     </EuiButton>
   ) : (