Merge commit '36c66dd290d3ce6eb1ccd310d0c658d4a32bb8eb' as 'repos/effect'
This commit is contained in:
-Puter
292
repos/effect/packages/platform-node/test/RpcServer.test.ts
Normal file
292
repos/effect/packages/platform-node/test/RpcServer.test.ts
Normal file
@@ -0,0 +1,292 @@
|
||||
import { NodeHttpServer, NodeSocket, NodeSocketServer } from "@effect/platform-node"
|
||||
import { assert, describe, it } from "@effect/vitest"
|
||||
import { Cause, Deferred, Effect, Fiber, Layer, Ref, Schedule, Schema, Stream } from "effect"
|
||||
import { Entity, EntityProxy, EntityProxyServer, Sharding } from "effect/unstable/cluster"
|
||||
import { HttpClient, HttpClientRequest, HttpRouter, HttpServer } from "effect/unstable/http"
|
||||
import { Rpc, RpcClient, RpcGroup, RpcSerialization, RpcServer, RpcTest } from "effect/unstable/rpc"
|
||||
import { SocketServer } from "effect/unstable/socket"
|
||||
import { e2eSuite, UsersClient } from "./fixtures/rpc-e2e.ts"
|
||||
import { RpcLive, User } from "./fixtures/rpc-schemas.ts"
|
||||
|
||||
describe("RpcServer", () => {
|
||||
// http ndjson
|
||||
const HttpProtocol = RpcServer.layerProtocolHttp({ path: "/rpc" }).pipe(
|
||||
Layer.provide(HttpRouter.layer)
|
||||
)
|
||||
const HttpNdjsonServer = RpcLive.pipe(
|
||||
Layer.provideMerge(HttpProtocol),
|
||||
Layer.provide(HttpRouter.serve(HttpProtocol, { disableListenLog: true, disableLogger: true }))
|
||||
)
|
||||
const HttpNdjsonClient = UsersClient.layer.pipe(
|
||||
Layer.provide(
|
||||
RpcClient.layerProtocolHttp({
|
||||
url: "",
|
||||
transformClient: HttpClient.mapRequest(HttpClientRequest.appendUrl("/rpc"))
|
||||
})
|
||||
)
|
||||
)
|
||||
const CustomDefectLayer = HttpNdjsonClient.pipe(
|
||||
Layer.provideMerge(HttpNdjsonServer),
|
||||
Layer.provide([NodeHttpServer.layerTest, RpcSerialization.layerNdjson])
|
||||
)
|
||||
e2eSuite(
|
||||
"e2e http ndjson",
|
||||
HttpNdjsonClient.pipe(
|
||||
Layer.provideMerge(HttpNdjsonServer),
|
||||
Layer.provide([NodeHttpServer.layerTest, RpcSerialization.layerNdjson])
|
||||
)
|
||||
)
|
||||
e2eSuite(
|
||||
"e2e http msgpack",
|
||||
HttpNdjsonClient.pipe(
|
||||
Layer.provideMerge(HttpNdjsonServer),
|
||||
Layer.provide([NodeHttpServer.layerTest, RpcSerialization.layerMsgPack])
|
||||
)
|
||||
)
|
||||
e2eSuite(
|
||||
"e2e http jsonrpc",
|
||||
HttpNdjsonClient.pipe(
|
||||
Layer.provideMerge(HttpNdjsonServer),
|
||||
Layer.provide([NodeHttpServer.layerTest, RpcSerialization.layerNdJsonRpc()])
|
||||
)
|
||||
)
|
||||
|
||||
// websocket
|
||||
const WsProtocol = RpcServer.layerProtocolWebsocket({ path: "/rpc" }).pipe(
|
||||
Layer.provide(HttpRouter.layer)
|
||||
)
|
||||
const HttpWsServer = RpcLive.pipe(
|
||||
Layer.provideMerge(WsProtocol),
|
||||
Layer.provide(HttpRouter.serve(WsProtocol, { disableListenLog: true, disableLogger: true }))
|
||||
)
|
||||
const HttpWsClient = UsersClient.layer.pipe(
|
||||
Layer.provide(RpcClient.layerProtocolSocket()),
|
||||
Layer.provide(
|
||||
Effect.gen(function*() {
|
||||
const server = yield* HttpServer.HttpServer
|
||||
const address = server.address as HttpServer.TcpAddress
|
||||
return NodeSocket.layerWebSocket(`http://127.0.0.1:${address.port}/rpc`)
|
||||
}).pipe(Layer.unwrap)
|
||||
)
|
||||
)
|
||||
e2eSuite(
|
||||
"e2e ws ndjson",
|
||||
HttpWsClient.pipe(
|
||||
Layer.provideMerge(HttpWsServer),
|
||||
Layer.provide([NodeHttpServer.layerTest, RpcSerialization.layerNdjson])
|
||||
)
|
||||
)
|
||||
e2eSuite(
|
||||
"e2e ws json",
|
||||
HttpWsClient.pipe(
|
||||
Layer.provideMerge(HttpWsServer),
|
||||
Layer.provide([NodeHttpServer.layerTest, RpcSerialization.layerJson])
|
||||
)
|
||||
)
|
||||
e2eSuite(
|
||||
"e2e ws msgpack",
|
||||
HttpWsClient.pipe(
|
||||
Layer.provideMerge(HttpWsServer),
|
||||
Layer.provide([NodeHttpServer.layerTest, RpcSerialization.layerMsgPack])
|
||||
)
|
||||
)
|
||||
e2eSuite(
|
||||
"e2e ws jsonrpc",
|
||||
HttpWsClient.pipe(
|
||||
Layer.provideMerge(HttpWsServer),
|
||||
Layer.provide([NodeHttpServer.layerTest, RpcSerialization.layerJsonRpc()])
|
||||
)
|
||||
)
|
||||
|
||||
// tcp
|
||||
const TcpServer = RpcLive.pipe(
|
||||
Layer.provideMerge(RpcServer.layerProtocolSocketServer),
|
||||
Layer.provideMerge(NodeSocketServer.layer({ port: 0 }))
|
||||
)
|
||||
const TcpClient = UsersClient.layer.pipe(
|
||||
Layer.provide(RpcClient.layerProtocolSocket()),
|
||||
Layer.provide(
|
||||
Effect.gen(function*() {
|
||||
const server = yield* SocketServer.SocketServer
|
||||
const address = server.address as SocketServer.TcpAddress
|
||||
return NodeSocket.layerNet({ port: address.port })
|
||||
}).pipe(Layer.unwrap)
|
||||
)
|
||||
)
|
||||
e2eSuite(
|
||||
"e2e tcp ndjson",
|
||||
TcpClient.pipe(
|
||||
Layer.provideMerge(TcpServer),
|
||||
Layer.provide([NodeHttpServer.layerTest, RpcSerialization.layerNdjson])
|
||||
)
|
||||
)
|
||||
e2eSuite(
|
||||
"e2e tcp msgpack",
|
||||
TcpClient.pipe(
|
||||
Layer.provideMerge(TcpServer),
|
||||
Layer.provide([NodeHttpServer.layerTest, RpcSerialization.layerMsgPack])
|
||||
)
|
||||
)
|
||||
e2eSuite(
|
||||
"e2e tcp jsonrpc",
|
||||
TcpClient.pipe(
|
||||
Layer.provideMerge(TcpServer),
|
||||
Layer.provide([NodeHttpServer.layerTest, RpcSerialization.layerNdJsonRpc()])
|
||||
)
|
||||
)
|
||||
|
||||
// worker
|
||||
// const WorkerClient = UsersClient.layer.pipe(
|
||||
// Layer.provide(RpcClient.layerProtocolWorker({ size: 1 })),
|
||||
// Layer.provide(
|
||||
// NodeWorker.layerPlatform(() =>
|
||||
// CP.fork(new URL("./fixtures/rpc-worker.ts", import.meta.url), {
|
||||
// execPath: "node"
|
||||
// })
|
||||
// )
|
||||
// ),
|
||||
// Layer.merge(Layer.succeed(RpcServer.Protocol, {
|
||||
// supportsAck: true
|
||||
// } as any))
|
||||
// )
|
||||
// e2eSuite("e2e worker", WorkerClient)
|
||||
|
||||
describe("RpcTest", () => {
|
||||
it.effect("works", () =>
|
||||
Effect.gen(function*() {
|
||||
const client = yield* UsersClient
|
||||
const user = yield* client.GetUser({ id: "1" })
|
||||
assert.deepStrictEqual(user, new User({ id: "1", name: "Logged in user" }))
|
||||
}).pipe(Effect.provide(UsersClient.layerTest)))
|
||||
})
|
||||
|
||||
describe("custom defect schema", () => {
|
||||
it.effect("preserves full defect with custom schema", () =>
|
||||
Effect.gen(function*() {
|
||||
const client = yield* UsersClient
|
||||
const cause = yield* client.ProduceDefectCustom().pipe(
|
||||
Effect.sandbox,
|
||||
Effect.flip
|
||||
)
|
||||
const defect = Cause.squash(cause)
|
||||
assert.instanceOf(defect, Error)
|
||||
assert.strictEqual(defect.name, "CustomDefect")
|
||||
assert.strictEqual(defect.message, "detailed error")
|
||||
assert.strictEqual(defect.stack, "Error: detailed error\n at handler.ts:1")
|
||||
}).pipe(Effect.provide(CustomDefectLayer)))
|
||||
})
|
||||
|
||||
describe("entity proxy", () => {
|
||||
it.effect("provides handler context for generated rpc handlers", () =>
|
||||
Effect.gen(function*() {
|
||||
const TestEntity = Entity.make("TestEntity", [Rpc.make("NoPayload")])
|
||||
const TestEntityRpcs = EntityProxy.toRpcGroup(TestEntity)
|
||||
const called = yield* Deferred.make<void>()
|
||||
const testClient = (entityId: string) => ({
|
||||
NoPayload: (payload: void, options?: { readonly discard?: boolean }) =>
|
||||
Effect.gen(function*() {
|
||||
assert.strictEqual(entityId, "id")
|
||||
assert.strictEqual(payload, undefined)
|
||||
assert.strictEqual(options?.discard, true)
|
||||
yield* Deferred.succeed(called, undefined)
|
||||
})
|
||||
})
|
||||
const sharding = Sharding.Sharding.of({
|
||||
...({} as Sharding.Sharding["Service"]),
|
||||
isShutdown: Effect.succeed(false),
|
||||
makeClient: () => Effect.succeed(testClient) as never,
|
||||
pollStorage: Effect.void
|
||||
})
|
||||
|
||||
const client = yield* RpcTest.makeClient(TestEntityRpcs).pipe(
|
||||
Effect.provide(EntityProxyServer.layerRpcHandlers(TestEntity)),
|
||||
Effect.provideService(Sharding.Sharding, sharding)
|
||||
)
|
||||
|
||||
yield* client["TestEntity.NoPayloadDiscard"]({
|
||||
entityId: "id",
|
||||
payload: undefined
|
||||
})
|
||||
yield* Deferred.await(called)
|
||||
}))
|
||||
})
|
||||
|
||||
describe("unknown-tag isolation", () => {
|
||||
const Ticker = Rpc.make("Ticker", {
|
||||
success: Schema.Number,
|
||||
stream: true
|
||||
})
|
||||
const Ghost = Rpc.make("Ghost", {
|
||||
payload: { value: Schema.String },
|
||||
success: Schema.String
|
||||
})
|
||||
|
||||
const serverGroup = RpcGroup.make(Ticker)
|
||||
const clientGroup = RpcGroup.make(Ticker, Ghost)
|
||||
|
||||
const TickerHandlers = serverGroup.toLayer({
|
||||
Ticker: () => Stream.fromSchedule(Schedule.spaced("60 millis"))
|
||||
})
|
||||
|
||||
const IsolationServer = RpcServer.layer(serverGroup).pipe(
|
||||
Layer.provide(TickerHandlers),
|
||||
Layer.provideMerge(RpcServer.layerProtocolSocketServer),
|
||||
Layer.provideMerge(NodeSocketServer.layer({ port: 0 })),
|
||||
Layer.provide(RpcSerialization.layerNdjson)
|
||||
)
|
||||
const IsolationClient = RpcClient.layerProtocolSocket().pipe(
|
||||
Layer.provide(
|
||||
Effect.gen(function*() {
|
||||
const server = yield* SocketServer.SocketServer
|
||||
const address = server.address as SocketServer.TcpAddress
|
||||
return NodeSocket.layerNet({ port: address.port })
|
||||
}).pipe(Layer.unwrap)
|
||||
),
|
||||
Layer.provide(RpcSerialization.layerNdjson)
|
||||
)
|
||||
|
||||
it.live(
|
||||
"an unknown request tag fails only its own request, not other in-flight streams on the same connection",
|
||||
() =>
|
||||
Effect.gen(function*() {
|
||||
const client = yield* RpcClient.make(clientGroup)
|
||||
|
||||
const received = yield* Ref.make<Array<number>>([])
|
||||
|
||||
const tickerFiber = yield* client.Ticker().pipe(
|
||||
Stream.runForEach((value) => Ref.update(received, (xs) => [...xs, value])),
|
||||
Effect.forkChild
|
||||
)
|
||||
|
||||
yield* Effect.retry(
|
||||
Effect.flatMap(
|
||||
Ref.get(received),
|
||||
(xs) => xs.length >= 2 ? Effect.void : Effect.fail("not enough ticks yet")
|
||||
),
|
||||
{ schedule: Schedule.spaced("50 millis"), times: 200 }
|
||||
)
|
||||
|
||||
const ticksBeforeGhost = (yield* Ref.get(received)).length
|
||||
assert.isAtLeast(ticksBeforeGhost, 2)
|
||||
|
||||
const ghostExit = yield* client.Ghost({ value: "boo" }).pipe(Effect.exit)
|
||||
assert.isTrue(ghostExit._tag === "Failure", "Ghost call should fail with the routing miss")
|
||||
|
||||
yield* Effect.sleep("300 millis")
|
||||
|
||||
const ticksAfterGhost = (yield* Ref.get(received)).length
|
||||
const tickerStatus = tickerFiber.pollUnsafe()
|
||||
|
||||
yield* Fiber.interrupt(tickerFiber)
|
||||
|
||||
assert.isUndefined(tickerStatus, "Ticker stream must still be running after the unknown-tag failure")
|
||||
assert.isAbove(
|
||||
ticksAfterGhost,
|
||||
ticksBeforeGhost,
|
||||
"Ticker stream must keep emitting after the unknown-tag failure"
|
||||
)
|
||||
}).pipe(Effect.provide(IsolationClient.pipe(Layer.provideMerge(IsolationServer)))),
|
||||
{ timeout: 30_000 }
|
||||
)
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user