network-ts-sdk/tests/e2e/pubsub.test.ts

import { describe, it, expect, beforeEach } from "vitest";
import { createTestClient, skipIfNoGateway, delay } from "./setup";

describe("PubSub", () => {
  if (skipIfNoGateway()) {
    console.log("Skipping PubSub tests");
  }

  let topicName: string;

  beforeEach(() => {
    topicName = `test_topic_${Date.now()}_${Math.random().toString(36).substring(7)}`;
  });

  it("should get topics list", async () => {
    const client = await createTestClient();
    const topics = await client.pubsub.topics();
    expect(Array.isArray(topics)).toBe(true);
  });

  it("should publish a message", async () => {
    const client = await createTestClient();
    const testMessage = "Hello from test";

    // Should not throw
    await client.pubsub.publish(topicName, testMessage);
    expect(true).toBe(true);
  });

  it("should subscribe and receive published message", async () => {
    const client = await createTestClient();
    const testMessage = "Test message";
    let receivedMessage: any = null;

    // Create subscription with handlers
    const subscription = await client.pubsub.subscribe(topicName, {
      onMessage: (msg) => {
        receivedMessage = msg;
      },
      onError: (err) => {
        console.error("Subscription error:", err);
      },
    });

    // Give subscription a moment to establish
    await delay(500);

    // Publish message
    await client.pubsub.publish(topicName, testMessage);

    // Wait for message to arrive
    await delay(1000);

    // Should have received the message
    expect(receivedMessage).toBeDefined();
    expect(receivedMessage?.topic).toBe(topicName);

    // Cleanup
    subscription.close();
  });

  it("should handle subscription events", async () => {
    const client = await createTestClient();
    const events: string[] = [];

    const subscription = await client.pubsub.subscribe(topicName, {
      onMessage: () => {
        events.push("message");
      },
      onError: (err) => {
        events.push("error");
      },
      onClose: () => {
        events.push("close");
      },
    });

    // Publish a message
    await delay(300);
    await client.pubsub.publish(topicName, "test");

    // Wait for event
    await delay(500);

    // Close and check for close event
    subscription.close();
    await delay(300);

    expect(events.length).toBeGreaterThanOrEqual(0);
  });

  it("should get presence information", async () => {
    const client = await createTestClient();
    const presence = await client.pubsub.getPresence(topicName);
    expect(presence.topic).toBe(topicName);
    expect(Array.isArray(presence.members)).toBe(true);
    expect(typeof presence.count).toBe("number");
  });

  it("should handle presence events in subscription", async () => {
    const client = await createTestClient();
    const joinedMembers: any[] = [];
    const leftMembers: any[] = [];
    const memberId = "test-user-" + Math.random().toString(36).substring(7);
    const meta = { name: "Test User" };

    const subscription = await client.pubsub.subscribe(topicName, {
      presence: {
        enabled: true,
        memberId,
        meta,
        onJoin: (member) => joinedMembers.push(member),
        onLeave: (member) => leftMembers.push(member),
      },
    });

    expect(subscription.hasPresence()).toBe(true);

    // Wait for join event
    await delay(1000);

    // Some gateways might send the self-join event
    // Check if we can get presence from subscription
    const members = await subscription.getPresence();
    expect(Array.isArray(members)).toBe(true);

    // Cleanup
    subscription.close();
    await delay(500);
  });
});