welshman/packages/net/__tests__/request.test.ts

import { describe, expect, it, vi, beforeEach, afterEach } from "vitest"
import { Nip01Signer } from '@welshman/signer'
import { LOCAL_RELAY_URL, makeEvent } from '@welshman/util'
import { ClientMessageType, RelayMessage } from "../src/message"
import { AdapterContext, AbstractAdapter, AdapterEvent } from "../src/adapter"
import { unireq, multireq, RequestEvent } from "../src/request"
import { Tracker } from "../src/tracker"

class MockAdapter extends AbstractAdapter {
  constructor(readonly url: string, readonly send) {
    super()
  }

  get sockets() {
    return []
  }

  get urls() {
    return [this.url]
  }

  receive = (message: RelayMessage) => {
    this.emit(AdapterEvent.Receive, message, this.url)
  }
}

describe("Unireq", () => {
  beforeEach(() => {
    vi.useFakeTimers()
  })

  afterEach(() => {
    vi.useRealTimers()
  })

  it("everything basically works", async () => {
    const sendSpy = vi.fn()
    const adapter = new MockAdapter('1', sendSpy)
    const req = unireq({
      relay: 'whatever',
      filter: {kinds: [1]},
      context: {getAdapter: () => adapter},
      autoClose: true,
    })

    const duplicateSpy = vi.fn()
    const invalidSpy = vi.fn()
    const filteredSpy = vi.fn()
    const eventSpy = vi.fn()
    const eoseSpy = vi.fn()
    const closeSpy = vi.fn()

    req.on(RequestEvent.Duplicate, duplicateSpy)
    req.on(RequestEvent.Invalid, invalidSpy)
    req.on(RequestEvent.Filtered, filteredSpy)
    req.on(RequestEvent.Event, eventSpy)
    req.on(RequestEvent.Eose, eoseSpy)
    req.on(RequestEvent.Close, closeSpy)

    await vi.runAllTimers()

    expect(sendSpy).toHaveBeenCalledWith([ClientMessageType.Req, req._id, {kinds: [1]}])

    const signer = Nip01Signer.ephemeral()
    const event1 = await signer.sign(makeEvent(1))
    const event2 = await signer.sign(makeEvent(7))
    const event3 = makeEvent(1)

    adapter.receive(["EVENT", req._id, event1])
    adapter.receive(["EVENT", req._id, event2])
    adapter.receive(["EVENT", req._id, event1])
    adapter.receive(["EVENT", req._id, event3])

    await vi.runAllTimers()

    expect(duplicateSpy).toHaveBeenCalledWith(event1)
    expect(filteredSpy).toHaveBeenCalledWith(event2)
    expect(invalidSpy).toHaveBeenCalledWith(event3)
    expect(eventSpy).toHaveBeenCalledWith(event1)
    expect(eoseSpy).toHaveBeenCalledTimes(0)

    adapter.receive(["EOSE", req._id])

    expect(eoseSpy).toHaveBeenCalledTimes(1)
    expect(closeSpy).toHaveBeenCalledTimes(1)
  })
})

describe("Multireq", () => {
  beforeEach(() => {
    vi.useFakeTimers()
  })

  afterEach(() => {
    vi.useRealTimers()
  })

  it("everything basically works", async () => {
    const send1Spy = vi.fn()
    const adapter1 = new MockAdapter('1', send1Spy)
    const send2Spy = vi.fn()
    const adapter2 = new MockAdapter('2', send2Spy)
    const req = multireq({
      autoClose: true,
      relays: ['1', '2'],
      filter: {kinds: [1]},
      context: {
        getAdapter: (url: string) => url === '1' ? adapter1 : adapter2
      },
    })

    const duplicateSpy = vi.fn()
    const invalidSpy = vi.fn()
    const filteredSpy = vi.fn()
    const eventSpy = vi.fn()
    const eoseSpy = vi.fn()
    const closeSpy = vi.fn()

    req.on(RequestEvent.Duplicate, duplicateSpy)
    req.on(RequestEvent.Invalid, invalidSpy)
    req.on(RequestEvent.Filtered, filteredSpy)
    req.on(RequestEvent.Event, eventSpy)
    req.on(RequestEvent.Eose, eoseSpy)
    req.on(RequestEvent.Close, closeSpy)

    await vi.runAllTimers()

    expect(send1Spy).toHaveBeenCalledWith([ClientMessageType.Req, req._children[0]._id, {kinds: [1]}])
    expect(send2Spy).toHaveBeenCalledWith([ClientMessageType.Req, req._children[1]._id, {kinds: [1]}])

    const signer = Nip01Signer.ephemeral()
    const event1 = await signer.sign(makeEvent(1))
    const event2 = await signer.sign(makeEvent(7))
    const event3 = makeEvent(1)
    const event4 = await signer.sign(makeEvent(1))

    adapter1.receive(["EVENT", req._children[0]._id, event1])
    adapter1.receive(["EVENT", req._children[0]._id, event2])
    adapter1.receive(["EVENT", req._children[0]._id, event3])
    adapter2.receive(["EVENT", req._children[1]._id, event1])
    adapter2.receive(["EVENT", req._children[1]._id, event4])

    await vi.runAllTimers()

    expect(duplicateSpy).toHaveBeenCalledWith(event1, '2')
    expect(filteredSpy).toHaveBeenCalledWith(event2, '1')
    expect(invalidSpy).toHaveBeenCalledWith(event3, '1')
    expect(eventSpy).toHaveBeenCalledWith(event1, '1')
    expect(eoseSpy).toHaveBeenCalledTimes(0)

    adapter1.receive(["EOSE", req._children[0]._id])
    adapter2.receive(["EOSE", req._children[1]._id])

    expect(eoseSpy).toHaveBeenCalledTimes(2)
    expect(closeSpy).toHaveBeenCalledTimes(1)
  })
})