117 lines
3.4 KiB
TypeScript
117 lines
3.4 KiB
TypeScript
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)
|
|
})
|
|
})
|