codex-storage
/
nim-libp2p
mirror of https://github.com/codex-storage/nim-libp2p.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix write limits test by using timeout, also a proposal to use this everywhere

This commit is contained in:
Giovanni Petrantoni 2020-02-07 15:41:24 +09:00 committed by Dmitriy Ryajov
parent 136ac3f550
commit 1140f7b884
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tests/testmplex.nim
View File

@ -1,4 +1,4 @@
import unittest, sequtils, sugar, strformat, options, strformat import unittest, sequtils, sugar, strformat, options, strformat, random
import chronos, nimcrypto/utils, chronicles import chronos, nimcrypto/utils, chronicles
import ../libp2p/[connection, import ../libp2p/[connection,
stream/lpstream, stream/lpstream,
@ -119,12 +119,16 @@ suite "Mplex":
test "e2e - read/write receiver": test "e2e - read/write receiver":
proc testNewStream(): Future[bool] {.async.} = proc testNewStream(): Future[bool] {.async.} =
let ma: MultiAddress = Multiaddress.init("/ip4/0.0.0.0/tcp/0") let
ma: MultiAddress = Multiaddress.init("/ip4/0.0.0.0/tcp/0")
lock = newFuture[void]()
timeout = sleepAsync(5_000)
proc connHandler(conn: Connection) {.async, gcsafe.} = proc connHandler(conn: Connection) {.async, gcsafe.} =
proc handleMplexListen(stream: Connection) {.async, gcsafe.} = proc handleMplexListen(stream: Connection) {.async, gcsafe.} =
let msg = await stream.readLp() let msg = await stream.readLp()
check cast[string](msg) == "Hello from stream!" check cast[string](msg) == "Hello from stream!"
lock.complete()
await stream.close() await stream.close()
let mplexListen = newMplex(conn) let mplexListen = newMplex(conn)
@ -172,6 +176,8 @@ suite "Mplex":
let stream = await mplexDial.newStream("", true) let stream = await mplexDial.newStream("", true)
let openState = cast[LPChannel](stream.stream).isOpen let openState = cast[LPChannel](stream.stream).isOpen
await stream.writeLp("Hello from stream!") await stream.writeLp("Hello from stream!")
await lock or timeout
check lock.finished
await conn.close() await conn.close()
check not openState # assert lazy check not openState # assert lazy
result = true result = true
@ -181,11 +187,15 @@ suite "Mplex":
test "e2e - write limits": test "e2e - write limits":
proc testNewStream(): Future[bool] {.async.} = proc testNewStream(): Future[bool] {.async.} =
let ma: MultiAddress = Multiaddress.init("/ip4/0.0.0.0/tcp/0") let
ma: MultiAddress = Multiaddress.init("/ip4/0.0.0.0/tcp/0")
lock = newFuture[void]()
timeout = sleepAsync(5_000)
proc connHandler(conn: Connection) {.async, gcsafe.} = proc connHandler(conn: Connection) {.async, gcsafe.} =
proc handleMplexListen(stream: Connection) {.async, gcsafe.} = proc handleMplexListen(stream: Connection) {.async, gcsafe.} =
let msg = await stream.readLp() let msg = await stream.readLp()
lock.complete()
check cast[string](msg) == "Hello from stream!" check cast[string](msg) == "Hello from stream!"
await stream.close() await stream.close()
@ -201,8 +211,12 @@ suite "Mplex":
let mplexDial = newMplex(conn) let mplexDial = newMplex(conn)
let stream = await mplexDial.newStream() let stream = await mplexDial.newStream()
let bigseq = newSeq[uint8](MaxMsgSize + 1) var bigseq = newSeqOfCap[uint8](MaxMsgSize + 1)
for _ in 0..<MaxMsgSize:
bigseq.add(uint8(rand(int('A')..int('z'))))
await stream.writeLp(bigseq) await stream.writeLp(bigseq)
await lock or timeout
check timeout.finished # this test has to timeout!
await conn.close() await conn.close()
result = true result = true