From 48d9de8dd10f1f9dc3fb693fbc00edd6ed66ccb1 Mon Sep 17 00:00:00 2001 From: esdrasrenan Date: Tue, 9 Dec 2025 23:30:25 -0300 Subject: [PATCH] test(machines): atualizar mock para suportar tabela machineHeartbeats MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Adicionar mock da nova tabela machineHeartbeats com metodo first() para corrigir teste getById apos refatoracao do heartbeat. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 --- tests/machines.getById.test.ts | 38 +++++++++++++++++++++++----------- 1 file changed, 26 insertions(+), 12 deletions(-) diff --git a/tests/machines.getById.test.ts b/tests/machines.getById.test.ts index 66e0140..6d44f69 100644 --- a/tests/machines.getById.test.ts +++ b/tests/machines.getById.test.ts @@ -55,18 +55,32 @@ describe("convex.machines.getById", () => { if (id === machine._id) return machine return null }), - query: vi.fn((_table: string) => ({ - withIndex: vi.fn((_name: string, _cb: unknown) => ({ - collect: vi.fn(async () => [ - { revoked: false, expiresAt: FIXED_NOW + 60_000, lastUsedAt: FIXED_NOW - 1000, usageCount: 5 }, - ]), - take: vi.fn(async () => [ - { revoked: false, expiresAt: FIXED_NOW + 60_000, lastUsedAt: FIXED_NOW - 1000, usageCount: 5 }, - ]), - })), - collect: vi.fn(async () => []), - take: vi.fn(async () => []), - })), + query: vi.fn((table: string) => { + // Mock para machineHeartbeats - retorna o lastHeartbeatAt da nova tabela + if (table === "machineHeartbeats") { + return { + withIndex: vi.fn((_name: string, _cb: unknown) => ({ + first: vi.fn(async () => ({ machineId: machine._id, lastHeartbeatAt: FIXED_NOW - 1000 })), + collect: vi.fn(async () => [{ machineId: machine._id, lastHeartbeatAt: FIXED_NOW - 1000 }]), + take: vi.fn(async () => [{ machineId: machine._id, lastHeartbeatAt: FIXED_NOW - 1000 }]), + })), + } + } + // Mock para machineTokens e outras tabelas + return { + withIndex: vi.fn((_name: string, _cb: unknown) => ({ + first: vi.fn(async () => null), + collect: vi.fn(async () => [ + { revoked: false, expiresAt: FIXED_NOW + 60_000, lastUsedAt: FIXED_NOW - 1000, usageCount: 5 }, + ]), + take: vi.fn(async () => [ + { revoked: false, expiresAt: FIXED_NOW + 60_000, lastUsedAt: FIXED_NOW - 1000, usageCount: 5 }, + ]), + })), + collect: vi.fn(async () => []), + take: vi.fn(async () => []), + } + }), } const ctx = { db } as unknown as Parameters[0]