|
|
@@ -6,6 +6,7 @@ import { Effect, Schedule } from "effect"
|
|
|
import { SessionRetry } from "../../src/session/retry"
|
|
|
import { MessageV2 } from "../../src/session/message-v2"
|
|
|
import { ProviderID } from "../../src/provider/schema"
|
|
|
+import { AppRuntime } from "../../src/effect/app-runtime"
|
|
|
import { SessionID } from "../../src/session/schema"
|
|
|
import { SessionStatus } from "../../src/session/status"
|
|
|
import { Instance } from "../../src/project/instance"
|
|
|
@@ -94,12 +95,16 @@ describe("session.retry.delay", () => {
|
|
|
parse: (err) => err as MessageV2.APIError,
|
|
|
set: (info) =>
|
|
|
Effect.promise(() =>
|
|
|
- SessionStatus.set(sessionID, {
|
|
|
- type: "retry",
|
|
|
- attempt: info.attempt,
|
|
|
- message: info.message,
|
|
|
- next: info.next,
|
|
|
- }),
|
|
|
+ AppRuntime.runPromise(
|
|
|
+ SessionStatus.Service.use((svc) =>
|
|
|
+ svc.set(sessionID, {
|
|
|
+ type: "retry",
|
|
|
+ attempt: info.attempt,
|
|
|
+ message: info.message,
|
|
|
+ next: info.next,
|
|
|
+ }),
|
|
|
+ ),
|
|
|
+ ),
|
|
|
),
|
|
|
}),
|
|
|
)
|
|
|
@@ -108,7 +113,7 @@ describe("session.retry.delay", () => {
|
|
|
}),
|
|
|
)
|
|
|
|
|
|
- expect(await SessionStatus.get(sessionID)).toMatchObject({
|
|
|
+ expect(await AppRuntime.runPromise(SessionStatus.Service.use((svc) => svc.get(sessionID)))).toMatchObject({
|
|
|
type: "retry",
|
|
|
attempt: 2,
|
|
|
message: "boom",
|