[test] Add tests for getconnectioncount, getnettotals and ping
* Split run_test into 4 separate tests * Add 2 tests, getconnectioncount and getnettotals * getnettotals - Strategy of test is to get the network stats before and after a ping. The difference in bytes sent/received is the bytes needed for a ping/pong.
This commit is contained in:
parent
4c924011f5
commit
85f950a5c4
1 changed files with 34 additions and 1 deletions
|
@ -32,6 +32,39 @@ class NetTest(BitcoinTestFramework):
|
|||
self.sync_all()
|
||||
|
||||
def run_test(self):
|
||||
self._test_connection_count()
|
||||
self._test_getnettotals()
|
||||
self._test_getnetworkinginfo()
|
||||
self._test_getaddednodeinfo()
|
||||
|
||||
def _test_connection_count(self):
|
||||
# connect_nodes_bi connects each node to the other
|
||||
assert_equal(self.nodes[0].getconnectioncount(), 2)
|
||||
|
||||
def _test_getnettotals(self):
|
||||
# check that getnettotals totalbytesrecv and totalbytessent
|
||||
# are consistent with getpeerinfo
|
||||
peer_info = self.nodes[0].getpeerinfo()
|
||||
assert_equal(len(peer_info), 2)
|
||||
net_totals = self.nodes[0].getnettotals()
|
||||
assert_equal(sum([peer['bytesrecv'] for peer in peer_info]),
|
||||
net_totals['totalbytesrecv'])
|
||||
assert_equal(sum([peer['bytessent'] for peer in peer_info]),
|
||||
net_totals['totalbytessent'])
|
||||
# test getnettotals and getpeerinfo by doing a ping
|
||||
# the bytes sent/received should change
|
||||
# note ping and pong are 32 bytes each
|
||||
self.nodes[0].ping()
|
||||
time.sleep(0.1)
|
||||
peer_info_after_ping = self.nodes[0].getpeerinfo()
|
||||
net_totals_after_ping = self.nodes[0].getnettotals()
|
||||
for before, after in zip(peer_info, peer_info_after_ping):
|
||||
assert_equal(before['bytesrecv_per_msg']['pong'] + 32, after['bytesrecv_per_msg']['pong'])
|
||||
assert_equal(before['bytessent_per_msg']['ping'] + 32, after['bytessent_per_msg']['ping'])
|
||||
assert_equal(net_totals['totalbytesrecv'] + 32*2, net_totals_after_ping['totalbytesrecv'])
|
||||
assert_equal(net_totals['totalbytessent'] + 32*2, net_totals_after_ping['totalbytessent'])
|
||||
|
||||
def _test_getnetworkinginfo(self):
|
||||
assert_equal(self.nodes[0].getnetworkinfo()['networkactive'], True)
|
||||
assert_equal(self.nodes[0].getnetworkinfo()['connections'], 2)
|
||||
|
||||
|
@ -49,7 +82,7 @@ class NetTest(BitcoinTestFramework):
|
|||
assert_equal(self.nodes[0].getnetworkinfo()['networkactive'], True)
|
||||
assert_equal(self.nodes[0].getnetworkinfo()['connections'], 2)
|
||||
|
||||
# test getaddednodeinfo
|
||||
def _test_getaddednodeinfo(self):
|
||||
assert_equal(self.nodes[0].getaddednodeinfo(), [])
|
||||
# add a node (node2) to node0
|
||||
ip_port = "127.0.0.1:{}".format(p2p_port(2))
|
||||
|
|
Loading…
Reference in a new issue