mirror of https://github.com/status-im/nim-eth.git
106 lines
2.9 KiB
Nim
106 lines
2.9 KiB
Nim
#
|
|
# Ethereum P2P
|
|
# (c) Copyright 2018
|
|
# Status Research & Development GmbH
|
|
#
|
|
# Licensed under either of
|
|
# Apache License, version 2.0, (LICENSE-APACHEv2)
|
|
# MIT license (LICENSE-MIT)
|
|
|
|
{.used.}
|
|
|
|
import
|
|
std/tables,
|
|
chronos, testutils/unittests,
|
|
../../eth/p2p,
|
|
../stubloglevel,
|
|
./p2p_test_helper
|
|
|
|
type
|
|
network = ref object of RootRef
|
|
count*: int
|
|
|
|
PeerState = ref object of RootRef
|
|
status*: string
|
|
|
|
p2pProtocol abc(version = 1,
|
|
rlpxName = "abc",
|
|
networkState = network):
|
|
|
|
onPeerConnected do (peer: Peer):
|
|
peer.networkState.count += 1
|
|
|
|
onPeerDisconnected do (peer: Peer, reason: DisconnectionReason) {.gcsafe.}:
|
|
peer.networkState.count -= 1
|
|
if true:
|
|
raise newException(CatchableError, "Fake abc exception")
|
|
|
|
p2pProtocol xyz(version = 1,
|
|
rlpxName = "xyz",
|
|
networkState = network,
|
|
peerState = PeerState):
|
|
|
|
onPeerConnected do (peer: Peer):
|
|
peer.networkState.count += 1
|
|
peer.state.status = "connected"
|
|
|
|
onPeerDisconnected do (peer: Peer, reason: DisconnectionReason) {.gcsafe.}:
|
|
peer.networkState.count -= 1
|
|
if true:
|
|
raise newException(CatchableError, "Fake xyz exception")
|
|
peer.state.status = "disconnected"
|
|
|
|
p2pProtocol hah(version = 1,
|
|
rlpxName = "hah",
|
|
networkState = network):
|
|
|
|
onPeerConnected do (peer: Peer):
|
|
if true:
|
|
raise newException(UselessPeerError, "Fake hah exception")
|
|
peer.networkState.count += 1
|
|
|
|
onPeerDisconnected do (peer: Peer, reason: DisconnectionReason) {.gcsafe.}:
|
|
peer.networkState.count -= 1
|
|
|
|
|
|
suite "Testing protocol handlers":
|
|
asyncTest "Failing disconnection handler":
|
|
let rng = newRng()
|
|
|
|
var node1 = setupTestNode(rng, abc, xyz)
|
|
var node2 = setupTestNode(rng, abc, xyz)
|
|
|
|
node2.startListening()
|
|
let res = await node1.rlpxConnect(newNode(node2.toENode()))
|
|
check res.isOk()
|
|
let peer = res.get()
|
|
check peer.state(xyz).status == "connected"
|
|
|
|
await peer.disconnect(SubprotocolReason, true)
|
|
check:
|
|
# we want to check that even though the exceptions in the disconnect
|
|
# handlers, each handler still ran
|
|
node1.protocolState(abc).count == 0
|
|
node1.protocolState(xyz).count == 0
|
|
peer.state(xyz).status == "connected"
|
|
|
|
asyncTest "Failing connection handler":
|
|
let rng = newRng()
|
|
|
|
var node1 = setupTestNode(rng, hah)
|
|
var node2 = setupTestNode(rng, hah)
|
|
node2.startListening()
|
|
let res = await node1.rlpxConnect(newNode(node2.toENode()))
|
|
check:
|
|
res.isErr()
|
|
# To check if the disconnection handler did not run
|
|
node1.protocolState(hah).count == 0
|
|
|
|
test "Override network state":
|
|
let rng = newRng()
|
|
var node = setupTestNode(rng, hah)
|
|
node.addCapability(hah, network(count: 3))
|
|
check node.protocolState(hah).count == 3
|
|
node.replaceNetworkState(hah, network(count: 7))
|
|
check node.protocolState(hah).count == 7
|