fix tests

This commit is contained in:
Dmitriy Ryajov 2020-03-27 10:02:14 -06:00
parent 3effb95f10
commit a92d616a3e
1 changed files with 5 additions and 3 deletions

View File

@ -59,9 +59,9 @@ proc readLp*(s: StreamTransport): Future[seq[byte]] {.async, gcsafe.} =
result.setLen(size) result.setLen(size)
if size > 0.uint: if size > 0.uint:
await s.readExactly(addr result[0], int(size)) await s.readExactly(addr result[0], int(size))
except LPStreamIncompleteError as exc: except TransportIncompleteError as exc:
trace "remote connection ended unexpectedly", exc = exc.msg trace "remote connection ended unexpectedly", exc = exc.msg
except LPStreamReadError as exc: except TransportError as exc:
trace "unable to read from remote connection", exc = exc.msg trace "unable to read from remote connection", exc = exc.msg
proc createNode*(privKey: Option[PrivateKey] = none(PrivateKey), proc createNode*(privKey: Option[PrivateKey] = none(PrivateKey),
@ -382,10 +382,12 @@ suite "Interop":
await daemonNode.connect(nativePeer.peerId, nativePeer.addrs) await daemonNode.connect(nativePeer.peerId, nativePeer.addrs)
var stream = await daemonNode.openStream(nativePeer.peerId, protos) var stream = await daemonNode.openStream(nativePeer.peerId, protos)
while count < 10: var count2 = 0
while count2 < 10:
discard await stream.transp.writeLp(test) discard await stream.transp.writeLp(test)
let line = await stream.transp.readLp() let line = await stream.transp.readLp()
check test == cast[string](line) check test == cast[string](line)
inc(count2)
result = 10 == (await wait(testFuture, 10.secs)) result = 10 == (await wait(testFuture, 10.secs))
await nativeNode.stop() await nativeNode.stop()