Jon Staab
118 lines
3.2 KiB
TypeScript
118 lines
3.2 KiB
TypeScript
import {PublishStatus} from "@welshman/net"
|
|
import {NOTE, makeEvent} from "@welshman/util"
|
|
import {LOCAL_RELAY_URL} from "@welshman/relay"
|
|
import {getPubkey, makeSecret} from "@welshman/signer"
|
|
import {afterEach, beforeEach, describe, expect, it, vi} from "vitest"
|
|
import {repository, tracker} from "../src/core"
|
|
import {addSession, dropSession, makeNip01Session} from "../src/session"
|
|
import {
|
|
abortThunk,
|
|
Thunk,
|
|
MergedThunk,
|
|
prepEvent,
|
|
publishThunk,
|
|
thunkQueue,
|
|
walkThunks,
|
|
} from "../src/thunk"
|
|
|
|
const secret = makeSecret()
|
|
|
|
const pubkey = getPubkey(secret)
|
|
|
|
const mockRequest = {
|
|
event: prepEvent({...makeEvent(NOTE), pubkey}),
|
|
relays: [LOCAL_RELAY_URL],
|
|
}
|
|
|
|
describe("thunk", () => {
|
|
beforeEach(() => {
|
|
vi.useFakeTimers()
|
|
addSession(makeNip01Session(secret))
|
|
})
|
|
|
|
afterEach(async () => {
|
|
thunkQueue.stop()
|
|
thunkQueue.clear()
|
|
await vi.runAllTimersAsync()
|
|
vi.useRealTimers()
|
|
vi.clearAllMocks()
|
|
thunkQueue.start()
|
|
dropSession(pubkey)
|
|
})
|
|
|
|
describe("MergedThunk", () => {
|
|
it("should abort all thunks when merged controller aborts", () => {
|
|
const thunk1 = new Thunk(mockRequest)
|
|
const thunk2 = new Thunk(mockRequest)
|
|
const merged = new MergedThunk([thunk1, thunk2])
|
|
|
|
merged.controller.abort()
|
|
|
|
expect(thunk1.controller.signal.aborted).toBe(true)
|
|
expect(thunk2.controller.signal.aborted).toBe(true)
|
|
})
|
|
})
|
|
|
|
describe("walkThunks", () => {
|
|
it("should iterate through nested thunks", () => {
|
|
const thunk1 = new Thunk(mockRequest)
|
|
const thunk2 = new Thunk(mockRequest)
|
|
const merged = new MergedThunk([thunk1, thunk2])
|
|
const thunks = Array.from(walkThunks([merged, thunk1]))
|
|
|
|
expect(thunks).toHaveLength(3)
|
|
})
|
|
})
|
|
|
|
describe("publishThunk", () => {
|
|
it("should create and publish a thunk", async () => {
|
|
const publishSpy = vi.spyOn(repository, "publish")
|
|
const result = publishThunk(mockRequest)
|
|
|
|
expect(publishSpy).toHaveBeenCalled()
|
|
expect(result).toHaveProperty("event")
|
|
expect(result).toHaveProperty("options")
|
|
})
|
|
|
|
it("should handle abort", () => {
|
|
const removeEventSpy = vi.spyOn(repository, "removeEvent")
|
|
const thunk = publishThunk(mockRequest)
|
|
|
|
thunk.controller.abort()
|
|
|
|
expect(removeEventSpy).toHaveBeenCalledWith(thunk.event.id)
|
|
})
|
|
})
|
|
|
|
describe("abortThunk", () => {
|
|
it("should abort a thunk and clean up", () => {
|
|
const thunk = new Thunk(mockRequest)
|
|
|
|
abortThunk(thunk)
|
|
|
|
expect(repository.removeEvent).toHaveBeenCalledWith(thunk.event.id)
|
|
})
|
|
})
|
|
|
|
it("should update status during publishing", async () => {
|
|
const track = vi.spyOn(tracker, "track")
|
|
const thunk = new Thunk(mockRequest)
|
|
|
|
// Start the publish process
|
|
thunkQueue.push(thunk)
|
|
|
|
// Wait for initial async operations
|
|
await vi.runAllTimersAsync()
|
|
|
|
expect(thunk.status[LOCAL_RELAY_URL]).toEqual(PublishStatus.Success)
|
|
|
|
// Verify tracker was called on success
|
|
expect(track).toHaveBeenCalledWith(thunk.event.id, LOCAL_RELAY_URL)
|
|
|
|
await vi.runAllTimersAsync()
|
|
|
|
const finalStatus = await thunk.result
|
|
expect(finalStatus).toEqual({[LOCAL_RELAY_URL]: PublishStatus.Success})
|
|
})
|
|
})
|