OpenClaw
b964b98a8b
All checks were successful
Build & Deploy to Staging / Build & Deploy to Staging (push) Successful in 13m7s
- downgradeByCustomer now queries DB when key not in memory cache, preventing cancelled customers from keeping Pro access in multi-pod setups - recover/verify endpoint falls back to DB lookup when cache miss on email - Added TDD tests for both fallback paths (4 new tests)
75 lines
2.4 KiB
TypeScript
75 lines
2.4 KiB
TypeScript
import { describe, it, expect, vi, beforeEach } from "vitest";
|
|
|
|
// Override the global setup.ts mock for keys — we need the REAL implementation
|
|
vi.unmock("../services/keys.js");
|
|
|
|
// Keep db mocked (setup.ts already does this, but be explicit about our mock)
|
|
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/logger.js", () => ({
|
|
default: { info: vi.fn(), error: vi.fn(), warn: vi.fn(), debug: vi.fn() },
|
|
}));
|
|
|
|
import { queryWithRetry } from "../services/db.js";
|
|
import { downgradeByCustomer } from "../services/keys.js";
|
|
|
|
const mockQuery = vi.mocked(queryWithRetry);
|
|
|
|
describe("downgradeByCustomer DB fallback (BUG-106)", () => {
|
|
beforeEach(() => {
|
|
vi.clearAllMocks();
|
|
});
|
|
|
|
it("returns true and updates DB when key is NOT in cache but IS in DB", async () => {
|
|
mockQuery
|
|
.mockResolvedValueOnce({
|
|
rows: [
|
|
{
|
|
key: "df_pro_abc123",
|
|
tier: "pro",
|
|
email: "user@example.com",
|
|
created_at: "2026-01-01T00:00:00.000Z",
|
|
stripe_customer_id: "cus_123",
|
|
},
|
|
],
|
|
rowCount: 1,
|
|
} as any)
|
|
.mockResolvedValueOnce({ rows: [], rowCount: 1 } as any); // UPDATE
|
|
|
|
const result = await downgradeByCustomer("cus_123");
|
|
|
|
expect(result).toBe(true);
|
|
expect(mockQuery).toHaveBeenCalledWith(
|
|
expect.stringContaining("SELECT"),
|
|
expect.arrayContaining(["cus_123"])
|
|
);
|
|
expect(mockQuery).toHaveBeenCalledWith(
|
|
expect.stringContaining("UPDATE"),
|
|
expect.arrayContaining(["cus_123"])
|
|
);
|
|
});
|
|
|
|
it("returns false when key is NOT in cache AND NOT in DB", async () => {
|
|
mockQuery.mockResolvedValueOnce({ rows: [], rowCount: 0 } as any);
|
|
|
|
const result = await downgradeByCustomer("cus_nonexistent");
|
|
|
|
expect(result).toBe(false);
|
|
expect(mockQuery).toHaveBeenCalledWith(
|
|
expect.stringContaining("SELECT"),
|
|
expect.arrayContaining(["cus_nonexistent"])
|
|
);
|
|
const updateCalls = mockQuery.mock.calls.filter((c) =>
|
|
(c[0] as string).includes("UPDATE")
|
|
);
|
|
expect(updateCalls).toHaveLength(0);
|
|
});
|
|
});
|