vrr/opencode
2
0
Fork 0
mirror of https://github.com/anomalyco/opencode.git synced 2026-08-04 13:09:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

test(provider): synchronize model refresh event

Browse source
This commit is contained in:
Aiden Cline 2026-06-11 21:50:22 -05:00
parent a03882ff48
commit 529106dae8
1 changed files with 19 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

39
packages/opencode/test/provider/digitalocean.test.ts
View file

@ -1,13 +1,28 @@
import { expect } from "bun:test"
import { Provider } from "../../src/provider/provider"
import { Effect } from "effect"
import { pollWithTimeout, testEffect } from "../lib/effect"
import { Deferred, Effect } from "effect"
import { testEffect } from "../lib/effect"
import { ProviderV2 } from "@opencode-ai/core/provider"
import { GlobalBus, type GlobalEvent } from "@/bus/global"
const DIGITALOCEAN = ProviderV2.ID.make("digitalocean")
const it = testEffect(Provider.defaultLayer)
const refreshedModels = Effect.fn("DigitalOceanTest.refreshedModels")(function* () {
const provider = yield* Provider.Service
const refreshed = yield* Deferred.make<void>()
const listener = (event: GlobalEvent) => {
if (event.payload.type !== "provider.models.updated" || event.payload.properties.providerID !== DIGITALOCEAN) return
Deferred.doneUnsafe(refreshed, Effect.void)
}
GlobalBus.on("event", listener)
yield* Effect.addFinalizer(() => Effect.sync(() => GlobalBus.off("event", listener)))
yield* provider.list()
yield* Deferred.await(refreshed)
return (yield* provider.list())[DIGITALOCEAN].models
})
const withEnv = <A, E, R>(values: Record<string, string>, effect: Effect.Effect<A, E, R>) =>
Effect.acquireUseRelease(
Effect.sync(() => {
@ -73,15 +88,7 @@ it.instance(
oauth_expires: String(Date.now() + 60 * 60 * 1000),
},
Effect.gen(function* () {
const provider = yield* Provider.Service
const models = yield* pollWithTimeout(
provider.list().pipe(
Effect.map((providers) =>
providers[DIGITALOCEAN].models["router:my-router"] ? providers[DIGITALOCEAN].models : undefined,
),
),
"digitalocean router models were not refreshed",
)
const models = yield* refreshedModels()
expect(models["router:my-router"]).toBeDefined()
expect(models["router:my-router"].api.id).toBe("router:my-router")
expect(models["router:my-router"].api.url).toBe("https://inference.do-ai.run/v1")
@ -103,15 +110,7 @@ it.instance(
oauth_expires: "1",
},
Effect.gen(function* () {
const provider = yield* Provider.Service
const models = yield* pollWithTimeout(
provider.list().pipe(
Effect.map((providers) =>
providers[DIGITALOCEAN].models["router:stale-router"] ? providers[DIGITALOCEAN].models : undefined,
),
),
"digitalocean cached router models were not restored",
)
const models = yield* refreshedModels()
expect(models["router:stale-router"]).toBeDefined()
}),
),