apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-03 23:11:00 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Use SubscriptionRef for client connection state

Browse source
This commit is contained in:
elpresidank 2026-06-04 05:30:31 -05:00
parent 68cbcde1f6
commit 0862250dab
4 changed files with 141 additions and 51 deletions
Download patch file Download diff file Expand all files Collapse all files

48
ts/packages/client/src/__tests__/rpc-timeout.test.ts
View file

@ -1,8 +1,8 @@
import { Effect } from "effect";
import { describe, expect, it, vi } from "vitest";
import { DispatchError, DispatchStreamChunk } from "../rpc/contract";
import { type DispatchInput, withDispatchRequestPolicy } from "../socket/effect-rpc-client";
import { makeBaseApiWithRpc } from "../socket/trustgraph-socket";
import { type DispatchInput, type RpcConnectionState, withDispatchRequestPolicy } from "../socket/effect-rpc-client";
import { type ConnectionState, makeBaseApiWithRpc } from "../socket/trustgraph-socket";
const input: DispatchInput = {
scope: "global",
@ -11,6 +11,50 @@ const input: DispatchInput = {
};
describe("Effect RPC request policy", () => {
it("replays and updates connection state through the SubscriptionRef-backed bridge", async () => {
let rpcListener: ((state: RpcConnectionState) => void) | undefined;
const api = makeBaseApiWithRpc("alice", undefined, "ws://example.test/rpc", {
dispatch: vi.fn(() => Promise.resolve({ ok: true })),
dispatchStream: vi.fn(() => Promise.resolve(undefined)),
close: vi.fn(() => Promise.resolve()),
subscribe: vi.fn((listener) => {
rpcListener = listener;
listener({ status: "connecting" });
return () => undefined;
}),
});
const observed: ConnectionState[] = [];
const unsubscribe = api.onConnectionStateChange((state) => {
observed.push(state);
});
expect(observed).toEqual([{ status: "connecting", hasApiKey: false }]);
const listener = rpcListener;
expect(listener).toBeDefined();
if (listener !== undefined) {
listener({ status: "connected" });
}
await Effect.runPromise(Effect.yieldNow);
expect(observed).toEqual([
{ status: "connecting", hasApiKey: false },
{ status: "unauthenticated", hasApiKey: false },
]);
unsubscribe();
await Effect.runPromise(Effect.yieldNow);
if (listener !== undefined) {
listener({ status: "failed", lastError: "boom" });
}
await Effect.runPromise(Effect.yieldNow);
expect(observed).toEqual([
{ status: "connecting", hasApiKey: false },
{ status: "unauthenticated", hasApiKey: false },
]);
});
it("threads BaseApi timeout and retry options into dispatch calls", async () => {
const dispatch = vi.fn(() => Promise.resolve({ ok: true }));
const api = makeBaseApiWithRpc("alice", undefined, "ws://example.test/rpc", {