Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 5 additions & 2 deletions assistant/src/__tests__/actor-token-service.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -652,11 +652,14 @@ describe('resolveLocalIpcGuardianContext', () => {
expect(ctx.sourceChannel).toBe('vellum');
});

test('returns fallback guardian context when no vellum binding exists', () => {
// No binding created — fresh DB state
test('returns guardian context with principal when no vellum binding exists (pre-bootstrap self-heal)', () => {
// No binding created — fresh DB state. Pre-bootstrap path self-heals
// by creating a vellum binding, then resolves through the shared pipeline
// with correct field names (conversationExternalId, actorExternalId).
const ctx = resolveLocalIpcGuardianContext();
expect(ctx.trustClass).toBe('guardian');
expect(ctx.sourceChannel).toBe('vellum');
expect(ctx.guardianPrincipalId).toBeDefined();
});

test('respects custom sourceChannel parameter', () => {
Expand Down
54 changes: 44 additions & 10 deletions assistant/src/__tests__/canonical-guardian-store.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -41,6 +41,10 @@ import { getDb, initializeDb, resetDb } from '../memory/db.js';

initializeDb();

// All decisionable kinds (tool_approval, pending_question, access_request)
// require a guardianPrincipalId. Use a constant for test fixtures.
const TEST_PRINCIPAL = 'test-principal-id';

function resetTables(): void {
const db = getDb();
db.run('DELETE FROM canonical_guardian_deliveries');
Expand Down Expand Up @@ -71,6 +75,7 @@ describe('canonical-guardian-store', () => {
conversationId: 'conv-1',
requesterExternalUserId: 'user-1',
guardianExternalUserId: 'guardian-1',
guardianPrincipalId: TEST_PRINCIPAL,
callSessionId: 'session-1',
pendingQuestionId: 'pq-1',
questionText: 'Can I run this tool?',
Expand All @@ -94,6 +99,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'access_request',
sourceType: 'channel',
guardianPrincipalId: TEST_PRINCIPAL,
});

expect(req.id).toBeTruthy();
Expand All @@ -111,6 +117,7 @@ describe('canonical-guardian-store', () => {
const created = createCanonicalGuardianRequest({
kind: 'tool_approval',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});

const fetched = getCanonicalGuardianRequest(created.id);
Expand All @@ -127,16 +134,16 @@ describe('canonical-guardian-store', () => {
// ── listCanonicalGuardianRequests ─────────────────────────────────

test('lists all requests with no filters', () => {
createCanonicalGuardianRequest({ kind: 'tool_approval', sourceType: 'voice' });
createCanonicalGuardianRequest({ kind: 'access_request', sourceType: 'channel' });
createCanonicalGuardianRequest({ kind: 'tool_approval', sourceType: 'voice', guardianPrincipalId: TEST_PRINCIPAL });
createCanonicalGuardianRequest({ kind: 'access_request', sourceType: 'channel', guardianPrincipalId: TEST_PRINCIPAL });

const all = listCanonicalGuardianRequests();
expect(all).toHaveLength(2);
});

test('filters by status', () => {
createCanonicalGuardianRequest({ kind: 'tool_approval', sourceType: 'voice' });
const req2 = createCanonicalGuardianRequest({ kind: 'access_request', sourceType: 'channel' });
createCanonicalGuardianRequest({ kind: 'tool_approval', sourceType: 'voice', guardianPrincipalId: TEST_PRINCIPAL });
const req2 = createCanonicalGuardianRequest({ kind: 'access_request', sourceType: 'channel', guardianPrincipalId: TEST_PRINCIPAL });
updateCanonicalGuardianRequest(req2.id, { status: 'approved' });

const pending = listCanonicalGuardianRequests({ status: 'pending' });
Expand All @@ -153,11 +160,13 @@ describe('canonical-guardian-store', () => {
kind: 'tool_approval',
sourceType: 'voice',
guardianExternalUserId: 'guardian-A',
guardianPrincipalId: TEST_PRINCIPAL,
});
createCanonicalGuardianRequest({
kind: 'tool_approval',
sourceType: 'voice',
guardianExternalUserId: 'guardian-B',
guardianPrincipalId: TEST_PRINCIPAL,
});

const filtered = listCanonicalGuardianRequests({ guardianExternalUserId: 'guardian-A' });
Expand All @@ -170,30 +179,32 @@ describe('canonical-guardian-store', () => {
kind: 'tool_approval',
sourceType: 'voice',
conversationId: 'conv-X',
guardianPrincipalId: TEST_PRINCIPAL,
});
createCanonicalGuardianRequest({
kind: 'tool_approval',
sourceType: 'voice',
conversationId: 'conv-Y',
guardianPrincipalId: TEST_PRINCIPAL,
});

const filtered = listCanonicalGuardianRequests({ conversationId: 'conv-X' });
expect(filtered).toHaveLength(1);
});

test('filters by sourceType', () => {
createCanonicalGuardianRequest({ kind: 'tool_approval', sourceType: 'voice' });
createCanonicalGuardianRequest({ kind: 'tool_approval', sourceType: 'channel' });
createCanonicalGuardianRequest({ kind: 'tool_approval', sourceType: 'desktop' });
createCanonicalGuardianRequest({ kind: 'tool_approval', sourceType: 'voice', guardianPrincipalId: TEST_PRINCIPAL });
createCanonicalGuardianRequest({ kind: 'tool_approval', sourceType: 'channel', guardianPrincipalId: TEST_PRINCIPAL });
createCanonicalGuardianRequest({ kind: 'tool_approval', sourceType: 'desktop', guardianPrincipalId: TEST_PRINCIPAL });

const voiceOnly = listCanonicalGuardianRequests({ sourceType: 'voice' });
expect(voiceOnly).toHaveLength(1);
});

test('filters by kind', () => {
createCanonicalGuardianRequest({ kind: 'tool_approval', sourceType: 'voice' });
createCanonicalGuardianRequest({ kind: 'pending_question', sourceType: 'voice' });
createCanonicalGuardianRequest({ kind: 'access_request', sourceType: 'channel' });
createCanonicalGuardianRequest({ kind: 'tool_approval', sourceType: 'voice', guardianPrincipalId: TEST_PRINCIPAL });
createCanonicalGuardianRequest({ kind: 'pending_question', sourceType: 'voice', guardianPrincipalId: TEST_PRINCIPAL });
createCanonicalGuardianRequest({ kind: 'access_request', sourceType: 'channel', guardianPrincipalId: TEST_PRINCIPAL });

const toolOnly = listCanonicalGuardianRequests({ kind: 'tool_approval' });
expect(toolOnly).toHaveLength(1);
Expand All @@ -204,16 +215,19 @@ describe('canonical-guardian-store', () => {
kind: 'tool_approval',
sourceType: 'voice',
guardianExternalUserId: 'guardian-A',
guardianPrincipalId: TEST_PRINCIPAL,
});
createCanonicalGuardianRequest({
kind: 'tool_approval',
sourceType: 'channel',
guardianExternalUserId: 'guardian-A',
guardianPrincipalId: TEST_PRINCIPAL,
});
createCanonicalGuardianRequest({
kind: 'access_request',
sourceType: 'voice',
guardianExternalUserId: 'guardian-A',
guardianPrincipalId: TEST_PRINCIPAL,
});

const filtered = listCanonicalGuardianRequests({
Expand All @@ -230,6 +244,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'tool_approval',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});

const updated = updateCanonicalGuardianRequest(req.id, {
Expand Down Expand Up @@ -260,6 +275,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'tool_approval',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});

const resolved = resolveCanonicalGuardianRequest(req.id, 'pending', {
Expand All @@ -278,6 +294,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'tool_approval',
sourceType: 'channel',
guardianPrincipalId: TEST_PRINCIPAL,
});

const resolved = resolveCanonicalGuardianRequest(req.id, 'pending', {
Expand All @@ -293,6 +310,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'tool_approval',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});

// Try to resolve with wrong expected status
Expand All @@ -311,6 +329,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'tool_approval',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});

// First resolve succeeds
Expand Down Expand Up @@ -353,6 +372,7 @@ describe('canonical-guardian-store', () => {
sourceChannel: 'twilio',
conversationId: 'conv-voice-1',
guardianExternalUserId: 'guardian-phone',
guardianPrincipalId: TEST_PRINCIPAL,
callSessionId: 'call-123',
pendingQuestionId: 'pq-456',
questionText: 'What is the gate code?',
Expand All @@ -374,6 +394,7 @@ describe('canonical-guardian-store', () => {
conversationId: 'conv-tg-1',
requesterExternalUserId: 'requester-tg-user',
guardianExternalUserId: 'guardian-tg-user',
guardianPrincipalId: TEST_PRINCIPAL,
toolName: 'execute_code',
inputDigest: 'sha256:abcdef',
expiresAt: new Date(Date.now() + 120_000).toISOString(),
Expand All @@ -394,6 +415,7 @@ describe('canonical-guardian-store', () => {
sourceType: 'desktop',
conversationId: 'conv-desktop-1',
guardianExternalUserId: 'guardian-desktop',
guardianPrincipalId: TEST_PRINCIPAL,
questionText: 'User wants to access settings',
});

Expand All @@ -408,6 +430,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'tool_approval',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});

const d1 = createCanonicalGuardianDelivery({
Expand Down Expand Up @@ -436,6 +459,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'tool_approval',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});

const deliveries = listCanonicalGuardianDeliveries(req.id);
Expand All @@ -446,10 +470,12 @@ describe('canonical-guardian-store', () => {
const pendingReq = createCanonicalGuardianRequest({
kind: 'pending_question',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});
const resolvedReq = createCanonicalGuardianRequest({
kind: 'pending_question',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});
updateCanonicalGuardianRequest(resolvedReq.id, { status: 'approved' });

Expand All @@ -476,6 +502,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'pending_question',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});

createCanonicalGuardianDelivery({
Expand All @@ -498,6 +525,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'tool_approval',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});
const delivery = createCanonicalGuardianDelivery({
requestId: req.id,
Expand Down Expand Up @@ -525,6 +553,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'pending_question',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});
createCanonicalGuardianDelivery({
requestId: req.id,
Expand All @@ -544,10 +573,12 @@ describe('canonical-guardian-store', () => {
const pendingReq = createCanonicalGuardianRequest({
kind: 'pending_question',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});
const resolvedReq = createCanonicalGuardianRequest({
kind: 'pending_question',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});
updateCanonicalGuardianRequest(resolvedReq.id, { status: 'approved' });

Expand All @@ -574,6 +605,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'pending_question',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});

// Two delivery rows targeting the same chat for the same request
Expand Down Expand Up @@ -602,6 +634,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'pending_question',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});
createCanonicalGuardianDelivery({
requestId: req.id,
Expand All @@ -620,6 +653,7 @@ describe('canonical-guardian-store', () => {
const req = createCanonicalGuardianRequest({
kind: 'pending_question',
sourceType: 'voice',
guardianPrincipalId: TEST_PRINCIPAL,
});
createCanonicalGuardianDelivery({
requestId: req.id,
Expand Down
2 changes: 2 additions & 0 deletions assistant/src/__tests__/channel-approval-routes.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2789,6 +2789,7 @@ describe('NL approval routing via destination-scoped canonical requests', () =>
sourceChannel: 'twilio',
conversationId: 'conv-voice-nl-1',
toolName: 'shell',
guardianPrincipalId: 'test-principal-id',
expiresAt: new Date(Date.now() + 60_000).toISOString(),
// guardianExternalUserId intentionally omitted
});
Expand Down Expand Up @@ -2842,6 +2843,7 @@ describe('NL approval routing via destination-scoped canonical requests', () =>
sourceType: 'voice',
sourceChannel: 'twilio',
toolName: 'shell',
guardianPrincipalId: 'test-principal-id',
expiresAt: new Date(Date.now() + 60_000).toISOString(),
});

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -118,6 +118,7 @@ function makeCanonicalRequest(overrides: Record<string, unknown> = {}) {
conversationId: 'conv-1',
requesterExternalUserId: 'requester-1',
guardianExternalUserId: 'guardian-1',
guardianPrincipalId: 'test-principal-id',
toolName: 'bash',
status: 'pending',
requestCode: generateCanonicalRequestCode(),
Expand Down
14 changes: 11 additions & 3 deletions assistant/src/__tests__/guardian-principal-id-roundtrip.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -116,13 +116,21 @@ describe('guardianPrincipalId roundtrip', () => {
expect(fetched!.decidedByPrincipalId).toBeNull();
});

test('creates request without guardianPrincipalId (defaults to null)', () => {
test('access_request requires guardianPrincipalId (decisionable kind)', () => {
// access_request is now decisionable — creating one without a principal
// should throw IntegrityError.
expect(() => createCanonicalGuardianRequest({
kind: 'access_request',
sourceType: 'desktop',
})).toThrow('guardianPrincipalId');

// With a principal, creation succeeds
const req = createCanonicalGuardianRequest({
kind: 'access_request',
sourceType: 'desktop',
guardianPrincipalId: 'access-req-principal',
});

expect(req.guardianPrincipalId).toBeNull();
expect(req.guardianPrincipalId).toBe('access-req-principal');
expect(req.decidedByPrincipalId).toBeNull();
});

Expand Down
Loading