Add tests for conditional rendering of data connection tabs

Signed-off-by: Ryan Liang <[email protected]>

public/components/datasources/components/__tests__/data_connection.test.tsx

@@ -3,7 +3,7 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 
-import { act } from '@testing-library/react';
+import { act, render, waitFor } from '@testing-library/react';
 import { configure } from 'enzyme';
 import Adapter from 'enzyme-adapter-react-16';
 import React from 'react';
@@ -65,4 +65,44 @@ describe('Data Connection Page test', () => {
     });
     expect(container).toMatchSnapshot();
   });
+
+  it('Does not render Associated Objects, Accelerations, and Installed Integrations tabs for Prometheus data source', async () => {
+    CatalogCacheManager.saveDataSourceCache(mockDataSourceCacheData);
+    CatalogCacheManager.saveAccelerationsCache(mockAccelerationCacheData);
+
+    (coreRefs.http!.get as jest.Mock).mockResolvedValue(describePrometheusDataConnection);
+
+    const { container, queryByText } = render(<DataConnection dataSource="PROMETHEUS" />);
+
+    await waitFor(() => {
+      expect(queryByText('Associated Objects')).not.toBeInTheDocument();
+      expect(queryByText('Accelerations')).not.toBeInTheDocument();
+      expect(queryByText('Installed Integrations')).not.toBeInTheDocument();
+
+      const accessControlTabs = Array.from(container.querySelectorAll('.euiTab__content')).filter(
+        (el) => el.textContent === 'Access control'
+      );
+      expect(accessControlTabs.length).toBeGreaterThan(0);
+    });
+  });
+
+  it('Renders all tabs for S3Glue data source', async () => {
+    CatalogCacheManager.saveDataSourceCache(mockDataSourceCacheData);
+    CatalogCacheManager.saveAccelerationsCache(mockAccelerationCacheData);
+
+    (coreRefs.http!.get as jest.Mock).mockResolvedValue(describeS3Dataconnection);
+
+    const { container, getByText } = render(<DataConnection dataSource="S3GLUE" />);
+
+    await waitFor(() => {
+      expect(getByText('Associated Objects')).toBeInTheDocument();
+      expect(getByText('Accelerations')).toBeInTheDocument();
+      expect(getByText('Installed Integrations')).toBeInTheDocument();
+
+      const accessControlTabs = Array.from(container.querySelectorAll('.euiTab__content')).filter(
+        (el) => el.textContent === 'Access control'
+      );
+      expect(accessControlTabs.length).toBeGreaterThan(0);
+    });
+  });
 });