mirror of
https://github.com/logos-storage/nim-ethers.git
synced 2026-01-04 22:53:08 +00:00
933103ca6d
Ensure that all errors bubble to the main `convertError` proc. Add websockets mocks and tests. Formatting updates via nph.
76 lines
2.4 KiB
Nim
76 lines
2.4 KiB
Nim
import std/tables
|
|
import pkg/serde
|
|
import pkg/chronos/apps/http/httpclient
|
|
import pkg/chronos/apps/http/httpserver
|
|
import pkg/stew/byteutils
|
|
import pkg/questionable
|
|
|
|
export httpserver
|
|
|
|
{.push raises: [].}
|
|
|
|
type
|
|
RpcResponse* = proc(request: HttpRequestRef): Future[HttpResponseRef] {.async: (raises: [CancelledError]).}
|
|
|
|
MockHttpServer* = object
|
|
server: HttpServerRef
|
|
rpcResponses: ref Table[string, RpcResponse]
|
|
|
|
RequestRx {.deserialize.} = object
|
|
jsonrpc*: string
|
|
id* : int
|
|
`method`* : string
|
|
|
|
|
|
proc init*(_: type MockHttpServer, address: TransportAddress): MockHttpServer =
|
|
var server: MockHttpServer
|
|
|
|
proc processRequest(r: RequestFence): Future[HttpResponseRef] {.async: (raises: [CancelledError]).} =
|
|
if r.isErr:
|
|
return defaultResponse()
|
|
|
|
let request = r.get()
|
|
try:
|
|
let body = string.fromBytes(await request.getBody())
|
|
without req =? RequestRx.fromJson(body), error:
|
|
return await request.respond(Http400, "Invalid request, must be valid json rpc request")
|
|
|
|
if not server.rpcResponses.contains(req.method):
|
|
return await request.respond(Http404, "Method not registered")
|
|
|
|
try:
|
|
let rpcResponseProc = server.rpcResponses[req.method]
|
|
return await rpcResponseProc(request)
|
|
except KeyError as e:
|
|
return await request.respond(Http500, "Method lookup error with key, error: " & e.msg)
|
|
|
|
except HttpProtocolError as e:
|
|
return defaultResponse(e)
|
|
except HttpTransportError as e:
|
|
return defaultResponse(e)
|
|
except HttpWriteError as exc:
|
|
return defaultResponse(exc)
|
|
|
|
let
|
|
socketFlags = {ServerFlags.TcpNoDelay, ServerFlags.ReuseAddr}
|
|
serverFlags = {HttpServerFlags.Http11Pipeline}
|
|
res = HttpServerRef.new(address, processRequest,
|
|
socketFlags = socketFlags,
|
|
serverFlags = serverFlags)
|
|
server = MockHttpServer(server: res.get(), rpcResponses: newTable[string, RpcResponse]())
|
|
return server
|
|
|
|
template registerRpcResponse*(server: MockHttpServer, `method`: string, rpcResponse: untyped): untyped =
|
|
server.rpcResponses[`method`] = rpcResponse
|
|
|
|
proc address*(server: MockHttpServer): TransportAddress =
|
|
server.server.instance.localAddress()
|
|
|
|
proc start*(server: MockHttpServer) =
|
|
server.server.start()
|
|
|
|
proc stop*(server: MockHttpServer) {.async: (raises: []).} =
|
|
await server.server.stop()
|
|
await server.server.closeWait()
|
|
|