feat(federation): rollbackFriendRequestCreate + side-effect registration

2026-04-25 21:51:04 +02:00
parent 9e3417485c
commit 08872f1e08
3 changed files with 164 additions and 0 deletions
@@ -0,0 +1,107 @@
+import { describe, it, expect, beforeEach, vi } from 'vitest';
+import Database from 'better-sqlite3';
+import { drizzle } from 'drizzle-orm/better-sqlite3';
+import { eq } from 'drizzle-orm';
+import fs from 'node:fs';
+import path from 'node:path';
+import { fileURLToPath } from 'node:url';
+import * as schema from '../db/schema.js';
+import { setWorkerId } from '../utils/snowflake.js';
+
+setWorkerId(1);
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+
+let sqlite: Database.Database;
+let testDb: ReturnType<typeof drizzle<typeof schema>>;
+const sendToUser = vi.fn();
+
+vi.mock('../db/index.js', () => ({
+  getDb: () => testDb,
+  getRawDb: () => sqlite,
+  schema,
+}));
+
+vi.mock('../ws/handler.js', () => ({
+  connectionManager: { sendToUser },
+}));
+
+function applyMigrations(db: Database.Database): void {
+  const dir = path.resolve(__dirname, '../../drizzle');
+  for (const f of fs.readdirSync(dir).filter(f => f.endsWith('.sql')).sort()) {
+    const sqlText = fs.readFileSync(path.join(dir, f), 'utf8');
+    for (const stmt of sqlText.split(/-->\s*statement-breakpoint/)) {
+      const clean = stmt.trim();
+      if (clean) db.exec(clean);
+    }
+  }
+}
+
+beforeEach(() => {
+  sqlite = new Database(':memory:');
+  testDb = drizzle(sqlite, { schema });
+  applyMigrations(sqlite);
+  sendToUser.mockReset();
+});
+
+describe('rollbackFriendRequestCreate', () => {
+  it('deletes the matching friend_requests row and emits friend_request_relay_failed to sender', async () => {
+    testDb.insert(schema.users).values([
+      { id: 'sender', username: 'bob', passwordHash: 'x', status: 'offline', isAdmin: 0, createdAt: Date.now() },
+      { id: 'stub', username: 'alice@orbit.test', passwordHash: '!federation-replicated',
+        status: 'offline', isAdmin: 0, homeInstance: 'orbit.test', homeUserId: 'remote-1', createdAt: Date.now() },
+    ] as typeof schema.users.$inferInsert[]).run();
+
+    testDb.insert(schema.friendRequests).values({
+      id: 'req-1',
+      fromId: 'sender',
+      toId: 'stub',
+      status: 'pending',
+      createdAt: Date.now(),
+      relayMessageId: 'friend_req:remote-1:sender:1234',
+    } as typeof schema.friendRequests.$inferInsert).run();
+
+    const { rollbackFriendRequestCreate } = await import('./federationRollback.js');
+    rollbackFriendRequestCreate('friend_req:remote-1:sender:1234', 'recipient_not_found');
+
+    const remaining = testDb.select().from(schema.friendRequests).where(eq(schema.friendRequests.id, 'req-1')).get();
+    expect(remaining).toBeUndefined();
+
+    expect(sendToUser).toHaveBeenCalledOnce();
+    const [userId, event] = sendToUser.mock.calls[0]!;
+    expect(userId).toBe('sender');
+    expect(event.type).toBe('friend_request_relay_failed');
+    expect(event.requestId).toBe('req-1');
+    expect(event.reason).toBe('user_not_found');
+    expect(event.targetHandle).toBe('alice@orbit.test');
+    expect(typeof event.message).toBe('string');
+  });
+
+  it('is idempotent: no-op if no row matches the messageId', async () => {
+    const { rollbackFriendRequestCreate } = await import('./federationRollback.js');
+    expect(() => rollbackFriendRequestCreate('no-such-msg', 'recipient_not_found')).not.toThrow();
+    expect(sendToUser).not.toHaveBeenCalled();
+  });
+
+  it('maps unknown reasons to peer_rejected', async () => {
+    testDb.insert(schema.users).values([
+      { id: 'sender', username: 'bob', passwordHash: 'x', status: 'offline', isAdmin: 0, createdAt: Date.now() },
+      { id: 'stub', username: 'alice@orbit.test', passwordHash: '!federation-replicated',
+        status: 'offline', isAdmin: 0, homeInstance: 'orbit.test', homeUserId: 'remote-1', createdAt: Date.now() },
+    ] as typeof schema.users.$inferInsert[]).run();
+    testDb.insert(schema.friendRequests).values({
+      id: 'req-2',
+      fromId: 'sender',
+      toId: 'stub',
+      status: 'pending',
+      createdAt: Date.now(),
+      relayMessageId: 'msg-x',
+    } as typeof schema.friendRequests.$inferInsert).run();
+
+    const { rollbackFriendRequestCreate } = await import('./federationRollback.js');
+    rollbackFriendRequestCreate('msg-x', 'attribution_mismatch');
+
+    expect(sendToUser).toHaveBeenCalledOnce();
+    const event = sendToUser.mock.calls[0]![1];
+    expect(event.reason).toBe('peer_rejected');
+  });
+});