@nostr-dev-kit/ndk/src/subscription/manager.test.ts

NDK - Nostr Development Kit

import { NDKRelay, NDKSubscription } from "../index.js";
import NDK, { type NDKEventId } from "../index.js";
import { NDKSubscriptionManager } from "./manager.js";

const ndk = new NDK();

describe("NDKSubscriptionManager", () => {
    let manager: NDKSubscriptionManager;

    beforeEach(() => {
        manager = new NDKSubscriptionManager();
    });

    it("should add a subscription", () => {
        const sub = new NDKSubscription(ndk, {});
        manager.add(sub);
        expect(manager.subscriptions.has(sub.internalId)).toBe(true);
    });

    // it("should remove a subscription on close", () => {
    //     const sub: NDKSubscription = {
    //         internalId: "sub2",
    //         on: jest.fn(),
    //         // Add other necessary properties
    //     };
    //     manager.add(sub);
    //     expect(manager.subscriptions.has("sub2")).toBe(true);

    //     // Simulate the 'close' event
    //     const closeCallback = sub.on.mock.calls.find(call => call[0] === "close")?.[1];
    //     if (closeCallback) {
    //         closeCallback();
    //     }

    //     expect(manager.subscriptions.has("sub2")).toBe(false);
    // });

    it("should record seen events", () => {
        const eventId: NDKEventId = "event1";
        const relay = new NDKRelay("wss://example.com", undefined, ndk);
        manager.seenEvent(eventId, relay);
        const seenRelays = manager.seenEvents.get(eventId);
        expect(seenRelays).toContain(relay);
    });
});