youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-06 10:28:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

test(billing): mock utility functions for plan limits in VectorSpaceFull tests

Browse Source 
- Introduce mock for getPlanVectorSpaceLimitMB to control plan limits for sandbox and team
- Update VectorSpaceFull test to reflect correct storage limit display based on plan type
This commit is contained in:
CodingOnStar
2026-01-20 17:32:12 +08:00
parent 9c84e1fe57
commit a8fda37abf
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
web/app/components/billing/vector-space-full/index.spec.tsx
View File

@ -21,6 +21,18 @@ vi.mock('../upgrade-btn', () => ({
default: () => <button data-testid="vector-upgrade-btn" type="button">Upgrade</button>, default: () => <button data-testid="vector-upgrade-btn" type="button">Upgrade</button>,
})) }))
// Mock utils to control threshold and plan limits
vi.mock('../utils', () => ({
getPlanVectorSpaceLimitMB: (planType: string) => {
// Return 5 for sandbox (threshold) and 100 for team
if (planType === 'sandbox')
return 5
if (planType === 'team')
return 100
return 0
},
}))
describe('VectorSpaceFull', () => { describe('VectorSpaceFull', () => {
const planMock = { const planMock = {
type: 'team', type: 'team',
@ -52,6 +64,6 @@ describe('VectorSpaceFull', () => {
render(<VectorSpaceFull />) render(<VectorSpaceFull />)
expect(screen.getByText('8')).toBeInTheDocument() expect(screen.getByText('8')).toBeInTheDocument()
expect(screen.getByText('10MB')).toBeInTheDocument() expect(screen.getByText('100MB')).toBeInTheDocument()
}) })
}) })