mirror of
https://github.com/logos-storage/nim-json-rpc.git
synced 2026-01-03 14:13:08 +00:00
88 lines
2.7 KiB
Nim
88 lines
2.7 KiB
Nim
# json-rpc
|
|
# Copyright (c) 2019-2023 Status Research & Development GmbH
|
|
# Licensed under either of
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE))
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT))
|
|
# at your option.
|
|
# This file may not be copied, modified, or distributed except according to
|
|
# those terms.
|
|
|
|
import
|
|
unittest2,
|
|
../json_rpc/[rpcserver, rpcclient, jsonmarshal]
|
|
|
|
const TestsCount = 100
|
|
|
|
proc simpleTest(address: string): Future[bool] {.async.} =
|
|
var client = newRpcHttpClient()
|
|
await client.connect("http://" & address)
|
|
var r = await client.call("noParamsProc", %[])
|
|
if r.string == "\"Hello world\"":
|
|
result = true
|
|
|
|
proc continuousTest(address: string): Future[int] {.async.} =
|
|
var client = newRpcHttpClient()
|
|
result = 0
|
|
for i in 0..<TestsCount:
|
|
await client.connect("http://" & address)
|
|
var r = await client.call("myProc", %[%"abc", %[1, 2, 3, i]])
|
|
if r.string == "\"Hello abc data: [1, 2, 3, " & $i & "]\"":
|
|
result += 1
|
|
await client.close()
|
|
|
|
proc invalidTest(address: string): Future[bool] {.async.} =
|
|
var client = newRpcHttpClient()
|
|
await client.connect("http://" & address)
|
|
var invalidA, invalidB: bool
|
|
try:
|
|
var r = await client.call("invalidProcA", %[])
|
|
discard r
|
|
except JsonRpcError:
|
|
invalidA = true
|
|
try:
|
|
var r = await client.call("invalidProcB", %[1, 2, 3])
|
|
discard r
|
|
except JsonRpcError:
|
|
invalidB = true
|
|
if invalidA and invalidB:
|
|
result = true
|
|
|
|
const bigChunkSize = 4 * 8192
|
|
|
|
proc chunkedTest(address: string): Future[bool] {.async.} =
|
|
var client = newRpcHttpClient()
|
|
await client.connect("http://" & address)
|
|
let r = await client.call("bigchunkMethod", %[])
|
|
let data = JrpcConv.decode(r.string, seq[byte])
|
|
return data.len == bigChunkSize
|
|
|
|
var httpsrv = newRpcHttpServer(["127.0.0.1:0"])
|
|
|
|
# Create RPC on server
|
|
httpsrv.rpc("myProc") do(input: string, data: array[0..3, int]):
|
|
result = %("Hello " & input & " data: " & $data)
|
|
httpsrv.rpc("noParamsProc") do():
|
|
result = %("Hello world")
|
|
|
|
httpsrv.rpc("bigchunkMethod") do() -> seq[byte]:
|
|
result = newSeq[byte](bigChunkSize)
|
|
for i in 0..<result.len:
|
|
result[i] = byte(i mod 255)
|
|
|
|
httpsrv.setMaxChunkSize(8192)
|
|
httpsrv.start()
|
|
let serverAddress = $httpsrv.localAddress()[0]
|
|
|
|
suite "JSON-RPC test suite":
|
|
test "Simple RPC call":
|
|
check waitFor(simpleTest(serverAddress)) == true
|
|
test "Continuous RPC calls (" & $TestsCount & " messages)":
|
|
check waitFor(continuousTest(serverAddress)) == TestsCount
|
|
test "Invalid RPC calls":
|
|
check waitFor(invalidTest(serverAddress)) == true
|
|
test "Http client can handle chunked transfer encoding":
|
|
check waitFor(chunkedTest(serverAddress)) == true
|
|
|
|
waitFor httpsrv.stop()
|
|
waitFor httpsrv.closeWait()
|