fix tests

openmetadata-ui/src/main/resources/ui/src/components/DataInsight/DataInsightSummary.test.tsx

@@ -37,6 +37,7 @@ jest.mock('react-router-dom', () => ({
 }));
 
 jest.mock('../../rest/DataInsightAPI', () => ({
+  ...jest.requireActual('../../rest/DataInsightAPI'),
   getAggregateChartData: jest.fn().mockImplementation(() => Promise.resolve()),
 }));
 

openmetadata-ui/src/main/resources/ui/src/components/DataInsight/DataInsightSummary.tsx

@@ -48,7 +48,7 @@ import './data-insight-detail.less';
 
 interface Props {
   chartFilter: ChartFilter;
-  onScrollToChart: (chartType: DataInsightChartType) => void;
+  onScrollToChart: (chartType: SystemChartType | DataInsightChartType) => void;
 }
 
 const DataInsightSummary: FC<Props> = ({ chartFilter, onScrollToChart }) => {

openmetadata-ui/src/main/resources/ui/src/constants/DataInsight.constants.ts

@@ -63,10 +63,10 @@ export const INITIAL_CHART_FILTER: ChartFilter = {
 };
 
 export const ENTITIES_CHARTS = [
-  DataInsightChartType.TotalEntitiesByType,
-  DataInsightChartType.PercentageOfEntitiesWithDescriptionByType,
-  DataInsightChartType.PercentageOfEntitiesWithOwnerByType,
-  DataInsightChartType.TotalEntitiesByTier,
+  SystemChartType.TotalDataAssets,
+  SystemChartType.PercentageOfDataAssetWithDescription,
+  SystemChartType.PercentageOfDataAssetWithOwner,
+  SystemChartType.TotalDataAssetsByTier,
 ];
 
 export const WEB_CHARTS = [

openmetadata-ui/src/main/resources/ui/src/pages/DataInsightPage/DataInsightHeader/DataInsightHeader.interface.ts

@@ -11,7 +11,8 @@
  *  limitations under the License.
  */
 import { DataInsightChartType } from '../../../generated/dataInsight/dataInsightChartResult';
+import { SystemChartType } from '../../../rest/DataInsightAPI';
 
 export interface DataInsightHeaderProps {
-  onScrollToChart: (chartType: DataInsightChartType) => void;
+  onScrollToChart: (chartType: SystemChartType | DataInsightChartType) => void;
 }

openmetadata-ui/src/main/resources/ui/src/pages/DataInsightPage/DataInsightPage.component.tsx

@@ -31,6 +31,7 @@ import { ERROR_PLACEHOLDER_TYPE } from '../../enums/common.enum';
 import { DataInsightChartType } from '../../generated/dataInsight/dataInsightChartResult';
 import { Operation } from '../../generated/entity/policies/policy';
 import { DataInsightTabs } from '../../interface/data-insight.interface';
+import { SystemChartType } from '../../rest/DataInsightAPI';
 import { getDataInsightPathWithFqn } from '../../utils/DataInsightUtils';
 import { checkPermission } from '../../utils/PermissionsUtils';
 import './data-insight.less';
@@ -73,10 +74,14 @@ const DataInsightPage = () => {
     [permissions]
   );
 
-  const [selectedChart, setSelectedChart] = useState<DataInsightChartType>();
+  const [selectedChart, setSelectedChart] = useState<
+    SystemChartType | DataInsightChartType
+  >();
 
-  const handleScrollToChart = (chartType: DataInsightChartType) => {
-    if (ENTITIES_CHARTS.includes(chartType)) {
+  const handleScrollToChart = (
+    chartType: SystemChartType | DataInsightChartType
+  ) => {
+    if (ENTITIES_CHARTS.includes(chartType as SystemChartType)) {
       history.push(getDataInsightPathWithFqn(DataInsightTabs.DATA_ASSETS));
     } else {
       history.push(getDataInsightPathWithFqn(DataInsightTabs.APP_ANALYTICS));

openmetadata-ui/src/main/resources/ui/src/pages/DataInsightPage/DataInsightProvider.tsx

@@ -254,7 +254,7 @@ const DataInsightProvider = ({ children }: DataInsightProviderProps) => {
         data: kpiList,
       },
       teamFilter: {
-        options: uniqBy(teamsOptions.options, 'id'),
+        options: uniqBy(teamsOptions.options, 'key'),
         selectedKeys: teamsOptions.selectedOptions,
         onChange: handleTeamChange,
         onGetInitialOptions: fetchDefaultTeamOptions,