status-im/nim-eth
2
0
Fork 0
mirror of https://github.com/status-im/nim-eth.git synced 2025-02-17 08:27:05 +00:00
Code Issues Projects Releases Wiki Activity

replace waitFor with await

Browse Source
This commit is contained in:
kdeme 2019-06-17 21:03:28 +02:00 committed by zah
parent 21c3c3946d
commit 1cb1168aec
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/p2p/test_protocol_handlers.nim
View File

@ -53,20 +53,20 @@ p2pProtocol hah(version = 1,
suite "Testing protocol handlers": suite "Testing protocol handlers":
asyncTest "Failing disconnection handler": asyncTest "Failing disconnection handler":
let bootENode = waitFor setupBootNode() let bootENode = await setupBootNode()
var node1 = setupTestNode(abc, xyz) var node1 = setupTestNode(abc, xyz)
var node2 = setupTestNode(abc, xyz) var node2 = setupTestNode(abc, xyz)
# node2 listening and node1 not, to avoid many incoming vs outgoing # node2 listening and node1 not, to avoid many incoming vs outgoing
var node1Connected = node1.connectToNetwork(@[bootENode], false, true) var node1Connected = node1.connectToNetwork(@[bootENode], false, true)
var node2Connected = node2.connectToNetwork(@[bootENode], true, true) var node2Connected = node2.connectToNetwork(@[bootENode], true, true)
waitFor node1Connected await node1Connected
waitFor node2Connected await node2Connected
check: check:
node1.peerPool.connectedNodes.len() == 1 node1.peerPool.connectedNodes.len() == 1
node2.peerPool.connectedNodes.len() == 1 node2.peerPool.connectedNodes.len() == 1
for peer in node1.peers(): for peer in node1.peers():
waitFor peer.disconnect(SubprotocolReason, true) await peer.disconnect(SubprotocolReason, true)
check: check:
# we want to check that even though the exceptions in the disconnect # we want to check that even though the exceptions in the disconnect
# handlers, each handler still ran # handlers, each handler still ran
@ -77,7 +77,7 @@ suite "Testing protocol handlers":
var node1 = setupTestNode(hah) var node1 = setupTestNode(hah)
var node2 = setupTestNode(hah) var node2 = setupTestNode(hah)
node2.startListening() node2.startListening()
let peer = waitFor node1.rlpxConnect(newNode(initENode(node2.keys.pubKey, let peer = await node1.rlpxConnect(newNode(initENode(node2.keys.pubKey,
node2.address))) node2.address)))
check: check:
peer.isNil == true peer.isNil == true