diff --git a/packages/base/src/page/Home/AIBanner/__tests__/__snapshots__/index.test.tsx.snap b/packages/base/src/page/Home/AIBanner/__tests__/__snapshots__/index.test.tsx.snap new file mode 100644 index 000000000..090ef7d1c --- /dev/null +++ b/packages/base/src/page/Home/AIBanner/__tests__/__snapshots__/index.test.tsx.snap @@ -0,0 +1,311 @@ +// Jest Snapshot v1, https://goo.gl/fbAQLP + +exports[`AIBanner should render AI banner after data loaded 1`] = ` + +
+
+
+
+ +
+
+
+
+ +`; + +exports[`AIBanner should render loading state on initial render 1`] = ` + +
+
+
+
+
+
+
    +
  • +
  • +
  • +
  • +
+
+
+
+
+
+
+ +`; diff --git a/packages/base/src/page/Home/AIBanner/__tests__/index.test.tsx b/packages/base/src/page/Home/AIBanner/__tests__/index.test.tsx new file mode 100644 index 000000000..a75b9d408 --- /dev/null +++ b/packages/base/src/page/Home/AIBanner/__tests__/index.test.tsx @@ -0,0 +1,146 @@ +import { act, cleanup, fireEvent, screen } from '@testing-library/react'; +import AIBanner from '..'; +import { baseSuperRender } from '../../../../testUtils/superRender'; +import { sqleMockApi } from '@actiontech/shared/lib/testUtil'; +import { mockUseCurrentUser } from '@actiontech/shared/lib/testUtil/mockHook/mockUseCurrentUser'; +import { mockUsePermission } from '@actiontech/shared/lib/testUtil/mockHook/mockUsePermission'; +import { mockUseRecentlyOpenedProjects } from '../../../Nav/SideMenu/testUtils/mockUseRecentlyOpenedProjects'; +import { useTypedNavigate } from '@actiontech/shared'; +import { ROUTE_PATHS } from '@actiontech/dms-kit'; +import EventEmitter from '../../../../utils/EventEmitter'; +import EmitterKey from '../../../../data/EmitterKey'; + +jest.mock('@actiontech/shared', () => ({ + ...jest.requireActual('@actiontech/shared'), + useTypedNavigate: jest.fn() +})); + +describe('AIBanner', () => { + const navigateSpy = jest.fn(); + let getAIHubBannerSpy: jest.SpyInstance; + + beforeEach(() => { + jest.useFakeTimers(); + getAIHubBannerSpy = sqleMockApi.statistic.getAIHubBanner(); + (useTypedNavigate as jest.Mock).mockImplementation(() => navigateSpy); + mockUseCurrentUser(); + mockUseRecentlyOpenedProjects({ currentProjectID: '1' }); + mockUsePermission( + { checkPagePermission: jest.fn().mockReturnValue(true) }, + { useSpyOnMockHooks: true } + ); + }); + + afterEach(() => { + jest.useRealTimers(); + jest.clearAllMocks(); + cleanup(); + }); + + it('should render loading state on initial render', () => { + const { baseElement } = baseSuperRender(); + expect(baseElement).toMatchSnapshot(); + expect(getAIHubBannerSpy).toHaveBeenCalledTimes(1); + }); + + it('should render AI banner after data loaded', async () => { + const { baseElement } = baseSuperRender(); + + await act(async () => jest.advanceTimersByTime(3000)); + + expect(baseElement).toMatchSnapshot(); + expect(screen.getByText('AI治理效能洞察')).toBeInTheDocument(); + expect( + screen.getByText('基于大模型实时监控规范与性能, AI驱动全链路质量闭环。') + ).toBeInTheDocument(); + expect(screen.getByText('查看完整报告')).toBeInTheDocument(); + expect(screen.getByText('风险拦截')).toBeInTheDocument(); + expect(screen.getByText('性能优化')).toBeInTheDocument(); + expect(screen.getByText('AI 性能引擎')).toBeInTheDocument(); + expect(screen.getByText('AI 智能修正')).toBeInTheDocument(); + }); + + it('should navigate to report statistics when clicking "查看完整报告"', async () => { + baseSuperRender(); + + await act(async () => jest.advanceTimersByTime(3000)); + + fireEvent.click(screen.getByText('查看完整报告')); + + expect(navigateSpy).toHaveBeenCalledWith( + ROUTE_PATHS.SQLE.REPORT_STATISTICS.index, + { queries: { tab: 'ai-governance' } } + ); + }); + + it('should navigate to SQL optimization when clicking "AI 性能引擎" with current project', async () => { + baseSuperRender(); + + await act(async () => jest.advanceTimersByTime(3000)); + + fireEvent.click(screen.getByText('AI 性能引擎')); + + expect(navigateSpy).toHaveBeenCalledWith( + ROUTE_PATHS.SQLE.SQL_AUDIT.create_optimization, + { params: { projectID: '1' } } + ); + }); + + it('should trigger smart correction handler when clicking "AI 智能修正"', async () => { + baseSuperRender(); + + await act(async () => jest.advanceTimersByTime(3000)); + + // 点击 AI 智能修正时，不会触发页面跳转（与 AI 性能引擎不同） + expect(() => + fireEvent.click(screen.getByText('AI 智能修正')) + ).not.toThrow(); + expect(navigateSpy).not.toHaveBeenCalled(); + }); + + it('should show paid feature prompt when all modules are disabled', async () => { + getAIHubBannerSpy.mockImplementation(() => + Promise.resolve({ + data: { + data: { + modules: [ + { + ai_module_type: 'performance_engine', + is_enabled: false, + banner_cards: [{ need_display: true }] + }, + { + ai_module_type: 'smart_correction', + is_enabled: false, + banner_cards: [{ need_display: true }] + } + ] + } + } + }) + ); + + baseSuperRender(); + await act(async () => jest.advanceTimersByTime(3000)); + + fireEvent.click(screen.getByText('查看完整报告')); + + expect( + screen.getByText('当前功能为付费增值模块，请联系商务获取详细信息') + ).toBeInTheDocument(); + }); + + it('should refresh AI banner data when DMS_Reload_Initial_Data event is emitted', async () => { + baseSuperRender(); + + await act(async () => jest.advanceTimersByTime(3000)); + expect(getAIHubBannerSpy).toHaveBeenCalledTimes(1); + + await act(async () => { + EventEmitter.emit(EmitterKey.DMS_Reload_Initial_Data); + await jest.advanceTimersByTime(3000); + }); + + expect(getAIHubBannerSpy).toHaveBeenCalledTimes(2); + }); +}); diff --git a/packages/base/src/page/Home/AIBanner/index.tsx b/packages/base/src/page/Home/AIBanner/index.tsx index 5a11d5a6e..293020afa 100644 --- a/packages/base/src/page/Home/AIBanner/index.tsx +++ b/packages/base/src/page/Home/AIBanner/index.tsx @@ -1,4 +1,4 @@ -import { useMemo, useState } from 'react'; +import { useEffect, useMemo, useState } from 'react'; import { useRequest } from 'ahooks'; import { Card, Space, Typography } from 'antd'; import { useTranslation } from 'react-i18next'; @@ -14,6 +14,8 @@ import { import { AiOutlined } from '@actiontech/icons'; import { AIBannerStyleWrapper } from './style'; import useRecentlyOpenedProjects from '../../Nav/SideMenu/useRecentlyOpenedProjects'; +import EventEmitter from '../../../utils/EventEmitter'; +import EmitterKey from '../../../data/EmitterKey'; import { ROUTE_PATHS } from '@actiontech/dms-kit'; import { AIModuleBannerCardsAiModuleTypeEnum } from '@actiontech/shared/lib/api/sqle/service/common.enum'; import { IAIModuleBannerCards } from '@actiontech/shared/lib/api/sqle/service/common.d'; @@ -45,12 +47,24 @@ const AIBanner: React.FC = () => { useState(false); const { checkPagePermission } = usePermission(); - const { data: bannerData, loading } = useRequest(() => { + const { + data: bannerData, + loading, + refresh + } = useRequest(() => { return AiHubService.GetAIHubBanner().then((res) => { return res.data?.data; }); }); + useEffect(() => { + const { unsubscribe } = EventEmitter.subscribe( + EmitterKey.DMS_Reload_Initial_Data, + refresh + ); + return unsubscribe; + }, [refresh]); + const bannerState = useMemo(() => { const modules = bannerData?.modules ?? []; const isModuleEnabled = (module?: IAIModuleBannerCards) => diff --git a/packages/base/src/page/Home/WorkflowStatCards/__tests__/index.test.tsx b/packages/base/src/page/Home/WorkflowStatCards/__tests__/index.test.tsx index fb02dd09a..1f031d914 100644 --- a/packages/base/src/page/Home/WorkflowStatCards/__tests__/index.test.tsx +++ b/packages/base/src/page/Home/WorkflowStatCards/__tests__/index.test.tsx @@ -5,6 +5,8 @@ import { sqleMockApi } from '@actiontech/shared/lib/testUtil'; import { useTypedNavigate } from '@actiontech/shared'; import { ROUTE_PATHS } from '@actiontech/dms-kit'; import { GetGlobalWorkflowListV2FilterCardEnum } from '@actiontech/shared/lib/api/sqle/service/GlobalDashboard/index.enum'; +import EventEmitter from '../../../../utils/EventEmitter'; +import EmitterKey from '../../../../data/EmitterKey'; jest.mock('@actiontech/shared', () => ({ ...jest.requireActual('@actiontech/shared'), @@ -152,4 +154,18 @@ describe('WorkflowStatCards', () => { } ); }); + + it('should refresh workflow statistics when DMS_Reload_Initial_Data event is emitted', async () => { + baseSuperRender(); + + await act(async () => jest.advanceTimersByTime(3000)); + expect(getGlobalWorkflowStatisticsSpy).toHaveBeenCalledTimes(1); + + await act(async () => { + EventEmitter.emit(EmitterKey.DMS_Reload_Initial_Data); + await jest.advanceTimersByTime(3000); + }); + + expect(getGlobalWorkflowStatisticsSpy).toHaveBeenCalledTimes(2); + }); }); diff --git a/packages/base/src/page/Home/WorkflowStatCards/index.tsx b/packages/base/src/page/Home/WorkflowStatCards/index.tsx index 4d9400e57..ec47a779d 100644 --- a/packages/base/src/page/Home/WorkflowStatCards/index.tsx +++ b/packages/base/src/page/Home/WorkflowStatCards/index.tsx @@ -1,4 +1,5 @@ import { useRequest } from 'ahooks'; +import { useEffect } from 'react'; import { useTranslation } from 'react-i18next'; import { Card, Typography } from 'antd'; import { @@ -13,6 +14,8 @@ import { useTypedNavigate } from '@actiontech/shared'; import { ROUTE_PATHS } from '@actiontech/dms-kit'; import { RightOutlined } from '@ant-design/icons'; import { WorkflowStatCardsWrapper, WorkflowCardItemWrapper } from './style'; +import EventEmitter from '../../../utils/EventEmitter'; +import EmitterKey from '../../../data/EmitterKey'; const WORKFLOW_ACCENT: Record = { [GetGlobalWorkflowListV2FilterCardEnum.pending_for_me]: '#fa8c16', @@ -25,10 +28,22 @@ const WorkflowStatCards: React.FC = () => { const { t } = useTranslation(); const navigate = useTypedNavigate(); - const { data: workflowStats, loading } = useRequest(() => + const { + data: workflowStats, + loading, + refresh + } = useRequest(() => GlobalDashboardService.GetGlobalWorkflowStatisticsV2({}) ); + useEffect(() => { + const { unsubscribe } = EventEmitter.subscribe( + EmitterKey.DMS_Reload_Initial_Data, + refresh + ); + return unsubscribe; + }, [refresh]); + const cards = [ { key: GetGlobalWorkflowListV2FilterCardEnum.pending_for_me,