update test for data_connection page

Signed-off-by: Shenoy Pratik <sgguruda@amazon.com>
ps48 · Mar 9, 2024 · 97a8fc7 · 97a8fc7
1 parent c65949c
commit 97a8fc7
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/public/components/datasources/components/__tests__/data_connection.test.tsx b/public/components/datasources/components/__tests__/data_connection.test.tsx
@@ -3,25 +3,22 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
+import { act } from '@testing-library/react';
 import { configure } from 'enzyme';
 import Adapter from 'enzyme-adapter-react-16';
-import { act } from '@testing-library/react';
 import React from 'react';
+import ReactDOM from 'react-dom';
+import { coreRefs } from '../../../../../public/framework/core_refs';
 import {
   describePrometheusDataConnection,
   describeS3Dataconnection,
 } from '../../../../../test/datasources';
 import { DataConnection } from '../manage/data_connection';
-import ReactDOM from 'react-dom';
-import { coreRefs } from '../../../../../public/framework/core_refs';
 
 jest.mock('../../../../plugin', () => ({
-  getRenderAccelerationDetailsFlyout: jest.fn(() =>
-    jest.fn().mockImplementation(() => console.log('Acceleration Details Flyout Rendered'))
-  ),
-  getRenderAssociatedObjectsDetailsFlyout: jest.fn(() =>
-    jest.fn().mockImplementation(() => console.log('Associated Objects Details Flyout Rendered'))
-  ),
+  getRenderAccelerationDetailsFlyout: jest.fn(),
+  getRenderAssociatedObjectsDetailsFlyout: jest.fn(),
+  getRenderCreateAccelerationFlyout: jest.fn(),
 }));
 
 jest.mock('../../../../../public/framework/core_refs', () => ({