chore: import upstream snapshot with attribution
CI / Shell Format Check (push) Has been cancelled
CI / Check Ruby (3.4) (push) Has been cancelled
CI / CI Config (push) Has been cancelled
CI / Test on Node ${{ matrix.node }} and ${{ matrix.os }}${{ matrix.shard && format(' (shard {0}/3)', matrix.shard) || '' }} (push) Has been cancelled
CI / Build on Node ${{ matrix.node }} (push) Has been cancelled
CI / Style Check (push) Has been cancelled
CI / Generate Assets (push) Has been cancelled
CI / Check Python (3.14) (push) Has been cancelled
CI / Check Python (3.9) (push) Has been cancelled
CI / Build Docs (push) Has been cancelled
CI / Code Scan Action (push) Has been cancelled
CI / Site tests (push) Has been cancelled
CI / webui tests (push) Has been cancelled
CI / Run Integration Tests (push) Has been cancelled
CI / Run Smoke Tests (push) Has been cancelled
CI / Go Tests (push) Has been cancelled
CI / Share Test (push) Has been cancelled
CI / Redteam (Production API) (push) Has been cancelled
CI / Redteam (Staging API) (push) Has been cancelled
CI / GitHub Actions Lint (push) Has been cancelled
CI / Check Ruby (3.0) (push) Has been cancelled
release-please / release-please (push) Has been cancelled
release-please / build (push) Has been cancelled
release-please / publish-npm (push) Has been cancelled
release-please / publish-npm-backfill (push) Has been cancelled
release-please / docker (push) Has been cancelled
release-please / publish-code-scan-action (push) Has been cancelled
release-please / attest-code-scan-action (push) Has been cancelled
Deploy local.promptfoo.app / Deploy to Cloudflare Pages (push) Has been cancelled
Test and Publish Multi-arch Docker Image / test (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-amd64 platform:linux/amd64 runner:ubuntu-latest]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-arm64 platform:linux/arm64 runner:ubuntu-24.04-arm]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / merge-docker-digests (push) Has been cancelled
Test and Publish Multi-arch Docker Image / Attest Multi-arch Image (push) Has been cancelled
Validate Renovate Config / Validate Renovate Configuration (push) Has been cancelled
CI / Shell Format Check (push) Has been cancelled
CI / Check Ruby (3.4) (push) Has been cancelled
CI / CI Config (push) Has been cancelled
CI / Test on Node ${{ matrix.node }} and ${{ matrix.os }}${{ matrix.shard && format(' (shard {0}/3)', matrix.shard) || '' }} (push) Has been cancelled
CI / Build on Node ${{ matrix.node }} (push) Has been cancelled
CI / Style Check (push) Has been cancelled
CI / Generate Assets (push) Has been cancelled
CI / Check Python (3.14) (push) Has been cancelled
CI / Check Python (3.9) (push) Has been cancelled
CI / Build Docs (push) Has been cancelled
CI / Code Scan Action (push) Has been cancelled
CI / Site tests (push) Has been cancelled
CI / webui tests (push) Has been cancelled
CI / Run Integration Tests (push) Has been cancelled
CI / Run Smoke Tests (push) Has been cancelled
CI / Go Tests (push) Has been cancelled
CI / Share Test (push) Has been cancelled
CI / Redteam (Production API) (push) Has been cancelled
CI / Redteam (Staging API) (push) Has been cancelled
CI / GitHub Actions Lint (push) Has been cancelled
CI / Check Ruby (3.0) (push) Has been cancelled
release-please / release-please (push) Has been cancelled
release-please / build (push) Has been cancelled
release-please / publish-npm (push) Has been cancelled
release-please / publish-npm-backfill (push) Has been cancelled
release-please / docker (push) Has been cancelled
release-please / publish-code-scan-action (push) Has been cancelled
release-please / attest-code-scan-action (push) Has been cancelled
Deploy local.promptfoo.app / Deploy to Cloudflare Pages (push) Has been cancelled
Test and Publish Multi-arch Docker Image / test (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-amd64 platform:linux/amd64 runner:ubuntu-latest]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-arm64 platform:linux/arm64 runner:ubuntu-24.04-arm]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / merge-docker-digests (push) Has been cancelled
Test and Publish Multi-arch Docker Image / Attest Multi-arch Image (push) Has been cancelled
Validate Renovate Config / Validate Renovate Configuration (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,204 @@
|
||||
import { describe, expect, it, vi } from 'vitest';
|
||||
import ModelAudit from '../../src/models/modelAudit';
|
||||
|
||||
import type { ModelAuditScanResults } from '../../src/types/modelAudit';
|
||||
|
||||
// Mock the database
|
||||
vi.mock('../../src/database', () => ({
|
||||
getDb: () => ({
|
||||
insert: () => ({
|
||||
values: () => ({
|
||||
run: vi.fn().mockResolvedValue({}),
|
||||
}),
|
||||
}),
|
||||
select: () => ({
|
||||
from: () => ({
|
||||
where: () => ({
|
||||
get: vi.fn().mockResolvedValue(null),
|
||||
orderBy: () => ({
|
||||
get: vi.fn().mockResolvedValue(null),
|
||||
}),
|
||||
}),
|
||||
orderBy: () => ({
|
||||
all: vi.fn().mockResolvedValue([]),
|
||||
limit: () => ({
|
||||
all: vi.fn().mockResolvedValue([]),
|
||||
}),
|
||||
}),
|
||||
}),
|
||||
}),
|
||||
}),
|
||||
}));
|
||||
|
||||
describe('ModelAudit', () => {
|
||||
describe('hasErrors logic', () => {
|
||||
it('should set hasErrors to true when has_errors is true in results', () => {
|
||||
const results: ModelAuditScanResults = {
|
||||
has_errors: true,
|
||||
issues: [{ severity: 'warning', message: 'Test warning' }],
|
||||
};
|
||||
|
||||
const audit = new ModelAudit({
|
||||
modelPath: '/test/path',
|
||||
results,
|
||||
});
|
||||
|
||||
expect(audit.hasErrors).toBe(true);
|
||||
});
|
||||
|
||||
it('should set hasErrors to true when has_errors is false but critical issues exist', () => {
|
||||
const results: ModelAuditScanResults = {
|
||||
has_errors: false, // CLI tool incorrectly says no errors
|
||||
issues: [
|
||||
{ severity: 'critical', message: 'Test critical issue' },
|
||||
{ severity: 'warning', message: 'Test warning' },
|
||||
],
|
||||
};
|
||||
|
||||
const audit = new ModelAudit({
|
||||
modelPath: '/test/path',
|
||||
results,
|
||||
});
|
||||
|
||||
expect(audit.hasErrors).toBe(true);
|
||||
});
|
||||
|
||||
it('should set hasErrors to true when has_errors is false but error issues exist', () => {
|
||||
const results: ModelAuditScanResults = {
|
||||
has_errors: false, // CLI tool incorrectly says no errors
|
||||
issues: [
|
||||
{ severity: 'error', message: 'Test error issue' },
|
||||
{ severity: 'warning', message: 'Test warning' },
|
||||
],
|
||||
};
|
||||
|
||||
const audit = new ModelAudit({
|
||||
modelPath: '/test/path',
|
||||
results,
|
||||
});
|
||||
|
||||
expect(audit.hasErrors).toBe(true);
|
||||
});
|
||||
|
||||
it('should set hasErrors to true when warning findings exist', () => {
|
||||
const results: ModelAuditScanResults = {
|
||||
has_errors: false,
|
||||
issues: [
|
||||
{ severity: 'warning', message: 'Test warning' },
|
||||
{ severity: 'info', message: 'Test info' },
|
||||
],
|
||||
};
|
||||
|
||||
const audit = new ModelAudit({
|
||||
modelPath: '/test/path',
|
||||
results,
|
||||
});
|
||||
|
||||
expect(audit.hasErrors).toBe(true);
|
||||
});
|
||||
|
||||
it('should set hasErrors to false when has_errors is false and no issues exist', () => {
|
||||
const results: ModelAuditScanResults = {
|
||||
has_errors: false,
|
||||
issues: [],
|
||||
};
|
||||
|
||||
const audit = new ModelAudit({
|
||||
modelPath: '/test/path',
|
||||
results,
|
||||
});
|
||||
|
||||
expect(audit.hasErrors).toBe(false);
|
||||
});
|
||||
|
||||
it('should set hasErrors to true when failed checks exist without issue rows', () => {
|
||||
const results: ModelAuditScanResults = {
|
||||
has_errors: false,
|
||||
failed_checks: 1,
|
||||
issues: [],
|
||||
checks: [{ name: 'pickle', status: 'failed', message: 'Check failed' }],
|
||||
};
|
||||
|
||||
const audit = new ModelAudit({
|
||||
modelPath: '/test/path',
|
||||
results,
|
||||
});
|
||||
|
||||
expect(audit.hasErrors).toBe(true);
|
||||
});
|
||||
|
||||
it('should handle missing issues array gracefully', () => {
|
||||
const results: ModelAuditScanResults = {
|
||||
has_errors: false,
|
||||
// issues not provided
|
||||
};
|
||||
|
||||
const audit = new ModelAudit({
|
||||
modelPath: '/test/path',
|
||||
results,
|
||||
});
|
||||
|
||||
expect(audit.hasErrors).toBe(false);
|
||||
});
|
||||
|
||||
it('should prioritize explicit hasErrors over computed value', () => {
|
||||
const results: ModelAuditScanResults = {
|
||||
has_errors: false,
|
||||
issues: [{ severity: 'critical', message: 'Test critical issue' }],
|
||||
};
|
||||
|
||||
const audit = new ModelAudit({
|
||||
modelPath: '/test/path',
|
||||
results,
|
||||
hasErrors: false, // Explicitly set to false
|
||||
});
|
||||
|
||||
// Should still be false because explicit value takes precedence
|
||||
expect(audit.hasErrors).toBe(false);
|
||||
});
|
||||
});
|
||||
|
||||
describe('ModelAudit.create', () => {
|
||||
it('should properly set hasErrors in create method with critical issues', async () => {
|
||||
const results: ModelAuditScanResults = {
|
||||
has_errors: false, // CLI tool incorrectly says no errors
|
||||
issues: [{ severity: 'critical', message: 'Test critical issue' }],
|
||||
};
|
||||
|
||||
const audit = await ModelAudit.create({
|
||||
modelPath: '/test/path',
|
||||
results,
|
||||
});
|
||||
|
||||
expect(audit.hasErrors).toBe(true);
|
||||
});
|
||||
|
||||
it('should properly set hasErrors in create method with error issues', async () => {
|
||||
const results: ModelAuditScanResults = {
|
||||
has_errors: false, // CLI tool incorrectly says no errors
|
||||
issues: [{ severity: 'error', message: 'Test error issue' }],
|
||||
};
|
||||
|
||||
const audit = await ModelAudit.create({
|
||||
modelPath: '/test/path',
|
||||
results,
|
||||
});
|
||||
|
||||
expect(audit.hasErrors).toBe(true);
|
||||
});
|
||||
|
||||
it('should keep warning findings visible in create method', async () => {
|
||||
const results: ModelAuditScanResults = {
|
||||
has_errors: false,
|
||||
issues: [{ severity: 'warning', message: 'Test warning' }],
|
||||
};
|
||||
|
||||
const audit = await ModelAudit.create({
|
||||
modelPath: '/test/path',
|
||||
results,
|
||||
});
|
||||
|
||||
expect(audit.hasErrors).toBe(true);
|
||||
});
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user