tt-help-cli-ycl 1.3.48 → 1.3.50

package/scripts/test-watch-db-smoke.mjs

@@ -0,0 +1,246 @@
+import assert from "node:assert/strict";
+import { mkdtempSync, rmSync } from "node:fs";
+import os from "node:os";
+import path from "node:path";
+
+import { createStore, closeStoreDb } from "../src/watch/data-store.js";
+import { startWatchServer } from "../src/watch/server.js";
+
+function createTempDbPath(prefix) {
+  const dir = mkdtempSync(path.join(os.tmpdir(), `${prefix}-`));
+  return {
+    dir,
+    dbPath: path.join(dir, "result.db"),
+  };
+}
+
+function cleanupTempDir(dir) {
+  rmSync(dir, { recursive: true, force: true });
+}
+
+function seedClaimStore(store) {
+  store.addUser({
+    uniqueId: "follow-tier2",
+    status: "pending",
+    followerCount: 100,
+    videoCount: 1,
+    guessedLocation: "DE",
+    sources: ["following"],
+  });
+  store.addUser({
+    uniqueId: "seller-tier2",
+    status: "pending",
+    followerCount: 50,
+    videoCount: 1,
+    guessedLocation: "DE",
+    ttSeller: true,
+    verified: false,
+    sources: ["comment"],
+  });
+  store.addUser({
+    uniqueId: "seed-tier1",
+    status: "pending",
+    followerCount: 10,
+    videoCount: 1,
+    guessedLocation: "PL",
+    sources: ["seed"],
+  });
+}
+
+function seedWatchStore(store) {
+  store.addUser({
+    uniqueId: "seed-tier1",
+    nickname: "Seed User",
+    status: "pending",
+    followerCount: 10,
+    videoCount: 1,
+    guessedLocation: "PL",
+    ttSeller: false,
+    verified: false,
+    sources: ["seed"],
+  });
+  store.addUser({
+    uniqueId: "seller-target",
+    nickname: "Seller Target",
+    status: "pending",
+    followerCount: 500,
+    videoCount: 1,
+    guessedLocation: "DE",
+    locationCreated: "DE",
+    ttSeller: true,
+    verified: false,
+    sources: ["comment"],
+  });
+  store.addUser({
+    uniqueId: "follow-tier2",
+    nickname: "Follow User",
+    status: "pending",
+    followerCount: 300,
+    videoCount: 1,
+    guessedLocation: "FR",
+    ttSeller: false,
+    verified: false,
+    sources: ["following"],
+  });
+  store.addUser({
+    uniqueId: "pending-update",
+    nickname: "Needs Update",
+    status: "pending",
+    guessedLocation: "US",
+    ttSeller: null,
+    verified: false,
+    userUpdateCount: 0,
+    sources: ["comment"],
+  });
+  store.addUser({
+    uniqueId: "done-es",
+    nickname: "Done ES",
+    status: "done",
+    followerCount: 800,
+    videoCount: 3,
+    guessedLocation: "ES",
+    locationCreated: "ES",
+    ttSeller: true,
+    verified: false,
+    processedAt: Date.now(),
+    processed: true,
+    sources: ["processed"],
+  });
+  store.addUser({
+    uniqueId: "restricted-it",
+    nickname: "Restricted IT",
+    status: "restricted",
+    locationCreated: "IT",
+    ttSeller: false,
+    verified: false,
+    sources: ["comment"],
+  });
+  store.addUser({
+    uniqueId: "error-user",
+    nickname: "Error User",
+    status: "error",
+    locationCreated: "US",
+    ttSeller: false,
+    verified: false,
+    sources: ["comment"],
+  });
+}
+
+async function testClaimPriorityAndRenewal() {
+  const { dir, dbPath } = createTempDbPath("tt-watch-claim");
+  try {
+    const store = createStore(dbPath);
+    seedClaimStore(store);
+
+    const first = store.claimNextJob("worker-a", 5 * 60 * 1000, null, true);
+    assert.ok(first, "expected a claim for logged-in worker");
+    assert.equal(first.uniqueId, "seed-tier1");
+
+    const renewed = store.claimNextJob("worker-a", 5 * 60 * 1000, null, true);
+    assert.ok(renewed, "expected renewal claim for same worker");
+    assert.equal(renewed.uniqueId, "seed-tier1");
+    assert.ok(
+      renewed.claimedAt >= first.claimedAt,
+      "expected renewal claimedAt to advance or stay equal",
+    );
+
+    const loggedOut = store.claimNextJob(
+      "worker-b",
+      5 * 60 * 1000,
+      null,
+      false,
+    );
+    assert.ok(loggedOut, "expected a claim for logged-out worker");
+    assert.equal(loggedOut.uniqueId, "follow-tier2");
+  } finally {
+    closeStoreDb();
+    cleanupTempDir(dir);
+  }
+}
+
+async function testWatchHttpEndpoints() {
+  const { dir, dbPath } = createTempDbPath("tt-watch-http");
+  let server;
+  try {
+    const store = createStore(dbPath);
+    seedWatchStore(store);
+
+    const started = await startWatchServer(dbPath, 0, store);
+    server = started.server;
+    const actualPort = server.address().port;
+    const baseUrl = `http://127.0.0.1:${actualPort}`;
+
+    const [statsRes, usersRes, targetRes, lightUsersRes] = await Promise.all([
+      fetch(`${baseUrl}/api/stats`),
+      fetch(`${baseUrl}/api/users?limit=3`),
+      fetch(`${baseUrl}/api/target-users`),
+      fetch(`${baseUrl}/api/users?limit=2&view=light`),
+    ]);
+
+    assert.equal(statsRes.status, 200);
+    assert.equal(usersRes.status, 200);
+    assert.equal(targetRes.status, 200);
+    assert.equal(lightUsersRes.status, 200);
+
+    const [stats, users, targets, lightUsers] = await Promise.all([
+      statsRes.json(),
+      usersRes.json(),
+      targetRes.json(),
+      lightUsersRes.json(),
+    ]);
+
+    assert.equal(stats.totalUsers, 7);
+    assert.equal(stats.pendingUsers, 4);
+    assert.equal(stats.processedUsers, 1);
+    assert.equal(stats.restrictedUsers, 1);
+    assert.equal(stats.errorUsers, 1);
+    assert.equal(stats.targetUsers, 2);
+    assert.equal(stats.userUpdateTasks, 1);
+    assert.ok(Array.isArray(stats.countryStats));
+    assert.ok(stats.countryStats.some((item) => item.country === "ES"));
+
+    assert.equal(users.total, 7);
+    assert.equal(users.users.length, 3);
+    assert.equal(users.users[0].uniqueId, "seller-target");
+
+    assert.equal(targets.total, 2);
+    assert.deepEqual(
+      targets.users.map((item) => item.uniqueId),
+      ["done-es", "seller-target"],
+    );
+
+    assert.equal(lightUsers.total, 7);
+    assert.equal(lightUsers.users.length, 2);
+    assert.deepEqual(Object.keys(lightUsers.users[0]).sort(), [
+      "followerCount",
+      "guessedLocation",
+      "locationCreated",
+      "nickname",
+      "pinned",
+      "processedAt",
+      "sources",
+      "status",
+      "ttSeller",
+      "uniqueId",
+      "verified",
+    ]);
+  } finally {
+    if (server) {
+      await new Promise((resolve) => server.close(resolve));
+    }
+    closeStoreDb();
+    cleanupTempDir(dir);
+  }
+}
+
+async function main() {
+  await testClaimPriorityAndRenewal();
+  await testWatchHttpEndpoints();
+  console.log("watch db smoke test passed");
+}
+
+main().catch((error) => {
+  console.error("watch db smoke test failed");
+  console.error(error);
+  process.exitCode = 1;
+});