Jelajahi Sumber

feat(contract-manager): Add optional provider address for entropy script (#2728)

* Add optional provider parameter to script
Amin Moghaddam 5 bulan lalu
induk
melakukan
1547b8e330

+ 8 - 2
contract_manager/scripts/latency_entropy_with_callback.ts

@@ -13,6 +13,11 @@ const parser = yargs(hideBin(process.argv))
       "Usage: $0 --private-key <private-key> --chain <chain-id> | --all-chains <testnet|mainnet>",
   )
   .options({
+    provider: {
+      type: "string",
+      desc: "Provider address to use for the request. Will use the default provider if not specified",
+      demandOption: false,
+    },
     chain: {
       type: "string",
       desc: "test latency for the contract on this chain",
@@ -30,8 +35,9 @@ const parser = yargs(hideBin(process.argv))
 async function testLatency(
   contract: EvmEntropyContract,
   privateKey: PrivateKey,
+  provider?: string,
 ) {
-  const provider = await contract.getDefaultProvider();
+  provider = provider || await contract.getDefaultProvider();
   const userRandomNumber = contract.generateUserRandomNumber();
   const requestResponse = await contract.requestRandomness(
     userRandomNumber,
@@ -106,7 +112,7 @@ async function main() {
   } else if (argv.chain) {
     const chain = DefaultStore.getChainOrThrow(argv.chain, EvmChain);
     const contract = findEntropyContract(chain);
-    await testLatency(contract, privateKey);
+    await testLatency(contract, privateKey, argv.provider);
   }
 }