fix(keys): add DB fallback to updateEmailByCustomer, updateKeyEmail, and recover route (BUG-108, BUG-109, BUG-110)
All checks were successful
Build & Deploy to Staging / Build & Deploy to Staging (push) Successful in 13m8s

- updateEmailByCustomer: DB fallback when stripe_customer_id not in cache
- updateKeyEmail: DB fallback when key not in cache
- POST /v1/recover: DB fallback when email not in cache (was only on verify)
- 6 TDD tests added (keys-email-update.test.ts, recover-initial-db-fallback.test.ts)
- 547 tests total, all passing
This commit is contained in:
DocFast CEO 2026-03-07 20:06:13 +01:00
parent 424a16ed8a
commit d376d586fe
4 changed files with 286 additions and 4 deletions

View file

@ -0,0 +1,105 @@
import { describe, it, expect, vi, beforeEach } from "vitest";
import request from "supertest";
import express from "express";
// Mock dependencies
vi.mock("../services/keys.js", () => ({
getAllKeys: vi.fn().mockReturnValue([]),
isValidKey: vi.fn(),
getKeyInfo: vi.fn(),
isProKey: vi.fn(),
createFreeKey: vi.fn(),
createProKey: vi.fn(),
downgradeByCustomer: vi.fn(),
findKeyByCustomerId: vi.fn(),
loadKeys: vi.fn(),
updateKeyEmail: vi.fn(),
updateEmailByCustomer: vi.fn(),
}));
vi.mock("../services/db.js", () => ({
default: { query: vi.fn(), connect: vi.fn(), on: vi.fn(), end: vi.fn() },
pool: { query: vi.fn(), connect: vi.fn(), on: vi.fn(), end: vi.fn() },
queryWithRetry: vi.fn().mockResolvedValue({ rows: [], rowCount: 0 }),
connectWithRetry: vi.fn().mockResolvedValue(undefined),
initDatabase: vi.fn().mockResolvedValue(undefined),
cleanupStaleData: vi.fn(),
isTransientError: vi.fn(),
}));
vi.mock("../services/verification.js", () => ({
createPendingVerification: vi.fn().mockResolvedValue({ code: "123456" }),
verifyCode: vi.fn(),
}));
vi.mock("../services/email.js", () => ({
sendVerificationEmail: vi.fn().mockResolvedValue(undefined),
}));
vi.mock("../services/logger.js", () => ({
default: { info: vi.fn(), error: vi.fn(), warn: vi.fn(), debug: vi.fn() },
}));
import { getAllKeys } from "../services/keys.js";
import { queryWithRetry } from "../services/db.js";
import { createPendingVerification } from "../services/verification.js";
import { sendVerificationEmail } from "../services/email.js";
import { recoverRouter } from "../routes/recover.js";
const mockGetAllKeys = vi.mocked(getAllKeys);
const mockQuery = vi.mocked(queryWithRetry);
const mockCreatePending = vi.mocked(createPendingVerification);
const mockSendEmail = vi.mocked(sendVerificationEmail);
function createApp() {
const app = express();
app.use(express.json());
app.use("/v1/recover", recoverRouter);
return app;
}
describe("POST /v1/recover DB fallback (BUG-110)", () => {
beforeEach(() => {
vi.clearAllMocks();
mockGetAllKeys.mockReturnValue([]);
mockCreatePending.mockResolvedValue({ code: "123456" } as any);
mockSendEmail.mockResolvedValue(undefined);
});
it("sends verification email via DB fallback when key not in cache but exists in DB", async () => {
mockGetAllKeys.mockReturnValue([]); // empty cache
mockQuery.mockResolvedValueOnce({
rows: [{ key: "df_pro_abc123" }],
rowCount: 1,
} as any);
const app = createApp();
const res = await request(app)
.post("/v1/recover")
.send({ email: "user@example.com" });
expect(res.status).toBe(200);
expect(res.body.status).toBe("recovery_sent");
expect(mockQuery).toHaveBeenCalledWith(
expect.stringContaining("SELECT"),
expect.arrayContaining(["user@example.com"])
);
expect(mockCreatePending).toHaveBeenCalledWith("user@example.com");
expect(mockSendEmail).toHaveBeenCalled();
});
it("does NOT send verification email when key not in cache AND not in DB", async () => {
mockGetAllKeys.mockReturnValue([]);
mockQuery.mockResolvedValueOnce({ rows: [], rowCount: 0 } as any);
const app = createApp();
const res = await request(app)
.post("/v1/recover")
.send({ email: "nobody@example.com" });
expect(res.status).toBe(200);
expect(res.body.status).toBe("recovery_sent");
expect(mockCreatePending).not.toHaveBeenCalled();
expect(mockSendEmail).not.toHaveBeenCalled();
});
});