bitvid/tests/nostr/videoEventBuffer.test.mjs

import { test } from 'node:test';
import { strict as assert } from 'node:assert';
import { VideoEventBuffer } from "../../js/nostr/videoEventBuffer.js";

class MockClient {
  constructor() {
    this.rawEvents = new Map();
    this.allEvents = new Map();
    this.activeMap = new Map();
    this.dirtyEventIds = new Set();
    this.tombstones = new Map();
    this.saveLocalDataCalls = [];
    this.populateNip71MetadataCalls = [];
  }

  getActiveKey(video) {
    return video.videoRootId || video.id;
  }

  recordTombstone(key, timestamp) {
    this.tombstones.set(key, timestamp);
  }

  applyTombstoneGuard(video) {
     const key = this.getActiveKey(video);
     const ts = this.tombstones.get(key);
     if (ts && video.created_at <= ts) {
         video.deleted = true;
     }
  }

  mergeNip71MetadataIntoVideo(video) {}
  applyRootCreatedAt(video) {}

  async populateNip71MetadataForVideos(videos) {
    this.populateNip71MetadataCalls.push(videos);
    return Promise.resolve();
  }

  saveLocalData(reason) {
    this.saveLocalDataCalls.push(reason);
  }
}

const validEvent = {
  id: "id1",
  pubkey: "pubkey1",
  created_at: 1000,
  kind: 30078,
  tags: [["t", "video"]],
  content: JSON.stringify({
    title: "Test Video",
    url: "https://example.com/video.mp4",
    magnet: "magnet:?xt=urn:btih:1234567890abcdef1234567890abcdef12345678&dn=video",
    thumbnail: "https://example.com/thumb.jpg",
    videoRootId: "root1"
  })
};

const newerEvent = {
  ...validEvent,
  id: "id2",
  created_at: 2000
};

const deletedEvent = {
  ...validEvent,
  id: "id3",
  created_at: 3000,
  content: JSON.stringify({
    deleted: true,
    videoRootId: "root1",
    url: "https://example.com/video.mp4",
    title: "Deleted Video"
  })
};

test('VideoEventBuffer', async (t) => {
  await t.test('Push and Flush', () => {
    const client = new MockClient();
    let onVideoCalls = [];
    const onVideo = (videos) => onVideoCalls.push(videos);
    const buffer = new VideoEventBuffer(client, onVideo);

    buffer.push(validEvent);
    assert.equal(buffer.buffer.length, 1);
    assert.notEqual(buffer.flushTimerId, null);

    // Force flush
    buffer.scheduleFlush(true);
    assert.equal(buffer.buffer.length, 0);
    assert.equal(client.allEvents.size, 1);
    assert.equal(client.activeMap.size, 1);
    assert.equal(onVideoCalls.length, 1);

    // videoRootId is "root1", so key is "root1" (based on MockClient.getActiveKey)
    assert.ok(client.activeMap.has("root1"));
    assert.equal(client.activeMap.get("root1").id, validEvent.id);
  });

  await t.test('Latest Wins', () => {
    const client = new MockClient();
    let onVideoCalls = [];
    const onVideo = (videos) => onVideoCalls.push(videos);
    const buffer = new VideoEventBuffer(client, onVideo);

    // Setup: activeMap has older event
    client.activeMap.set("root1", { ...validEvent, created_at: 1000, videoRootId: "root1" });

    buffer.push(newerEvent);
    buffer.scheduleFlush(true); // Force flush

    // Should update because newerEvent.created_at (2000) > 1000
    assert.equal(client.activeMap.get("root1").id, newerEvent.id);
    assert.equal(client.activeMap.get("root1").created_at, 2000);
  });

  await t.test('Tombstone Handling', () => {
      const client = new MockClient();
      const buffer = new VideoEventBuffer(client, () => {});

      // Process deleted event
      buffer.push(deletedEvent);
      buffer.scheduleFlush(true);

      // Should verify tombstone recorded
      assert.ok(client.tombstones.has("root1"));
      assert.equal(client.tombstones.get("root1"), deletedEvent.created_at);

      // Should NOT be in activeMap
      assert.ok(!client.activeMap.has("root1"));

      // Now try pushing an older event
      buffer.push(newerEvent); // created_at 2000 < 3000
      buffer.scheduleFlush(true);

      // Should be guarded by tombstone
      assert.ok(!client.activeMap.has("root1"));
  });

  await t.test('Cleanup', () => {
    const client = new MockClient();
    const buffer = new VideoEventBuffer(client, () => {});

    buffer.push(validEvent);
    assert.notEqual(buffer.flushTimerId, null);

    buffer.cleanup();
    assert.equal(buffer.flushTimerId, null);
    assert.equal(buffer.buffer.length, 0); // Should have flushed
    assert.equal(client.allEvents.size, 1);
  });
});