52 lines
2.0 KiB
TypeScript
52 lines
2.0 KiB
TypeScript
|
import { beforeEach, describe, expect, test } from "bun:test"
|
||
|
|
import {
|
||
|
|
createMessageQueueTarget,
|
||
|
|
getMessageQueueKey,
|
||
|
|
migrateMessageQueueState,
|
||
|
|
parseMessageQueueKey,
|
||
|
|
useMessageQueueStore,
|
||
|
|
} from "./messageQueueStore"
|
||
|
|
|
||
|
|
beforeEach(() => {
|
||
|
|
useMessageQueueStore.setState({ queuedMessages: {}, quarantinedLegacyMessages: {} })
|
||
|
|
})
|
||
|
|
|
||
|
|
describe("message queue runtime ownership", () => {
|
||
|
|
test("isolates colliding session IDs by runtime and directory", () => {
|
||
|
|
const a = createMessageQueueTarget("session-1", "/repo", "runtime-a")!
|
||
|
|
const b = createMessageQueueTarget("session-1", "/repo", "runtime-b")!
|
||
|
|
useMessageQueueStore.getState().addToQueue(a, { content: "from A" })
|
||
|
|
useMessageQueueStore.getState().addToQueue(b, { content: "from B" })
|
||
|
|
|
||
|
|
expect(useMessageQueueStore.getState().getQueueForTarget(a)[0]?.content).toBe("from A")
|
||
|
|
expect(useMessageQueueStore.getState().getQueueForTarget(b)[0]?.content).toBe("from B")
|
||
|
|
})
|
||
|
|
|
||
|
|
test("round trips a composite queue key", () => {
|
||
|
|
const target = createMessageQueueTarget("session-1", "/repo", "runtime-a")!
|
||
|
|
expect(parseMessageQueueKey(getMessageQueueKey(target))).toEqual(target)
|
||
|
|
})
|
||
|
|
|
||
|
|
test("quarantines legacy session-only queues instead of activating them", () => {
|
||
|
|
const migrated = migrateMessageQueueState({
|
||
|
|
queuedMessages: {
|
||
|
|
"session-1": [{ id: "queued-1", content: "legacy", createdAt: 1 }],
|
||
|
|
},
|
||
|
|
}, 1)
|
||
|
|
|
||
|
|
expect(migrated.queuedMessages).toEqual({})
|
||
|
|
expect(migrated.quarantinedLegacyMessages?.["session-1"]?.[0]?.content).toBe("legacy")
|
||
|
|
})
|
||
|
|
|
||
|
|
test("bounds each queue to the newest 20 messages", () => {
|
||
|
|
const target = createMessageQueueTarget("session-1", "/repo", "runtime-a")!
|
||
|
|
for (let index = 0; index < 25; index += 1) {
|
||
|
|
useMessageQueueStore.getState().addToQueue(target, { content: `message-${index}` })
|
||
|
|
}
|
||
|
|
|
||
|
|
const queue = useMessageQueueStore.getState().getQueueForTarget(target)
|
||
|
|
expect(queue).toHaveLength(20)
|
||
|
|
expect(queue[0]?.content).toBe("message-5")
|
||
|
|
})
|
||
|
|
})
|