diff options
| author | Martin Zumsande <mzumsande@gmail.com> | 2024-02-02 11:59:05 -0500 |
|---|---|---|
| committer | Martin Zumsande <mzumsande@gmail.com> | 2024-02-02 13:24:29 -0500 |
| commit | cc87ee4c3934028e78a59de509951ff7226ec80d (patch) | |
| tree | 472d43efb28b4602046ac7d61237ac69f630741b /test/functional/rpc_setban.py | |
| parent | 5b8c5970bdfc817cac9b59f699925c4426c59b61 (diff) | |
| download | bitcoin-cc87ee4c3934028e78a59de509951ff7226ec80d.tar.xz | |
test: fix intermittent failure in rpc_setban.py --v2transport
When initiating a v2 connection and being immediately disconnected,
a node cannot know if the disconnect happens because the peer only
supports v1, or because it has banned you, so it schedules to reconnect with v1.
If the test doesn't wait for that, the reconnect can happen at a bad time,
resulting in failure in a later connect_nodes call.
Also add the test with --v2transport to the test runner.
Diffstat (limited to 'test/functional/rpc_setban.py')
| -rwxr-xr-x | test/functional/rpc_setban.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/test/functional/rpc_setban.py b/test/functional/rpc_setban.py index b4f3d77e5b..bc426d7371 100755 --- a/test/functional/rpc_setban.py +++ b/test/functional/rpc_setban.py @@ -4,6 +4,7 @@ # file COPYING or http://www.opensource.org/licenses/mit-license.php. """Test the setban rpc call.""" +from contextlib import ExitStack from test_framework.test_framework import BitcoinTestFramework from test_framework.util import ( p2p_port, @@ -29,7 +30,13 @@ class SetBanTests(BitcoinTestFramework): self.nodes[1].setban("127.0.0.1", "add") # Node 0 should not be able to reconnect - with self.nodes[1].assert_debug_log(expected_msgs=['dropped (banned)\n'], timeout=50): + context = ExitStack() + context.enter_context(self.nodes[1].assert_debug_log(expected_msgs=['dropped (banned)\n'], timeout=50)) + # When disconnected right after connecting, a v2 node will attempt to reconnect with v1. + # Wait for that to happen so that it cannot mess with later tests. + if self.options.v2transport: + context.enter_context(self.nodes[0].assert_debug_log(expected_msgs=['trying v1 connection'], timeout=50)) + with context: self.restart_node(1, []) self.nodes[0].addnode("127.0.0.1:" + str(p2p_port(1)), "onetry") |