Update EVM tests

Author: ales-otf

evm-tests/test/neuron.precompile.reveal-weights.test.ts

@@ -1,5 +1,5 @@
 import * as assert from "assert";
-import { getAliceSigner, getDevnetApi, getRandomSubstrateKeypair } from "../src/substrate"
+import { getAliceSigner, getDevnetApi, getRandomSubstrateKeypair, waitForTransactionWithRetry } from "../src/substrate"
 import { devnet } from "@polkadot-api/descriptors"
 import { PolkadotSigner, TypedApi } from "polkadot-api";
 import { convertPublicKeyToSs58, convertH160ToSS58 } from "../src/address-utils"
@@ -70,6 +70,20 @@ describe("Test neuron precompile reveal weights", () => {
         await startCall(api, netuid, coldkey)
 
         console.log("test the case on subnet ", netuid)
+        // Disable admin freeze window and owner hyperparam rate limiting for tests
+        {
+            const alice = getAliceSigner()
+
+            // Set AdminFreezeWindow to 0
+            const setFreezeWindow = api.tx.AdminUtils.sudo_set_admin_freeze_window({ window: 0 })
+            const sudoFreezeTx = api.tx.Sudo.sudo({ call: setFreezeWindow.decodedCall })
+            await waitForTransactionWithRetry(api, sudoFreezeTx, alice)
+
+            // Set OwnerHyperparamRateLimit to 0
+            const setOwnerRateLimit = api.tx.AdminUtils.sudo_set_owner_hparam_rate_limit({ limit: BigInt(0) })
+            const sudoOwnerRateTx = api.tx.Sudo.sudo({ call: setOwnerRateLimit.decodedCall })
+            await waitForTransactionWithRetry(api, sudoOwnerRateTx, alice)
+        }
 
         await setWeightsSetRateLimit(api, netuid, BigInt(0))
 
@@ -164,4 +178,4 @@ describe("Test neuron precompile reveal weights", () => {
             assert.ok(weight[1] !== undefined)
         }
     })
-});
+});

evm-tests/test/neuron.precompile.set-weights.test.ts

@@ -1,6 +1,6 @@
 import * as assert from "assert";
 
-import { getDevnetApi, getRandomSubstrateKeypair } from "../src/substrate"
+import { getAliceSigner, getDevnetApi, getRandomSubstrateKeypair, waitForTransactionWithRetry } from "../src/substrate"
 import { devnet } from "@polkadot-api/descriptors"
 import { TypedApi } from "polkadot-api";
 import { convertH160ToSS58, convertPublicKeyToSs58, } from "../src/address-utils"
@@ -38,6 +38,20 @@ describe("Test neuron precompile contract, set weights function", () => {
         await burnedRegister(api, netuid, convertH160ToSS58(wallet.address), coldkey)
         const uid = await api.query.SubtensorModule.Uids.getValue(netuid, convertH160ToSS58(wallet.address))
         assert.notEqual(uid, undefined)
+        // Disable admin freeze window and owner hyperparam rate limiting for tests
+        {
+            const alice = getAliceSigner()
+
+            // Set AdminFreezeWindow to 0
+            const setFreezeWindow = api.tx.AdminUtils.sudo_set_admin_freeze_window({ window: 0 })
+            const sudoFreezeTx = api.tx.Sudo.sudo({ call: setFreezeWindow.decodedCall })
+            await waitForTransactionWithRetry(api, sudoFreezeTx, alice)
+
+            // Set OwnerHyperparamRateLimit to 0
+            const setOwnerRateLimit = api.tx.AdminUtils.sudo_set_owner_hparam_rate_limit({ limit: BigInt(0) })
+            const sudoOwnerRateTx = api.tx.Sudo.sudo({ call: setOwnerRateLimit.decodedCall })
+            await waitForTransactionWithRetry(api, sudoOwnerRateTx, alice)
+        }
         // disable reveal and enable direct set weights
         await setCommitRevealWeightsEnabled(api, netuid, false)
         await setWeightsSetRateLimit(api, netuid, BigInt(0))
@@ -68,4 +82,4 @@ describe("Test neuron precompile contract, set weights function", () => {
             });
         }
     })
-});
+});

evm-tests/test/staking.precompile.reward.test.ts

@@ -1,5 +1,5 @@
 import * as assert from "assert";
-import { getDevnetApi, getRandomSubstrateKeypair } from "../src/substrate"
+import { getAliceSigner, getDevnetApi, getRandomSubstrateKeypair, waitForTransactionWithRetry } from "../src/substrate"
 import { devnet } from "@polkadot-api/descriptors"
 import { TypedApi } from "polkadot-api";
 import { convertPublicKeyToSs58 } from "../src/address-utils"
@@ -39,6 +39,20 @@ describe("Test neuron precompile reward", () => {
         await startCall(api, netuid, coldkey)
 
         console.log("test the case on subnet ", netuid)
+        // Disable admin freeze window and owner hyperparam rate limiting for tests
+        {
+            const alice = getAliceSigner()
+
+            // Set AdminFreezeWindow to 0
+            const setFreezeWindow = api.tx.AdminUtils.sudo_set_admin_freeze_window({ window: 0 })
+            const sudoFreezeTx = api.tx.Sudo.sudo({ call: setFreezeWindow.decodedCall })
+            await waitForTransactionWithRetry(api, sudoFreezeTx, alice)
+
+            // Set OwnerHyperparamRateLimit to 0
+            const setOwnerRateLimit = api.tx.AdminUtils.sudo_set_owner_hparam_rate_limit({ limit: BigInt(0) })
+            const sudoOwnerRateTx = api.tx.Sudo.sudo({ call: setOwnerRateLimit.decodedCall })
+            await waitForTransactionWithRetry(api, sudoOwnerRateTx, alice)
+        }
 
         await setTxRateLimit(api, BigInt(0))
         await setTempo(api, root_netuid, root_tempo)