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,70 @@
|
||||
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
||||
import { getShareAuthorizedBlob } from '../../src/blobs';
|
||||
import { createBlobInlineCache, inlineBlobRefsForShare } from '../../src/util/inlineBlobsForShare';
|
||||
|
||||
import type { StoredBlob } from '../../src/blobs';
|
||||
|
||||
vi.mock('../../src/blobs', () => ({
|
||||
getShareAuthorizedBlob: vi.fn(),
|
||||
}));
|
||||
|
||||
vi.mock('../../src/logger', () => ({
|
||||
default: {
|
||||
warn: vi.fn(),
|
||||
},
|
||||
}));
|
||||
|
||||
describe('inlineBlobRefsForShare', () => {
|
||||
const hash = 'a'.repeat(64);
|
||||
const uri = `promptfoo://blob/${hash}`;
|
||||
const bytes = Buffer.from('image-bytes');
|
||||
const storedBlob: StoredBlob = {
|
||||
data: bytes,
|
||||
metadata: {
|
||||
createdAt: '2026-06-08T00:00:00.000Z',
|
||||
key: 'blob-key',
|
||||
mimeType: 'image/png',
|
||||
provider: 'filesystem',
|
||||
sizeBytes: bytes.length,
|
||||
},
|
||||
};
|
||||
|
||||
beforeEach(() => {
|
||||
vi.resetAllMocks();
|
||||
vi.mocked(getShareAuthorizedBlob).mockResolvedValue(storedBlob);
|
||||
});
|
||||
|
||||
it('inlines authorized blob URIs as data URLs', async () => {
|
||||
const result = await inlineBlobRefsForShare(
|
||||
{ output: uri },
|
||||
createBlobInlineCache(),
|
||||
'local-eval-1',
|
||||
);
|
||||
|
||||
expect(result.output).toBe(`data:image/png;base64,${bytes.toString('base64')}`);
|
||||
expect(getShareAuthorizedBlob).toHaveBeenCalledWith(hash, 'local-eval-1');
|
||||
});
|
||||
|
||||
it('does not inline a copied blob URI that is not share-authorized for the eval', async () => {
|
||||
vi.mocked(getShareAuthorizedBlob).mockResolvedValue(null);
|
||||
|
||||
const result = await inlineBlobRefsForShare(
|
||||
{ output: uri },
|
||||
createBlobInlineCache(),
|
||||
'local-eval-1',
|
||||
);
|
||||
|
||||
expect(result.output).toBe(uri);
|
||||
expect(getShareAuthorizedBlob).toHaveBeenCalledWith(hash, 'local-eval-1');
|
||||
});
|
||||
|
||||
it('caches the authorization decision across calls sharing a cache', async () => {
|
||||
vi.mocked(getShareAuthorizedBlob).mockResolvedValue(null);
|
||||
const cache = createBlobInlineCache();
|
||||
|
||||
await inlineBlobRefsForShare({ output: uri }, cache, 'local-eval-1');
|
||||
await inlineBlobRefsForShare([uri], cache, 'local-eval-1');
|
||||
|
||||
expect(getShareAuthorizedBlob).toHaveBeenCalledOnce();
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user