mirror of
https://github.com/anomalyco/opencode.git
synced 2026-07-18 10:23:29 +00:00
8c94e9005f
Co-authored-by: Brendan Allan <14191578+Brendonovich@users.noreply.github.com> Co-authored-by: Kit Langton <kit.langton@gmail.com> Co-authored-by: opencode-agent[bot] <opencode-agent[bot]@users.noreply.github.com> Co-authored-by: Affan Ali <93028901+affanali2k3@users.noreply.github.com> Co-authored-by: affanali2k3 <affanalikhanxx@gmail.com> Co-authored-by: Frank <frank@anoma.ly> Co-authored-by: opencode-agent[bot] <219766164+opencode-agent[bot]@users.noreply.github.com> Co-authored-by: 𝓛𝓲𝓽𝓽𝓵𝓮 𝓕𝓻𝓪𝓷𝓴 <little-frank@opencord.local> Co-authored-by: Aiden Cline <63023139+rekram1-node@users.noreply.github.com> Co-authored-by: Jay V <air@live.ca> Co-authored-by: Dax Raad <d@ironbay.co> Co-authored-by: Aarav Sareen <96787824+arvsrn@users.noreply.github.com> Co-authored-by: OpeOginni <107570612+OpeOginni@users.noreply.github.com> Co-authored-by: Luke Parker <10430890+Hona@users.noreply.github.com> Co-authored-by: Ben Guthrie <benjee.012@gmail.com> Co-authored-by: Dax <mail@thdxr.com> Co-authored-by: Filip <34747899+neriousy@users.noreply.github.com> Co-authored-by: Max Anderson <max.a.anderson95@gmail.com> Co-authored-by: Brendan Allan <git@brendonovich.dev> Co-authored-by: Jack <jack@anoma.ly> Co-authored-by: Shoubhit Dash <shoubhit2005@gmail.com> Co-authored-by: Dustin Deus <deusdustin@gmail.com> Co-authored-by: starptech <starptech@starptechs-MBP.fritz.box> Co-authored-by: Aiden Cline <aidenpcline@gmail.com> Co-authored-by: usrnk1 <7547651+usrnk1@users.noreply.github.com> Co-authored-by: Jay <53023+jayair@users.noreply.github.com> Co-authored-by: runvip <164729189+runvip@users.noreply.github.com> Co-authored-by: opencode <opencode@sst.dev> Co-authored-by: Julian Coy <julian@ex-machina.co> Co-authored-by: Vladimir Glafirov <vglafirov@gitlab.com>
62 lines
2.4 KiB
TypeScript
62 lines
2.4 KiB
TypeScript
import { describe, expect } from "bun:test"
|
|
import { Effect, Layer } from "effect"
|
|
import { LayerNode } from "@opencode-ai/core/effect/layer-node"
|
|
import { PtyID } from "@opencode-ai/core/pty/schema"
|
|
import { PtyTicket } from "@opencode-ai/core/pty/ticket"
|
|
import { WorkspaceV2 } from "@opencode-ai/core/workspace"
|
|
import { testEffect } from "../lib/effect"
|
|
|
|
const it = testEffect(LayerNode.compile(PtyTicket.node))
|
|
const itExpiring = testEffect(
|
|
LayerNode.compile(PtyTicket.node, [[PtyTicket.node, Layer.effect(PtyTicket.Service, PtyTicket.make(5))]]),
|
|
)
|
|
|
|
describe("PTY websocket tickets", () => {
|
|
it.live("consumes tickets once", () =>
|
|
Effect.gen(function* () {
|
|
const tickets = yield* PtyTicket.Service
|
|
const scope = { ptyID: PtyID.ascending(), directory: "/tmp/a" }
|
|
const issued = yield* tickets.issue(scope)
|
|
|
|
expect(yield* tickets.consume({ ...scope, ticket: issued.ticket })).toBe(true)
|
|
expect(yield* tickets.consume({ ...scope, ticket: issued.ticket })).toBe(false)
|
|
}),
|
|
)
|
|
|
|
it.live("rejects tickets scoped to a different request", () =>
|
|
Effect.gen(function* () {
|
|
const tickets = yield* PtyTicket.Service
|
|
const ptyID = PtyID.ascending()
|
|
const issued = yield* tickets.issue({ ptyID, directory: "/tmp/a" })
|
|
|
|
expect(yield* tickets.consume({ ptyID, directory: "/tmp/b", ticket: issued.ticket })).toBe(false)
|
|
expect(yield* tickets.consume({ ptyID, directory: "/tmp/a", ticket: issued.ticket })).toBe(true)
|
|
}),
|
|
)
|
|
|
|
itExpiring.live("rejects tickets after the TTL elapses", () =>
|
|
Effect.gen(function* () {
|
|
const tickets = yield* PtyTicket.Service
|
|
const ptyID = PtyID.ascending()
|
|
const issued = yield* tickets.issue({ ptyID })
|
|
|
|
yield* Effect.promise(() => new Promise((resolve) => setTimeout(resolve, 25)))
|
|
|
|
expect(yield* tickets.consume({ ptyID, ticket: issued.ticket })).toBe(false)
|
|
}),
|
|
)
|
|
|
|
it.live("rejects tickets scoped to a different workspace", () =>
|
|
Effect.gen(function* () {
|
|
const tickets = yield* PtyTicket.Service
|
|
const ptyID = PtyID.ascending()
|
|
const workspaceID = WorkspaceV2.ID.ascending()
|
|
const issued = yield* tickets.issue({ ptyID, workspaceID })
|
|
|
|
expect(yield* tickets.consume({ ptyID, workspaceID: WorkspaceV2.ID.ascending(), ticket: issued.ticket })).toBe(
|
|
false,
|
|
)
|
|
expect(yield* tickets.consume({ ptyID, workspaceID, ticket: issued.ticket })).toBe(true)
|
|
}),
|
|
)
|
|
})
|