def test_disconnect_with_failure(self, client0: RpcClient):
n = self.connect_nodes(client0, 1)
assert client0.disconnect_peer(self.nodes[1].key,
client0.UPDATE_NODE_OP_FAILURE)
# Node 1 is still in trusted node table, only marked as failure.
# But it may be auto connected again (by timer).
node = client0.get_node(self.nodes[1].key)
assert node[0] == "trusted"
if node[1]["lastContact"].get("failure"):
# Node 1 marked as failure
assert node[1]["lastConnected"].get("failure")
assert node[1]["streamToken"] == n[1]["streamToken"]
else:
# Node 1 auto connected by timer, so timestamp changed
assert self.compare_node_time(
node[1]["lastConnected"]["success"],
n[1]["lastConnected"]["success"]) == 1
assert self.compare_node_time(node[1]["lastContact"]["success"],
n[1]["lastContact"]["success"]) == 1
# Node 0 still create outgoing connection to Node 1
time.sleep((self.test_house_keeping_ms + 100) / 1000)
assert client0.get_peer(self.nodes[1].key) is not None
def test_disconnect_with_demote(self, client0: RpcClient):
n = self.connect_nodes(client0, 2)
assert client0.disconnect_peer(self.nodes[2].key, client0.UPDATE_NODE_OP_DEMOTE)
# demote to untrusted node table
node = client0.get_node(self.nodes[2].key)
assert node[0] == "untrusted"
assert node[1]["lastConnected"].get("failure")
assert node[1]["lastContact"].get("demoted")
assert node[1]["streamToken"] == n[1]["streamToken"]
# Node 0 will not create outgoing connection to Node 2
time.sleep((self.test_house_keeping_ms + 100) / 1000)
assert client0.get_peer(self.nodes[2].key) is None
def test_disconnect_with_remove(self, client0: RpcClient):
n = self.connect_nodes(client0, 3)
assert client0.disconnect_peer(self.nodes[3].key, client0.UPDATE_NODE_OP_REMOVE)
# On node 0: node 3 is blacklisted, and cannot immediately add it again
assert client0.get_node(self.nodes[3].key) is None
self.nodes[0].addnode(self.nodes[3].key, get_peer_addr(self.nodes[3]))
assert client0.get_node(self.nodes[3].key) is None
# On node 3: add node 0 as trusted node, so that try to create
# outgoing connection to node 0.
client3 = RpcClient(self.nodes[3])
self.nodes[3].addnode(self.nodes[0].key, get_peer_addr(self.nodes[0]))
node0 = client3.get_node(self.nodes[0].key)
assert node0[0] == "trusted"
# Node 3 create more outgoing connection, but it's blacklisted in node 0.
time.sleep((self.test_house_keeping_ms + 100) / 1000)
peer0 = client3.get_peer(self.nodes[0].key)
# refused during handshake or not handshaked yet
assert peer0 is None or len(peer0["protocols"]) == 0
