Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 5 additions & 2 deletions resources/scenarios/commander.py
Original file line number Diff line number Diff line change
Expand Up @@ -233,6 +233,9 @@ def setup(self):
ch.setFormatter(ColorFormatter())
self.log.addHandler(ch)

# This is the namespace the scenario was deployed in
self.namespace = NAMESPACE

# Keep a separate index of tanks by pod name
self.tanks: dict[str, TestNode] = {}
self.lns: dict[str, LNNode] = {}
Expand Down Expand Up @@ -265,10 +268,10 @@ def setup(self):
node.init_peers = int(tank["init_peers"])

self.nodes.append(node)
self.tanks[tank["tank"]] = node
self.tanks[tank["tank"] + "." + tank["namespace"]] = node

for ln in WARNET["lightning"]:
self.lns[ln.name] = ln
self.lns[ln.name + "." + ln.namespace] = ln

self.num_nodes = len(self.nodes)

Expand Down
8 changes: 6 additions & 2 deletions resources/scenarios/ln_init.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,10 +47,10 @@ def run_test(self):
if self.options.miner:
self.log.info(f"Parsed 'miner' argument: {self.options.miner}")
mining_tank = self.tanks[self.options.miner]
elif "miner" in self.tanks:
elif f"miner.{self.namespace}" in self.tanks:
# or choose the tank with the right name
self.log.info("Found tank named 'miner'")
mining_tank = self.tanks["miner"]
mining_tank = self.tanks[f"miner.{self.namespace}"]
else:
mining_tank = self.nodes[0]
self.log.info(f"Using tank {mining_tank.tank} as miner")
Expand Down Expand Up @@ -211,6 +211,10 @@ def get_ln_uri(self, ln):
for ch in self.channels:
src = self.lns[ch["source"]]
tgt = self.lns[ch["target"]]
if "." not in src:
src += f".{self.namespace}"
if "." not in tgt:
tgt += f".{self.namespace}"
# Avoid duplicates and reciprocals
if (src, tgt) not in connections and (tgt, src) not in connections:
connections.append((src, tgt))
Expand Down
20 changes: 10 additions & 10 deletions resources/scenarios/test_scenarios/connect_dag.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,16 +56,16 @@ def run_test(self):
self.connect_nodes(6, 7)
self.sync_all()

zero_peers = self.tanks["tank-0000"].getpeerinfo()
one_peers = self.tanks["tank-0001"].getpeerinfo()
two_peers = self.tanks["tank-0002"].getpeerinfo()
three_peers = self.tanks["tank-0003"].getpeerinfo()
four_peers = self.tanks["tank-0004"].getpeerinfo()
five_peers = self.tanks["tank-0005"].getpeerinfo()
six_peers = self.tanks["tank-0006"].getpeerinfo()
seven_peers = self.tanks["tank-0007"].getpeerinfo()
eight_peers = self.tanks["tank-0008"].getpeerinfo()
nine_peers = self.tanks["tank-0009"].getpeerinfo()
zero_peers = self.tanks[f"tank-0000.{self.namespace}"].getpeerinfo()
one_peers = self.tanks[f"tank-0001.{self.namespace}"].getpeerinfo()
two_peers = self.tanks[f"tank-0002.{self.namespace}"].getpeerinfo()
three_peers = self.tanks[f"tank-0003.{self.namespace}"].getpeerinfo()
four_peers = self.tanks[f"tank-0004.{self.namespace}"].getpeerinfo()
five_peers = self.tanks[f"tank-0005.{self.namespace}"].getpeerinfo()
six_peers = self.tanks[f"tank-0006.{self.namespace}"].getpeerinfo()
seven_peers = self.tanks[f"tank-0007.{self.namespace}"].getpeerinfo()
eight_peers = self.tanks[f"tank-0008.{self.namespace}"].getpeerinfo()
nine_peers = self.tanks[f"tank-0009.{self.namespace}"].getpeerinfo()

for node in self.nodes:
self.log.info(f"Node {node.index}: tank={node.tank} ip={node.rpchost}")
Expand Down
4 changes: 3 additions & 1 deletion resources/scenarios/test_scenarios/p2p_interface.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,9 @@ def run_test(self):
self.log.info("Adding the p2p connection")

p2p_block_store = self.nodes[0].add_p2p_connection(
P2PStoreBlock(), dstaddr=self.tanks["tank-0000"].rpchost, dstport=18444
P2PStoreBlock(),
dstaddr=self.tanks[f"tank-0000.{self.namespace}"].rpchost,
dstport=18444,
)

self.log.info("test that an invalid GETDATA doesn't prevent processing of future messages")
Expand Down
4 changes: 3 additions & 1 deletion resources/scenarios/test_scenarios/signet_grinder.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,9 @@ def set_test_params(self):
self.num_nodes = 0

def run_test(self):
self.generatetoaddress(self.tanks["miner"], 1, "tb1qjfplwf7a2dpjj04cx96rysqeastvycc0j50cch")
self.generatetoaddress(
self.tanks[f"miner.{self.namespace}"], 1, "tb1qjfplwf7a2dpjj04cx96rysqeastvycc0j50cch"
)


def main():
Expand Down
Loading