Adapt local-v and mainnet-local-v to collatorStaking

paritytech · Aug 21, 2024 · ff74623 · ff74623
1 parent a5dd877
commit ff74623
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/javascript/packages/orchestrator/src/chain-decorators/local-v.ts b/javascript/packages/orchestrator/src/chain-decorators/local-v.ts
@@ -54,14 +54,10 @@ export async function addCollatorSelection(specPath: string, node: Node) {
   try {
     const chainSpec = readAndParseChainSpec(specPath);
     const runtimeConfig = getRuntimeConfig(chainSpec);
-    if (!runtimeConfig?.collatorSelection?.invulnerables) return;
-
     const { eth_account } = node.accounts;
 
-    if (runtimeConfig.collatorSelection)
-      runtimeConfig.collatorSelection.invulnerables.push(eth_account.address);
-    if (runtimeConfig.collatorStaking)
-      runtimeConfig.collatorStaking.invulnerables.push(eth_account.address);
+    runtimeConfig?.collatorSelection?.invulnerables?.push(eth_account.address);
+    runtimeConfig?.collatorStaking?.invulnerables?.push(eth_account.address);
 
     new CreateLogTable({
       colWidths: [30, 20, 70],

diff --git a/javascript/packages/orchestrator/src/chain-decorators/mainnet-local-v.ts b/javascript/packages/orchestrator/src/chain-decorators/mainnet-local-v.ts
@@ -54,11 +54,10 @@ export async function addCollatorSelection(specPath: string, node: Node) {
   try {
     const chainSpec = readAndParseChainSpec(specPath);
     const runtimeConfig = getRuntimeConfig(chainSpec);
-    if (!runtimeConfig?.collatorSelection?.invulnerables) return;
-
     const { eth_account } = node.accounts;
 
-    runtimeConfig.collatorSelection.invulnerables.push(eth_account.address);
+    runtimeConfig?.collatorSelection?.invulnerables?.push(eth_account.address);
+    runtimeConfig?.collatorStaking?.invulnerables?.push(eth_account.address);
 
     new CreateLogTable({
       colWidths: [30, 20, 70],