Changeset View
Changeset View
Standalone View
Standalone View
test/functional/p2p_nobloomfilter_messages.py
| Show All 30 Lines | class P2PNoBloomFilterMessages(BitcoinTestFramework): | ||||
| 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(self.nodes[0].getconnectioncount(), 0) | assert_equal(self.nodes[0].getconnectioncount(), 0) | ||||
| def run_test(self): | def run_test(self): | ||||
| self.log.info( | self.log.info("Test that peer is disconnected if it sends mempool message") | ||||
| "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( | self.log.info("Test that peer is disconnected if it sends filterload message") | ||||
| "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( | self.log.info("Test that peer is disconnected if it sends filteradd message") | ||||
| "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( | self.log.info( | ||||
| "Test that peer is disconnected if it sends a filterclear message") | "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() | ||||