File tree Expand file tree Collapse file tree 3 files changed +16
-11
lines changed Expand file tree Collapse file tree 3 files changed +16
-11
lines changed Original file line number Diff line number Diff line change 9
9
import sys
10
10
import tempfile
11
11
from pathlib import Path
12
+ from typing import Dict
12
13
13
14
from test_framework .authproxy import AuthServiceProxy
14
15
from test_framework .p2p import NetworkThread
@@ -80,6 +81,9 @@ def setup(self):
80
81
ch .setFormatter (formatter )
81
82
self .log .addHandler (ch )
82
83
84
+ # Keep a separate index of tanks by pod name
85
+ self .tanks : Dict [str , TestNode ] = {}
86
+
83
87
for i , tank in enumerate (WARNET ):
84
88
self .log .info (
85
89
f"Adding TestNode #{ i } from pod { tank ['tank' ]} with IP { tank ['rpc_host' ]} "
@@ -106,6 +110,7 @@ def setup(self):
106
110
node .rpc_connected = True
107
111
node .init_peers = tank ["init_peers" ]
108
112
self .nodes .append (node )
113
+ self .tanks [tank ["tank" ]] = node
109
114
110
115
self .num_nodes = len (self .nodes )
111
116
Original file line number Diff line number Diff line change @@ -56,16 +56,16 @@ def run_test(self):
56
56
self .connect_nodes (6 , 7 )
57
57
self .sync_all ()
58
58
59
- zero_peers = self .nodes [ 0 ].getpeerinfo ()
60
- one_peers = self .nodes [ 1 ].getpeerinfo ()
61
- two_peers = self .nodes [ 2 ].getpeerinfo ()
62
- three_peers = self .nodes [ 3 ].getpeerinfo ()
63
- four_peers = self .nodes [ 4 ].getpeerinfo ()
64
- five_peers = self .nodes [ 5 ].getpeerinfo ()
65
- six_peers = self .nodes [ 6 ].getpeerinfo ()
66
- seven_peers = self .nodes [ 7 ].getpeerinfo ()
67
- eight_peers = self .nodes [ 8 ].getpeerinfo ()
68
- nine_peers = self .nodes [ 9 ].getpeerinfo ()
59
+ zero_peers = self .tanks [ "tank-0000" ].getpeerinfo ()
60
+ one_peers = self .tanks [ "tank-0001" ].getpeerinfo ()
61
+ two_peers = self .tanks [ "tank-0002" ].getpeerinfo ()
62
+ three_peers = self .tanks [ "tank-0003" ].getpeerinfo ()
63
+ four_peers = self .tanks [ "tank-0004" ].getpeerinfo ()
64
+ five_peers = self .tanks [ "tank-0005" ].getpeerinfo ()
65
+ six_peers = self .tanks [ "tank-0006" ].getpeerinfo ()
66
+ seven_peers = self .tanks [ "tank-0007" ].getpeerinfo ()
67
+ eight_peers = self .tanks [ "tank-0008" ].getpeerinfo ()
68
+ nine_peers = self .tanks [ "tank-0009" ].getpeerinfo ()
69
69
70
70
for node in self .nodes :
71
71
self .log .info (f"Node { node .index } : tank={ node .tank } ip={ node .rpchost } " )
Original file line number Diff line number Diff line change @@ -34,7 +34,7 @@ def run_test(self):
34
34
self .log .info ("Adding the p2p connection" )
35
35
36
36
p2p_block_store = self .nodes [0 ].add_p2p_connection (
37
- P2PStoreBlock (), dstaddr = self .nodes [ 0 ].rpchost , dstport = 18444
37
+ P2PStoreBlock (), dstaddr = self .tanks [ "tank-0000" ].rpchost , dstport = 18444
38
38
)
39
39
40
40
self .log .info ("test that an invalid GETDATA doesn't prevent processing of future messages" )
You can’t perform that action at this time.
0 commit comments