|
4 | 4 | # file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
5 | 5 | """Test invalid p2p messages for nodes with bloom filters disabled.
|
6 | 6 |
|
7 |
| -Test that, when bloom filters are not enabled, nodes are disconnected if: |
| 7 | +Test that, when bloom filters are not enabled, peers are disconnected if: |
8 | 8 | 1. They send a p2p mempool message
|
9 | 9 | 2. They send a p2p filterload message
|
10 | 10 | 3. They send a p2p filteradd message
|
|
17 | 17 | from test_framework.util import assert_equal
|
18 | 18 |
|
19 | 19 |
|
20 |
| -class P2PNobloomfilterMessages(BitcoinTestFramework): |
| 20 | +class P2PNoBloomFilterMessages(BitcoinTestFramework): |
21 | 21 | def set_test_params(self):
|
22 | 22 | self.setup_clean_chain = True
|
23 | 23 | self.num_nodes = 1
|
24 | 24 | self.extra_args = [["-peerbloomfilters=0"]]
|
25 | 25 |
|
26 | 26 | def test_message_causes_disconnect(self, message):
|
27 |
| - # Add a p2p connection that sends a message and check that it disconnects |
| 27 | + """Add a p2p connection that sends a message and check that it disconnects.""" |
28 | 28 | peer = self.nodes[0].add_p2p_connection(P2PInterface())
|
29 | 29 | peer.send_message(message)
|
30 | 30 | peer.wait_for_disconnect()
|
31 |
| - assert_equal(len(self.nodes[0].getpeerinfo()), 0) |
| 31 | + assert_equal(self.nodes[0].getconnectioncount(), 0) |
32 | 32 |
|
33 | 33 | def run_test(self):
|
34 |
| - self.log.info("Test that node is disconnected if it sends mempool message") |
| 34 | + self.log.info("Test that peer is disconnected if it sends mempool message") |
35 | 35 | self.test_message_causes_disconnect(msg_mempool())
|
36 | 36 |
|
37 |
| - self.log.info("Test that node is disconnected if it sends filterload message") |
| 37 | + self.log.info("Test that peer is disconnected if it sends filterload message") |
38 | 38 | self.test_message_causes_disconnect(msg_filterload())
|
39 | 39 |
|
40 |
| - self.log.info("Test that node is disconnected if it sends filteradd message") |
| 40 | + self.log.info("Test that peer is disconnected if it sends filteradd message") |
41 | 41 | self.test_message_causes_disconnect(msg_filteradd(data=b'\xcc'))
|
42 | 42 |
|
43 | 43 | self.log.info("Test that peer is disconnected if it sends a filterclear message")
|
44 | 44 | self.test_message_causes_disconnect(msg_filterclear())
|
45 | 45 |
|
| 46 | + |
46 | 47 | if __name__ == '__main__':
|
47 |
| - P2PNobloomfilterMessages().main() |
| 48 | + P2PNoBloomFilterMessages().main() |
0 commit comments