[test] logging and style followups for bloomfilter tests

-Use peer to refer to mininodes instead of node
because they are not bitcoind nodes.
-Use log.debug for logs that give helpful but
not super necessary information.
-Adhere to style guidelines (newlines, capitalization).
This commit is contained in:
gzhao408 committed 2020-06-16 08:23:12 -07:00
1 parent 4b30c41b4e
commit e81942d2e1
2 files changed
+12 -11

No files matched your search

+3 -3
View File
@@ -124,11 +124,11 @@ class FilterTest(BitcoinTestFramework):
self.log.info("Check that a node with bloom filters enabled services p2p mempool messages") self.log.info("Check that a node with bloom filters enabled services p2p mempool messages")
filter_peer = P2PBloomFilter() filter_peer = P2PBloomFilter()
self.log.info("Create a tx relevant to the peer before connecting") self.log.debug("Create a tx relevant to the peer before connecting")
filter_address = self.nodes[0].decodescript(filter_peer.watch_script_pubkey)['addresses'][0] filter_address = self.nodes[0].decodescript(filter_peer.watch_script_pubkey)['addresses'][0]
txid = self.nodes[0].sendtoaddress(filter_address, 90) txid = self.nodes[0].sendtoaddress(filter_address, 90)
self.log.info("Send a mempool msg after connecting and check that the tx is received") self.log.debug("Send a mempool msg after connecting and check that the tx is received")
self.nodes[0].add_p2p_connection(filter_peer) self.nodes[0].add_p2p_connection(filter_peer)
filter_peer.send_and_ping(filter_peer.watch_filter_init) filter_peer.send_and_ping(filter_peer.watch_filter_init)
self.nodes[0].p2p.send_message(msg_mempool()) self.nodes[0].p2p.send_message(msg_mempool())
@@ -227,8 +227,8 @@ class FilterTest(BitcoinTestFramework):
self.test_frelay_false(filter_peer_without_nrelay) self.test_frelay_false(filter_peer_without_nrelay)
self.test_filter(filter_peer_without_nrelay) self.test_filter(filter_peer_without_nrelay)
self.log.info('Test msg_mempool')
self.test_msg_mempool() self.test_msg_mempool()
if __name__ == '__main__': if __name__ == '__main__':
FilterTest().main() FilterTest().main()
@@ -4,7 +4,7 @@
# file COPYING or http://www.opensource.org/licenses/mit-license.php. # file COPYING or http://www.opensource.org/licenses/mit-license.php.
"""Test invalid p2p messages for nodes with bloom filters disabled. """Test invalid p2p messages for nodes with bloom filters disabled.
Test that, when bloom filters are not enabled, nodes are disconnected if: Test that, when bloom filters are not enabled, peers are disconnected if:
1. They send a p2p mempool message 1. They send a p2p mempool message
2. They send a p2p filterload message 2. They send a p2p filterload message
3. They send a p2p filteradd message 3. They send a p2p filteradd message
@@ -17,31 +17,32 @@ from test_framework.test_framework import BitcoinTestFramework
from test_framework.util import assert_equal from test_framework.util import assert_equal
class P2PNobloomfilterMessages(BitcoinTestFramework): class P2PNoBloomFilterMessages(BitcoinTestFramework):
def set_test_params(self): def set_test_params(self):
self.setup_clean_chain = True self.setup_clean_chain = True
self.num_nodes = 1 self.num_nodes = 1
self.extra_args = [["-peerbloomfilters=0"]] self.extra_args = [["-peerbloomfilters=0"]]
def test_message_causes_disconnect(self, message): def test_message_causes_disconnect(self, message):
# Add a p2p connection that sends a message and check that it disconnects """Add a p2p connection that sends a message and check that it disconnects."""
peer = self.nodes[0].add_p2p_connection(P2PInterface()) peer = self.nodes[0].add_p2p_connection(P2PInterface())
peer.send_message(message) peer.send_message(message)
peer.wait_for_disconnect() peer.wait_for_disconnect()
assert_equal(len(self.nodes[0].getpeerinfo()), 0) assert_equal(self.nodes[0].getconnectioncount(), 0)
def run_test(self): def run_test(self):
self.log.info("Test that node is disconnected if it sends mempool message") self.log.info("Test that peer is disconnected if it sends mempool message")
self.test_message_causes_disconnect(msg_mempool()) self.test_message_causes_disconnect(msg_mempool())
self.log.info("Test that node is disconnected if it sends filterload message") self.log.info("Test that peer is disconnected if it sends filterload message")
self.test_message_causes_disconnect(msg_filterload()) self.test_message_causes_disconnect(msg_filterload())
self.log.info("Test that node is disconnected if it sends filteradd message") self.log.info("Test that peer is disconnected if it sends filteradd message")
self.test_message_causes_disconnect(msg_filteradd(data=b'\xcc')) self.test_message_causes_disconnect(msg_filteradd(data=b'\xcc'))
self.log.info("Test that peer is disconnected if it sends a filterclear message") self.log.info("Test that peer is disconnected if it sends a filterclear message")
self.test_message_causes_disconnect(msg_filterclear()) self.test_message_causes_disconnect(msg_filterclear())
if __name__ == '__main__': if __name__ == '__main__':
P2PNobloomfilterMessages().main() P2PNoBloomFilterMessages().main()