feat(trading): add trade history and daily FIFO retention cron

- trading_trades table (migration 0001) persists every buy/sell via optional onTrade callback
- /history [n] command shows caller's last N trades (default 10, max 50), HTML-escaped
- daily cron at 0 17 * * * trims to 1000/user + 10000/global via FIFO delete
- persistence failure logs but does not fail the trade reply
This commit is contained in:
2026-04-15 13:29:15 +07:00
parent 8235c9602e
commit d040ce4161
9 changed files with 821 additions and 10 deletions

View File

@@ -0,0 +1,195 @@
/**
* @file Tests for trading/retention — trimTradesHandler per-user and global caps.
*
* Uses enhanced fake-d1 which supports:
* - SELECT DISTINCT user_id
* - SELECT id ... WHERE user_id = ? ORDER BY ts DESC LIMIT -1 OFFSET N
* - SELECT id ... ORDER BY ts DESC LIMIT -1 OFFSET N
* - DELETE FROM ... WHERE id IN (?, ...)
*/
import { beforeEach, describe, expect, it, vi } from "vitest";
import { createSqlStore } from "../../../src/db/create-sql-store.js";
import { trimTradesHandler } from "../../../src/modules/trading/retention.js";
import { makeFakeD1 } from "../../fakes/fake-d1.js";
// ─── helpers ─────────────────────────────────────────────────────────────────
/** Build a fresh sql store backed by a new fake D1. */
function makeTestSql() {
const fakeDb = makeFakeD1();
const sql = createSqlStore("trading", { DB: fakeDb });
return { fakeDb, sql };
}
/**
* Build N trade rows for a given user_id.
* ts increases from 1..N so that the LAST row (index N-1) has the highest ts
* and is therefore the "newest".
*
* @param {number} userId
* @param {number} count
* @param {number} [idOffset=0] — added to each id to keep ids globally unique.
* @returns {object[]}
*/
function makeTrades(userId, count, idOffset = 0) {
return Array.from({ length: count }, (_, i) => ({
id: idOffset + i + 1,
user_id: userId,
symbol: "TCB",
side: "buy",
qty: 1,
price_vnd: 1000,
ts: i + 1, // ts=1 is oldest, ts=count is newest
}));
}
/** Convenience: run trimTradesHandler with muted console output. */
async function runTrim(sql, caps) {
const spy = vi.spyOn(console, "log").mockImplementation(() => {});
try {
await trimTradesHandler({}, { sql }, caps);
} finally {
spy.mockRestore();
}
}
/** Return all rows remaining in trading_trades via the fake table map. */
function allRows(fakeDb) {
return fakeDb.tables.get("trading_trades") ?? [];
}
// ─── sql=null guard ───────────────────────────────────────────────────────────
describe("trimTradesHandler — sql=null", () => {
it("returns without throwing when sql is null", async () => {
const logSpy = vi.spyOn(console, "log").mockImplementation(() => {});
await expect(trimTradesHandler({}, { sql: null })).resolves.toBeUndefined();
logSpy.mockRestore();
});
});
// ─── per-user trim ────────────────────────────────────────────────────────────
describe("trimTradesHandler — per-user trim", () => {
it("removes rows beyond cap, keeping newest", async () => {
const { fakeDb, sql } = makeTestSql();
// 5 trades for user 1, cap=3 → keeps ts=3,4,5 (newest 3), deletes ts=1,2.
fakeDb.seed("trading_trades", makeTrades(1, 5));
await runTrim(sql, { perUserCap: 3, globalCap: 1000 });
const remaining = allRows(fakeDb);
expect(remaining).toHaveLength(3);
// Remaining should be the 3 rows with highest ts.
const tsSorted = remaining.map((r) => r.ts).sort((a, b) => b - a);
expect(tsSorted).toEqual([5, 4, 3]);
});
it("leaves users with rows <= cap untouched", async () => {
const { fakeDb, sql } = makeTestSql();
// User 2 has 2 trades, cap=3 → none deleted.
fakeDb.seed("trading_trades", makeTrades(2, 2));
await runTrim(sql, { perUserCap: 3, globalCap: 1000 });
expect(allRows(fakeDb)).toHaveLength(2);
});
it("handles exactly cap rows — no deletion", async () => {
const { fakeDb, sql } = makeTestSql();
fakeDb.seed("trading_trades", makeTrades(1, 3));
await runTrim(sql, { perUserCap: 3, globalCap: 1000 });
expect(allRows(fakeDb)).toHaveLength(3);
});
it("trims multiple users independently", async () => {
const { fakeDb, sql } = makeTestSql();
// User 1: 5 trades (cap 3 → keep 3). User 2: 2 trades (cap 3 → keep 2).
fakeDb.seed("trading_trades", [...makeTrades(1, 5, 0), ...makeTrades(2, 2, 5)]);
await runTrim(sql, { perUserCap: 3, globalCap: 1000 });
const remaining = allRows(fakeDb);
// 3 from user1 + 2 from user2 = 5.
expect(remaining).toHaveLength(5);
const user1 = remaining.filter((r) => r.user_id === 1);
const user2 = remaining.filter((r) => r.user_id === 2);
expect(user1).toHaveLength(3);
expect(user2).toHaveLength(2);
});
});
// ─── global FIFO trim ─────────────────────────────────────────────────────────
describe("trimTradesHandler — global trim", () => {
it("keeps only globalCap newest rows across all users", async () => {
const { fakeDb, sql } = makeTestSql();
// 15 trades total: user1=8, user2=7. globalCap=10 → keep 10 newest.
const trades = [...makeTrades(1, 8, 0), ...makeTrades(2, 7, 8)];
// Re-assign ts values to be globally unique so ordering is deterministic.
trades.forEach((t, i) => {
t.ts = i + 1;
});
fakeDb.seed("trading_trades", trades);
await runTrim(sql, { perUserCap: 1000, globalCap: 10 });
expect(allRows(fakeDb)).toHaveLength(10);
// All remaining rows should have the top-10 ts values (6..15).
const tsSorted = allRows(fakeDb)
.map((r) => r.ts)
.sort((a, b) => a - b);
expect(tsSorted).toEqual([6, 7, 8, 9, 10, 11, 12, 13, 14, 15]);
});
it("leaves rows untouched when count <= globalCap", async () => {
const { fakeDb, sql } = makeTestSql();
fakeDb.seed("trading_trades", makeTrades(1, 5));
await runTrim(sql, { perUserCap: 1000, globalCap: 10 });
expect(allRows(fakeDb)).toHaveLength(5);
});
});
// ─── combined pass ────────────────────────────────────────────────────────────
describe("trimTradesHandler — combined per-user + global", () => {
it("applies per-user first, then global", async () => {
const { fakeDb, sql } = makeTestSql();
// User 1: 6 trades (perUserCap=4 → trim to 4).
// User 2: 6 trades (perUserCap=4 → trim to 4).
// After per-user: 8 rows total.
// globalCap=6 → trim to 6.
const trades = [...makeTrades(1, 6, 0), ...makeTrades(2, 6, 6)];
// Assign globally monotone ts values.
trades.forEach((t, i) => {
t.ts = i + 1;
});
fakeDb.seed("trading_trades", trades);
await runTrim(sql, { perUserCap: 4, globalCap: 6 });
expect(allRows(fakeDb)).toHaveLength(6);
});
});
// ─── idempotence ──────────────────────────────────────────────────────────────
describe("trimTradesHandler — idempotence", () => {
it("second run is a no-op when already within caps", async () => {
const { fakeDb, sql } = makeTestSql();
fakeDb.seed("trading_trades", makeTrades(1, 5));
await runTrim(sql, { perUserCap: 3, globalCap: 10 });
const afterFirst = allRows(fakeDb).length;
await runTrim(sql, { perUserCap: 3, globalCap: 10 });
const afterSecond = allRows(fakeDb).length;
expect(afterFirst).toBe(afterSecond);
});
});