[tests] disconnect_ban: remove dependency on urllib
Github-Pull: #10143
Rebased-From: 5cc3ee24d2
This commit is contained in:
parent
98bd0c338b
commit
04226938a3
|
@ -6,14 +6,12 @@
|
|||
#
|
||||
# Test node handling
|
||||
#
|
||||
import urllib.parse
|
||||
|
||||
from test_framework.mininode import wait_until
|
||||
from test_framework.test_framework import BitcoinTestFramework
|
||||
from test_framework.util import (assert_equal,
|
||||
assert_raises_jsonrpc,
|
||||
connect_nodes_bi,
|
||||
p2p_port,
|
||||
start_node,
|
||||
stop_node,
|
||||
)
|
||||
|
@ -79,18 +77,13 @@ class NodeHandlingTest(BitcoinTestFramework):
|
|||
###########################
|
||||
# RPC disconnectnode test #
|
||||
###########################
|
||||
url = urllib.parse.urlparse(self.nodes[1].url)
|
||||
self.nodes[0].disconnectnode(url.hostname + ":" + str(p2p_port(1)))
|
||||
address1 = self.nodes[0].getpeerinfo()[0]['addr']
|
||||
self.nodes[0].disconnectnode(address=address1)
|
||||
wait_until(lambda: len(self.nodes[0].getpeerinfo()) == 1)
|
||||
for node in self.nodes[0].getpeerinfo():
|
||||
assert(node['addr'] != url.hostname + ":" + str(p2p_port(1)))
|
||||
assert not [node for node in self.nodes[0].getpeerinfo() if node['addr'] == address1]
|
||||
|
||||
connect_nodes_bi(self.nodes, 0, 1) # reconnect the node
|
||||
found = False
|
||||
for node in self.nodes[0].getpeerinfo():
|
||||
if node['addr'] == url.hostname + ":" + str(p2p_port(1)):
|
||||
found = True
|
||||
assert(found)
|
||||
assert [node for node in self.nodes[0].getpeerinfo() if node['addr'] == address1]
|
||||
|
||||
if __name__ == '__main__':
|
||||
NodeHandlingTest().main()
|
||||
|
|
Loading…
Reference in a new issue