fix: tests

This commit is contained in:
yyh
2026-03-11 20:28:37 +08:00
parent 5d07ccce59
commit 715a0fabfc
4 changed files with 90 additions and 100 deletions

View File

@ -127,6 +127,14 @@ vi.mock('@/service/use-common', () => ({
],
},
}),
useCurrentWorkspace: () => ({
data: {
trial_credits: 1000,
trial_credits_used: 100,
next_credit_reset_date: undefined,
},
isPending: false,
}),
}))
vi.mock('@/app/components/header/account-setting/model-provider-page/hooks', () => ({

View File

@ -30,6 +30,21 @@ vi.mock('@/context/app-context', () => ({
},
}))
vi.mock('@/service/use-common', async () => {
const actual = await vi.importActual<typeof import('@/service/use-common')>('@/service/use-common')
return {
...actual,
useCurrentWorkspace: () => ({
data: {
trial_credits: 1000,
trial_credits_used: 100,
next_credit_reset_date: undefined,
},
isPending: false,
}),
}
})
// Mock model-provider-page hooks
vi.mock('@/app/components/header/account-setting/model-provider-page/hooks', () => ({
useModelList: () => ({ data: [], mutate: vi.fn(), isLoading: false }),