mirror of
https://github.com/anomalyco/opencode.git
synced 2026-07-10 03:38:31 +00:00
246 lines
9.5 KiB
TypeScript
246 lines
9.5 KiB
TypeScript
import { expect, test } from "bun:test"
|
|
import { DateTime, Effect, Stream } from "effect"
|
|
import { HttpClient, HttpClientResponse } from "effect/unstable/http"
|
|
import { AbsolutePath, Agent, Location, Model, OpenCode, Prompt, Session, SessionMessage } from "../src/effect"
|
|
|
|
test("sessions.get returns the decoded Effect projection", async () => {
|
|
const httpClient = HttpClient.make((request) =>
|
|
Effect.succeed(HttpClientResponse.fromWeb(request, Response.json(session))),
|
|
)
|
|
const result = await Effect.gen(function* () {
|
|
const client = yield* OpenCode.make({ baseUrl: "http://localhost:3000" })
|
|
return yield* client.sessions.get({ sessionID: Session.ID.make("ses_test") })
|
|
}).pipe(Effect.provideService(HttpClient.HttpClient, httpClient), Effect.runPromise)
|
|
|
|
expect(DateTime.toEpochMillis(result.time.created)).toBe(1_717_171_717_000)
|
|
})
|
|
|
|
test("events.subscribe exposes and decodes the native Effect event stream", async () => {
|
|
const httpClient = HttpClient.make((request) =>
|
|
Effect.succeed(
|
|
HttpClientResponse.fromWeb(
|
|
request,
|
|
new Response(
|
|
`data: ${JSON.stringify({ id: "evt_connected", type: "server.connected", data: {} })}\n\n` +
|
|
`data: ${JSON.stringify(modelSwitchedEvent)}\n\n`,
|
|
{ headers: { "content-type": "text/event-stream" } },
|
|
),
|
|
),
|
|
),
|
|
)
|
|
const events = await Effect.gen(function* () {
|
|
const client = yield* OpenCode.make({ baseUrl: "http://localhost:3000" })
|
|
return yield* client.events.subscribe().pipe(Stream.runCollect)
|
|
}).pipe(Effect.provideService(HttpClient.HttpClient, httpClient), Effect.runPromise)
|
|
|
|
expect(Array.from(events).map((event) => event.type)).toEqual(["server.connected", "session.next.model.switched"])
|
|
const durable = events[1]
|
|
if (durable?.type !== "session.next.model.switched") throw new Error("Expected model event")
|
|
expect(DateTime.toEpochMillis(durable.data.timestamp)).toBe(1_717_171_717_000)
|
|
expect(durable.durable).toEqual({ aggregateID: "ses_test", seq: 1, version: 1 })
|
|
})
|
|
|
|
test("events.subscribe terminates on Effect protocol decode failures", async () => {
|
|
const httpClient = HttpClient.make((request) =>
|
|
Effect.succeed(
|
|
HttpClientResponse.fromWeb(
|
|
request,
|
|
new Response(`data: {"type":"server.connected"}\n\n`, {
|
|
headers: { "content-type": "text/event-stream" },
|
|
}),
|
|
),
|
|
),
|
|
)
|
|
const error = await Effect.gen(function* () {
|
|
const client = yield* OpenCode.make({ baseUrl: "http://localhost:3000" })
|
|
return yield* client.events.subscribe().pipe(Stream.runCollect, Effect.flip)
|
|
}).pipe(Effect.provideService(HttpClient.HttpClient, httpClient), Effect.runPromise)
|
|
|
|
expect(error._tag).toBe("ClientError")
|
|
})
|
|
|
|
test("session methods retain decoded Effect inputs and outputs", async () => {
|
|
const historyQueries: Array<Record<string, string>> = []
|
|
let historyPage = 0
|
|
const httpClient = HttpClient.make((request) => {
|
|
const url = request.url
|
|
if (url.includes("/event")) {
|
|
return Effect.succeed(
|
|
HttpClientResponse.fromWeb(
|
|
request,
|
|
new Response(`data: ${JSON.stringify(modelSwitchedEvent)}\n\n`, {
|
|
headers: { "content-type": "text/event-stream" },
|
|
}),
|
|
),
|
|
)
|
|
}
|
|
if (url.includes("/history")) {
|
|
historyPage++
|
|
historyQueries.push(Object.fromEntries(request.urlParams.params))
|
|
return Effect.succeed(
|
|
HttpClientResponse.fromWeb(
|
|
request,
|
|
Response.json(
|
|
historyPage === 1 ? { data: [modelSwitchedEvent], hasMore: true } : { data: [], hasMore: false },
|
|
),
|
|
),
|
|
)
|
|
}
|
|
if (url.includes("/prompt")) {
|
|
return Effect.succeed(HttpClientResponse.fromWeb(request, Response.json(admission)))
|
|
}
|
|
if (url.includes("/context")) {
|
|
return Effect.succeed(HttpClientResponse.fromWeb(request, Response.json({ data: [] })))
|
|
}
|
|
if (url.includes("/message/")) {
|
|
return Effect.succeed(HttpClientResponse.fromWeb(request, Response.json({ data: modelSwitchedMessage })))
|
|
}
|
|
if (url.endsWith("/api/session/active")) {
|
|
return Effect.succeed(
|
|
HttpClientResponse.fromWeb(request, Response.json({ data: { ses_test: { type: "running" } } })),
|
|
)
|
|
}
|
|
if (request.method === "POST" && url.endsWith("/api/session")) {
|
|
return Effect.succeed(HttpClientResponse.fromWeb(request, Response.json(session)))
|
|
}
|
|
if (request.method === "POST") {
|
|
return Effect.succeed(HttpClientResponse.fromWeb(request, new Response(null, { status: 204 })))
|
|
}
|
|
return Effect.succeed(
|
|
HttpClientResponse.fromWeb(request, Response.json({ data: [session.data], cursor: { next: "next" } })),
|
|
)
|
|
})
|
|
const result = await Effect.gen(function* () {
|
|
const client = yield* OpenCode.make({ baseUrl: "http://localhost:3000" })
|
|
const page = yield* client.sessions.list({ limit: 10 })
|
|
const active = yield* client.sessions.active()
|
|
const created = yield* client.sessions.create({
|
|
location: Location.Ref.make({ directory: AbsolutePath.make("/tmp/project") }),
|
|
})
|
|
yield* client.sessions.switchAgent({ sessionID: Session.ID.make("ses_test"), agent: Agent.ID.make("build") })
|
|
yield* client.sessions.switchModel({
|
|
sessionID: Session.ID.make("ses_test"),
|
|
model: Model.Ref.make({ id: "claude", providerID: "anthropic" }),
|
|
})
|
|
const admitted = yield* client.sessions.prompt({
|
|
sessionID: Session.ID.make("ses_test"),
|
|
prompt: Prompt.make({ text: "Hello" }),
|
|
resume: false,
|
|
})
|
|
yield* client.sessions.compact({ sessionID: Session.ID.make("ses_test") })
|
|
yield* client.sessions.wait({ sessionID: Session.ID.make("ses_test") })
|
|
const context = yield* client.sessions.context({ sessionID: Session.ID.make("ses_test") })
|
|
const history = yield* client.sessions.history({
|
|
sessionID: Session.ID.make("ses_test"),
|
|
after: 0,
|
|
limit: 1,
|
|
})
|
|
const historyNext = history.hasMore
|
|
? yield* client.sessions.history({
|
|
sessionID: Session.ID.make("ses_test"),
|
|
after: history.data.at(-1)?.durable?.seq,
|
|
limit: 2,
|
|
})
|
|
: undefined
|
|
const events = yield* client.sessions
|
|
.events({ sessionID: Session.ID.make("ses_test"), after: 0 })
|
|
.pipe(Stream.runCollect)
|
|
yield* client.sessions.interrupt({ sessionID: Session.ID.make("ses_test") })
|
|
const message = yield* client.sessions.message({
|
|
sessionID: Session.ID.make("ses_test"),
|
|
messageID: SessionMessage.ID.make("msg_model"),
|
|
})
|
|
return { page, active, created, admitted, context, history, historyNext, events, message }
|
|
}).pipe(Effect.provideService(HttpClient.HttpClient, httpClient), Effect.runPromise)
|
|
|
|
expect(DateTime.toEpochMillis(result.page.data[0].time.created)).toBe(1_717_171_717_000)
|
|
expect(result.active).toEqual({ ses_test: { type: "running" } })
|
|
expect(Object.getPrototypeOf(result.page.data[0])).toBe(Object.prototype)
|
|
expect(Object.getPrototypeOf(result.created)).toBe(Object.prototype)
|
|
expect(result.created.id).toBe("ses_test")
|
|
expect(Object.getPrototypeOf(result.admitted)).toBe(Object.prototype)
|
|
expect(Object.getPrototypeOf(result.admitted.prompt)).toBe(Object.prototype)
|
|
expect(DateTime.toEpochMillis(result.admitted.timeCreated)).toBe(1_717_171_717_000)
|
|
expect(result.context).toEqual([])
|
|
expect(DateTime.toEpochMillis(result.history.data[0].data.timestamp)).toBe(1_717_171_717_000)
|
|
expect(result.history).toEqual(expect.objectContaining({ hasMore: true }))
|
|
expect(result.historyNext).toEqual({ data: [], hasMore: false })
|
|
expect(historyQueries[0]).toEqual({ limit: "1", after: "0" })
|
|
expect(historyQueries[1]).toEqual({ limit: "2", after: "1" })
|
|
expect(DateTime.toEpochMillis(result.events[0].data.timestamp)).toBe(1_717_171_717_000)
|
|
expect(result.message).toEqual(expect.objectContaining({ id: "msg_model", type: "model-switched" }))
|
|
})
|
|
|
|
test("sessions.history retains the typed SessionNotFoundError", async () => {
|
|
const httpClient = HttpClient.make((request) =>
|
|
Effect.succeed(
|
|
HttpClientResponse.fromWeb(
|
|
request,
|
|
Response.json(
|
|
{ _tag: "SessionNotFoundError", sessionID: "ses_missing", message: "Session not found" },
|
|
{ status: 404 },
|
|
),
|
|
),
|
|
),
|
|
)
|
|
const error = await Effect.gen(function* () {
|
|
const client = yield* OpenCode.make({ baseUrl: "http://localhost:3000" })
|
|
return yield* client.sessions
|
|
.history({
|
|
sessionID: Session.ID.make("ses_missing"),
|
|
})
|
|
.pipe(Effect.flip)
|
|
}).pipe(Effect.provideService(HttpClient.HttpClient, httpClient), Effect.runPromise)
|
|
|
|
expect(error._tag).toBe("SessionNotFoundError")
|
|
})
|
|
|
|
const session = {
|
|
data: {
|
|
id: "ses_test",
|
|
projectID: "project",
|
|
cost: 0,
|
|
tokens: {
|
|
input: 1,
|
|
output: 2,
|
|
reasoning: 3,
|
|
cache: { read: 4, write: 5 },
|
|
},
|
|
time: {
|
|
created: 1_717_171_717_000,
|
|
updated: 1_717_171_717_000,
|
|
},
|
|
title: "Test",
|
|
location: { directory: "/tmp/project" },
|
|
},
|
|
}
|
|
|
|
const admission = {
|
|
data: {
|
|
admittedSeq: 0,
|
|
id: "msg_test",
|
|
sessionID: "ses_test",
|
|
prompt: { text: "Hello" },
|
|
delivery: "steer",
|
|
timeCreated: 1_717_171_717_000,
|
|
},
|
|
}
|
|
|
|
const modelSwitchedMessage = {
|
|
id: "msg_model",
|
|
type: "model-switched",
|
|
time: { created: 1_717_171_717_000 },
|
|
model: { id: "claude", providerID: "anthropic" },
|
|
}
|
|
|
|
const modelSwitchedEvent = {
|
|
id: "evt_model",
|
|
type: "session.next.model.switched",
|
|
durable: { aggregateID: "ses_test", seq: 1, version: 1 },
|
|
data: {
|
|
timestamp: 1_717_171_717_000,
|
|
sessionID: "ses_test",
|
|
messageID: "msg_model",
|
|
model: { id: "claude", providerID: "anthropic" },
|
|
},
|
|
}
|