141 lines
5.2 KiB
TypeScript
141 lines
5.2 KiB
TypeScript
import { describe, expect } from "bun:test"
|
|
import { EventV2Bridge } from "../../src/event-v2-bridge"
|
|
import { Config } from "../../src/config/config"
|
|
import { Plugin } from "../../src/plugin"
|
|
import { Pty } from "../../src/pty"
|
|
import type { PtyID } from "../../src/pty/schema"
|
|
import { Cause, Effect, Exit, Layer, Queue } from "effect"
|
|
import { testEffect } from "../lib/effect"
|
|
|
|
type PtyEvent = { type: "created" | "exited" | "deleted"; id: PtyID }
|
|
|
|
const it = testEffect(
|
|
Pty.layer.pipe(
|
|
Layer.provideMerge(EventV2Bridge.defaultLayer),
|
|
Layer.provideMerge(Config.defaultLayer),
|
|
Layer.provideMerge(Plugin.defaultLayer),
|
|
),
|
|
)
|
|
const ptyTest = process.platform === "win32" ? it.instance.skip : it.instance
|
|
|
|
const subscribePtyEvents = Effect.fn("PtySessionTest.subscribePtyEvents")(function* () {
|
|
const source = yield* EventV2Bridge.Service
|
|
const events = yield* Queue.unbounded<PtyEvent>()
|
|
|
|
const unsubscribe = yield* source.listen((event) => {
|
|
if (event.type === Pty.Event.Created.type)
|
|
Queue.offerUnsafe(events, { type: "created", id: (event.data as typeof Pty.Event.Created.data.Type).info.id })
|
|
if (event.type === Pty.Event.Exited.type)
|
|
Queue.offerUnsafe(events, { type: "exited", id: (event.data as typeof Pty.Event.Exited.data.Type).id })
|
|
if (event.type === Pty.Event.Deleted.type)
|
|
Queue.offerUnsafe(events, { type: "deleted", id: (event.data as typeof Pty.Event.Deleted.data.Type).id })
|
|
return Effect.void
|
|
})
|
|
yield* Effect.addFinalizer(() => unsubscribe)
|
|
|
|
return events
|
|
})
|
|
|
|
const createPty = Effect.fn("PtySessionTest.createPty")(function* (input: Pty.CreateInput) {
|
|
const pty = yield* Pty.Service
|
|
return yield* Effect.acquireRelease(pty.create(input), (info) => pty.remove(info.id).pipe(Effect.ignore))
|
|
})
|
|
|
|
const waitForEvents = (events: Queue.Queue<PtyEvent>, id: PtyID, count: number) => {
|
|
return Effect.gen(function* () {
|
|
const picked: Array<PtyEvent["type"]> = []
|
|
while (picked.length < count) {
|
|
const evt = yield* Queue.take(events)
|
|
if (evt.id === id) picked.push(evt.type)
|
|
}
|
|
return picked
|
|
}).pipe(
|
|
Effect.timeoutOrElse({
|
|
duration: "5 seconds",
|
|
orElse: () => Effect.fail(new Error("timeout waiting for pty events")),
|
|
}),
|
|
)
|
|
}
|
|
|
|
describe("pty", () => {
|
|
it.instance(
|
|
"returns typed not found errors for missing sessions",
|
|
() =>
|
|
Effect.gen(function* () {
|
|
const pty = yield* Pty.Service
|
|
const id = "pty_missing" as PtyID
|
|
let closed = false
|
|
const socket = {
|
|
readyState: 1,
|
|
send: () => {},
|
|
close: () => {
|
|
closed = true
|
|
},
|
|
}
|
|
|
|
const get = yield* pty.get(id).pipe(Effect.exit)
|
|
expect(Exit.isFailure(get)).toBe(true)
|
|
if (Exit.isFailure(get)) expect(Cause.squash(get.cause)).toMatchObject({ _tag: "Pty.NotFoundError", ptyID: id })
|
|
|
|
const update = yield* pty.update(id, { title: "missing" }).pipe(Effect.exit)
|
|
expect(Exit.isFailure(update)).toBe(true)
|
|
if (Exit.isFailure(update))
|
|
expect(Cause.squash(update.cause)).toMatchObject({ _tag: "Pty.NotFoundError", ptyID: id })
|
|
|
|
const remove = yield* pty.remove(id).pipe(Effect.exit)
|
|
expect(Exit.isFailure(remove)).toBe(true)
|
|
if (Exit.isFailure(remove))
|
|
expect(Cause.squash(remove.cause)).toMatchObject({ _tag: "Pty.NotFoundError", ptyID: id })
|
|
|
|
const resize = yield* pty.resize(id, 80, 24).pipe(Effect.exit)
|
|
expect(Exit.isFailure(resize)).toBe(true)
|
|
if (Exit.isFailure(resize))
|
|
expect(Cause.squash(resize.cause)).toMatchObject({ _tag: "Pty.NotFoundError", ptyID: id })
|
|
|
|
const write = yield* pty.write(id, "input").pipe(Effect.exit)
|
|
expect(Exit.isFailure(write)).toBe(true)
|
|
if (Exit.isFailure(write))
|
|
expect(Cause.squash(write.cause)).toMatchObject({ _tag: "Pty.NotFoundError", ptyID: id })
|
|
|
|
const connect = yield* pty.connect(id, socket).pipe(Effect.exit)
|
|
expect(Exit.isFailure(connect)).toBe(true)
|
|
if (Exit.isFailure(connect))
|
|
expect(Cause.squash(connect.cause)).toMatchObject({ _tag: "Pty.NotFoundError", ptyID: id })
|
|
expect(closed).toBe(true)
|
|
}),
|
|
{ git: true },
|
|
)
|
|
|
|
ptyTest(
|
|
"publishes created, exited, deleted in order for a short-lived process",
|
|
() =>
|
|
Effect.gen(function* () {
|
|
const events = yield* subscribePtyEvents()
|
|
const info = yield* createPty({
|
|
command: "/usr/bin/env",
|
|
args: ["sh", "-c", "sleep 0.1"],
|
|
title: "sleep",
|
|
})
|
|
|
|
expect(yield* waitForEvents(events, info.id, 3)).toEqual(["created", "exited", "deleted"])
|
|
}),
|
|
{ git: true },
|
|
)
|
|
|
|
ptyTest(
|
|
"publishes created, exited, deleted in order for /bin/sh + remove",
|
|
() =>
|
|
Effect.gen(function* () {
|
|
const pty = yield* Pty.Service
|
|
const events = yield* subscribePtyEvents()
|
|
const info = yield* createPty({ command: "/bin/sh", title: "sh" })
|
|
|
|
expect(yield* waitForEvents(events, info.id, 1)).toEqual(["created"])
|
|
yield* pty.write(info.id, "exit\n")
|
|
expect(yield* waitForEvents(events, info.id, 2)).toEqual(["exited", "deleted"])
|
|
yield* pty.remove(info.id).pipe(Effect.ignore)
|
|
}),
|
|
{ git: true },
|
|
)
|
|
})
|