Spaces:
Sleeping
Sleeping
File size: 3,489 Bytes
b6ecafa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 | import { describe, it, expect, vi, beforeEach } from 'vitest'
const mockGet = vi.fn()
const mockPrepare = vi.fn(() => ({ get: mockGet }))
vi.mock('@/lib/db', () => ({
getDatabase: () => ({ prepare: mockPrepare }),
}))
import {
getActiveProfile,
shouldScanSecrets,
shouldAuditMcpCalls,
shouldBlockOnSecretDetection,
getRateLimitMultiplier,
} from '@/lib/hook-profiles'
describe('getActiveProfile', () => {
beforeEach(() => {
vi.clearAllMocks()
})
it('returns standard profile by default', () => {
mockGet.mockReturnValue(undefined)
const profile = getActiveProfile()
expect(profile.level).toBe('standard')
expect(profile.scanSecrets).toBe(true)
expect(profile.auditMcpCalls).toBe(true)
expect(profile.blockOnSecretDetection).toBe(false)
expect(profile.rateLimitMultiplier).toBe(1.0)
})
it('returns minimal profile when set', () => {
mockGet.mockReturnValue({ value: 'minimal' })
const profile = getActiveProfile()
expect(profile.level).toBe('minimal')
expect(profile.scanSecrets).toBe(false)
expect(profile.auditMcpCalls).toBe(false)
expect(profile.blockOnSecretDetection).toBe(false)
expect(profile.rateLimitMultiplier).toBe(2.0)
})
it('returns strict profile when set', () => {
mockGet.mockReturnValue({ value: 'strict' })
const profile = getActiveProfile()
expect(profile.level).toBe('strict')
expect(profile.scanSecrets).toBe(true)
expect(profile.auditMcpCalls).toBe(true)
expect(profile.blockOnSecretDetection).toBe(true)
expect(profile.rateLimitMultiplier).toBe(0.5)
})
it('falls back to standard for unknown profile value', () => {
mockGet.mockReturnValue({ value: 'nonexistent' })
const profile = getActiveProfile()
expect(profile.level).toBe('standard')
})
})
describe('shouldScanSecrets', () => {
it('returns true for standard profile', () => {
mockGet.mockReturnValue({ value: 'standard' })
expect(shouldScanSecrets()).toBe(true)
})
it('returns false for minimal profile', () => {
mockGet.mockReturnValue({ value: 'minimal' })
expect(shouldScanSecrets()).toBe(false)
})
it('returns true for strict profile', () => {
mockGet.mockReturnValue({ value: 'strict' })
expect(shouldScanSecrets()).toBe(true)
})
})
describe('shouldAuditMcpCalls', () => {
it('returns false for minimal profile', () => {
mockGet.mockReturnValue({ value: 'minimal' })
expect(shouldAuditMcpCalls()).toBe(false)
})
it('returns true for standard profile', () => {
mockGet.mockReturnValue({ value: 'standard' })
expect(shouldAuditMcpCalls()).toBe(true)
})
})
describe('shouldBlockOnSecretDetection', () => {
it('returns false for standard profile', () => {
mockGet.mockReturnValue({ value: 'standard' })
expect(shouldBlockOnSecretDetection()).toBe(false)
})
it('returns true for strict profile', () => {
mockGet.mockReturnValue({ value: 'strict' })
expect(shouldBlockOnSecretDetection()).toBe(true)
})
})
describe('getRateLimitMultiplier', () => {
it('returns 1.0 for standard', () => {
mockGet.mockReturnValue({ value: 'standard' })
expect(getRateLimitMultiplier()).toBe(1.0)
})
it('returns 2.0 for minimal', () => {
mockGet.mockReturnValue({ value: 'minimal' })
expect(getRateLimitMultiplier()).toBe(2.0)
})
it('returns 0.5 for strict', () => {
mockGet.mockReturnValue({ value: 'strict' })
expect(getRateLimitMultiplier()).toBe(0.5)
})
})
|