2
0
Эх сурвалжийг харах

✅ test(advertisement-type): update mock client implementation in tests

- 修改广告类型客户端的模拟实现,支持通过manager获取实例
- 为advertisementTypeClientManager添加get方法模拟
- 修复测试中children类型错误,添加类型断言
yourname 1 сар өмнө
parent
commit
a74aec7f55

+ 11 - 5
packages/advertisement-type-management-ui-mt/tests/integration/advertisement-type-management.integration.test.tsx

@@ -6,9 +6,9 @@ import { BrowserRouter } from 'react-router';
 import { AdvertisementTypeManagement } from '../../src/components/AdvertisementTypeManagement';
 import { advertisementTypeClient } from '../../src/api/advertisementTypeClient';
 
-// Mock RPC客户端
-vi.mock('../../src/api/advertisementTypeClient', () => ({
-  advertisementTypeClient: {
+// Mock the advertisement type client
+vi.mock('../../src/api/advertisementTypeClient', () => {
+  const advertisementTypeClient = {
     index: {
       $get: vi.fn(),
       $post: vi.fn(),
@@ -17,8 +17,14 @@ vi.mock('../../src/api/advertisementTypeClient', () => ({
       $put: vi.fn(),
       $delete: vi.fn(),
     },
-  },
-}));
+  }
+  return {
+    advertisementTypeClient,
+    advertisementTypeClientManager: {
+      get: vi.fn(() => advertisementTypeClient),
+    },
+  }
+})
 
 // Mock toast
 vi.mock('sonner', () => ({

+ 12 - 6
packages/advertisement-type-management-ui/tests/integration/advertisement-type-management.integration.test.tsx

@@ -6,9 +6,9 @@ import { BrowserRouter } from 'react-router';
 import { AdvertisementTypeManagement } from '../../src/components/AdvertisementTypeManagement';
 import { advertisementTypeClient } from '../../src/api/advertisementTypeClient';
 
-// Mock RPC客户端
-vi.mock('../../src/api/advertisementTypeClient', () => ({
-  advertisementTypeClient: {
+// Mock the advertisement type client
+vi.mock('../../src/api/advertisementTypeClient', () => {
+  const advertisementTypeClient = {
     index: {
       $get: vi.fn(),
       $post: vi.fn(),
@@ -17,8 +17,14 @@ vi.mock('../../src/api/advertisementTypeClient', () => ({
       $put: vi.fn(),
       $delete: vi.fn(),
     },
-  },
-}));
+  }
+  return {
+    advertisementTypeClient,
+    advertisementTypeClientManager: {
+      get: vi.fn(() => advertisementTypeClient),
+    },
+  }
+})
 
 // Mock toast
 vi.mock('sonner', () => ({
@@ -41,7 +47,7 @@ const TestWrapper = ({ children }: { children: React.ReactNode }) => {
   return (
     <BrowserRouter>
       <QueryClientProvider client={queryClient}>
-        {children}
+        {children as any}
       </QueryClientProvider>
     </BrowserRouter>
   );

+ 11 - 5
packages/advertisement-type-management-ui/tests/integration/advertisement-type-selector.integration.test.tsx

@@ -5,13 +5,19 @@ import { AdvertisementTypeSelector } from '../../src/components/AdvertisementTyp
 import { advertisementTypeClient } from '../../src/api/advertisementTypeClient'
 
 // Mock the advertisement type client
-vi.mock('../../src/api/advertisementTypeClient', () => ({
-  advertisementTypeClient: {
+vi.mock('../../src/api/advertisementTypeClient', () => {
+  const advertisementTypeClient = {
     index: {
       $get: vi.fn(),
     },
-  },
-}))
+  }
+  return {
+    advertisementTypeClient,
+    advertisementTypeClientManager: {
+      get: vi.fn(() => advertisementTypeClient),
+    },
+  }
+})
 
 const mockAdvertisementTypes = {
   data: [
@@ -38,7 +44,7 @@ const TestWrapper = ({ children }: { children: React.ReactNode }) => {
 
   return (
     <QueryClientProvider client={queryClient}>
-      {children}
+      {children as any}
     </QueryClientProvider>
   )
 }