openAgent

1import { describe, expect } from "bun:test"
2import { Effect, Layer } from "effect"
3import { LayerNode } from "@opencode-ai/core/effect/layer-node"
4import { PtyID } from "@opencode-ai/core/pty/schema"
5import { PtyTicket } from "@opencode-ai/core/pty/ticket"
6import { WorkspaceV2 } from "@opencode-ai/core/workspace"
7import { testEffect } from "../lib/effect"
8 
9const it = testEffect(LayerNode.compile(PtyTicket.node))
10const itExpiring = testEffect(
11  LayerNode.compile(PtyTicket.node, [[PtyTicket.node, Layer.effect(PtyTicket.Service, PtyTicket.make(5))]]),
12)
13 
14describe("PTY websocket tickets", () => {
15  it.live("consumes tickets once", () =>
16    Effect.gen(function* () {
17      const tickets = yield* PtyTicket.Service
18      const scope = { ptyID: PtyID.ascending(), directory: "/tmp/a" }
19      const issued = yield* tickets.issue(scope)
20 
21      expect(yield* tickets.consume({ ...scope, ticket: issued.ticket })).toBe(true)
22      expect(yield* tickets.consume({ ...scope, ticket: issued.ticket })).toBe(false)
23    }),
24  )
25 
26  it.live("rejects tickets scoped to a different request", () =>
27    Effect.gen(function* () {
28      const tickets = yield* PtyTicket.Service
29      const ptyID = PtyID.ascending()
30      const issued = yield* tickets.issue({ ptyID, directory: "/tmp/a" })
31 
32      expect(yield* tickets.consume({ ptyID, directory: "/tmp/b", ticket: issued.ticket })).toBe(false)
33      expect(yield* tickets.consume({ ptyID, directory: "/tmp/a", ticket: issued.ticket })).toBe(true)
34    }),
35  )
36 
37  itExpiring.live("rejects tickets after the TTL elapses", () =>
38    Effect.gen(function* () {
39      const tickets = yield* PtyTicket.Service
40      const ptyID = PtyID.ascending()
41      const issued = yield* tickets.issue({ ptyID })
42 
43      yield* Effect.promise(() => new Promise((resolve) => setTimeout(resolve, 25)))
44 
45      expect(yield* tickets.consume({ ptyID, ticket: issued.ticket })).toBe(false)
46    }),
47  )
48 
49  it.live("rejects tickets scoped to a different workspace", () =>
50    Effect.gen(function* () {
51      const tickets = yield* PtyTicket.Service
52      const ptyID = PtyID.ascending()
53      const workspaceID = WorkspaceV2.ID.ascending()
54      const issued = yield* tickets.issue({ ptyID, workspaceID })
55 
56      expect(yield* tickets.consume({ ptyID, workspaceID: WorkspaceV2.ID.ascending(), ticket: issued.ticket })).toBe(
57        false,
58      )
59      expect(yield* tickets.consume({ ptyID, workspaceID, ticket: issued.ticket })).toBe(true)
60    }),
61  )
62})
63