Skip to content

Commit 9079bfe

Browse files
committed
ln_init: dont assume node[0] is the miner
1 parent 93ea530 commit 9079bfe

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

resources/scenarios/ln_init.py

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,12 @@ def set_test_params(self):
2626
def add_options(self, parser):
2727
parser.description = "Fund LN wallets and open channels"
2828
parser.usage = "warnet run /path/to/ln_init.py"
29+
parser.add_argument(
30+
"--miner",
31+
dest="miner",
32+
type=str,
33+
help="Select one tank by name as the blockchain miner",
34+
)
2935

3036
def run_test(self):
3137
##
@@ -38,7 +44,18 @@ def run_test(self):
3844
# MINER
3945
##
4046
self.log.info("Setting up miner...")
41-
miner = self.ensure_miner(self.nodes[0])
47+
if self.options.miner:
48+
self.log.info(f"Parsed 'miner' argument: {self.options.miner}")
49+
mining_tank = self.tanks[self.options.miner]
50+
elif "miner" in self.tanks:
51+
# or choose the tank with the right name
52+
self.log.info("Found tank named 'miner'")
53+
mining_tank = self.tanks["miner"]
54+
else:
55+
mining_tank = self.nodes[0]
56+
self.log.info(f"Using tank {mining_tank.tank} as miner")
57+
58+
miner = self.ensure_miner(mining_tank)
4259
miner_addr = miner.getnewaddress()
4360

4461
def gen(n):

0 commit comments

Comments
 (0)