mirror of https://github.com/status-im/nim-eth.git
61 lines
1.5 KiB
Nim
61 lines
1.5 KiB
Nim
{.used.}
|
|
|
|
import
|
|
unittest2,
|
|
chronos/unittest2/asynctests,
|
|
../../eth/common/keys,
|
|
../../eth/p2p/rlpxtransport
|
|
|
|
suite "RLPx transport":
|
|
setup:
|
|
let
|
|
rng = newRng()
|
|
keys1 = KeyPair.random(rng[])
|
|
keys2 = KeyPair.random(rng[])
|
|
server = createStreamServer(initTAddress("127.0.0.1:0"), {ReuseAddr})
|
|
|
|
teardown:
|
|
waitFor server.closeWait()
|
|
|
|
asyncTest "Connect/accept":
|
|
const msg = @[byte 0, 1, 2, 3]
|
|
proc serveClient(server: StreamServer) {.async.} =
|
|
let transp = await server.accept()
|
|
let a = await RlpxTransport.accept(rng, keys1, transp)
|
|
await a.sendMsg(msg)
|
|
await a.closeWait()
|
|
|
|
let serverFut = server.serveClient()
|
|
defer:
|
|
await serverFut.wait(1.seconds)
|
|
|
|
let client =
|
|
await RlpxTransport.connect(rng, keys2, server.localAddress(), keys1.pubkey)
|
|
|
|
defer:
|
|
await client.closeWait()
|
|
let rmsg = await client.recvMsg().wait(1.seconds)
|
|
|
|
check:
|
|
msg == rmsg
|
|
|
|
await serverFut
|
|
|
|
asyncTest "Detect invalid pubkey":
|
|
proc serveClient(server: StreamServer) {.async.} =
|
|
let transp = await server.accept()
|
|
discard await RlpxTransport.accept(rng, keys1, transp)
|
|
raiseAssert "should fail to accept due to pubkey error"
|
|
|
|
let serverFut = server.serveClient()
|
|
defer:
|
|
expect(RlpxTransportError):
|
|
await serverFut.wait(1.seconds)
|
|
|
|
let keys3 = KeyPair.random(rng[])
|
|
|
|
# accept side should close connections
|
|
expect(TransportError):
|
|
discard
|
|
await RlpxTransport.connect(rng, keys2, server.localAddress(), keys3.pubkey)
|