45 lines
1.4 KiB
TypeScript
45 lines
1.4 KiB
TypeScript
import { mkdtemp, rm } from "node:fs/promises";
|
|
import { tmpdir } from "node:os";
|
|
import { join } from "node:path";
|
|
import { afterEach, describe, expect, it } from "vitest";
|
|
import { WatchlistStore } from "../src/agent/watchlistStore";
|
|
|
|
let tempDir: string | undefined;
|
|
|
|
afterEach(async () => {
|
|
if (tempDir) {
|
|
await rm(tempDir, { force: true, recursive: true });
|
|
tempDir = undefined;
|
|
}
|
|
});
|
|
|
|
describe("WatchlistStore", () => {
|
|
it("adds normalized stocks once and persists them", async () => {
|
|
tempDir = await mkdtemp(join(tmpdir(), "stock-agent-"));
|
|
const path = join(tempDir, "watchlist.json");
|
|
const store = new WatchlistStore(path);
|
|
|
|
await store.add("aapl", "Apple");
|
|
await store.add("AAPL", "Duplicate");
|
|
await store.add("00700.HK", "Tencent");
|
|
|
|
expect(await store.list()).toEqual([
|
|
expect.objectContaining({ display: "AAPL", market: "US", name: "Apple" }),
|
|
expect.objectContaining({ display: "00700.HK", market: "HK", name: "Tencent" })
|
|
]);
|
|
|
|
const reloaded = new WatchlistStore(path);
|
|
expect(await reloaded.list()).toHaveLength(2);
|
|
});
|
|
|
|
it("removes stocks by any supported symbol form", async () => {
|
|
tempDir = await mkdtemp(join(tmpdir(), "stock-agent-"));
|
|
const store = new WatchlistStore(join(tempDir, "watchlist.json"));
|
|
|
|
await store.add("600519", "Kweichow Moutai");
|
|
await store.remove("600519.SS");
|
|
|
|
expect(await store.list()).toEqual([]);
|
|
});
|
|
});
|