add test cases

Author: open-junius

pallets/subtensor/src/tests/evm.rs

@@ -234,3 +234,109 @@ fn test_associate_evm_key_using_wrong_hash_function() {
         );
     });
 }
+
+#[test]
+fn test_associate_evm_key_rate_limit_exceeded() {
+    new_test_ext(1).execute_with(|| {
+        let netuid = NetUid::from(1);
+
+        let tempo: u16 = 2;
+        let modality: u16 = 2;
+        add_network(netuid, tempo, modality);
+        System::set_block_number(EvmKeyAssociateRateLimit::get());
+
+        let coldkey = U256::from(1);
+        let hotkey = U256::from(2);
+        SubtensorModule::create_account_if_non_existent(&coldkey, &hotkey);
+
+        register_ok_neuron(netuid, hotkey, coldkey, 0);
+
+        let pair = ecdsa::Pair::generate().0;
+        let public = pair.public();
+        let evm_key = public_to_evm_key(&public);
+        let block_number = frame_system::Pallet::<Test>::block_number();
+        let hashed_block_number = keccak_256(block_number.encode().as_ref());
+        let hotkey_bytes = hotkey.encode();
+
+        let message = [hotkey_bytes.as_ref(), hashed_block_number.as_ref()].concat();
+        let signature = sign_evm_message(&pair, message);
+
+        // First association should succeed
+        assert_ok!(SubtensorModule::associate_evm_key(
+            RuntimeOrigin::signed(hotkey),
+            netuid,
+            evm_key,
+            block_number,
+            signature.clone(),
+        ));
+
+        System::set_block_number(System::block_number() + 1);
+        let block_number = frame_system::Pallet::<Test>::block_number();
+        let hashed_block_number = keccak_256(block_number.encode().as_ref());
+        let hotkey_bytes = hotkey.encode();
+        let message = [hotkey_bytes.as_ref(), hashed_block_number.as_ref()].concat();
+        let signature = sign_evm_message(&pair, message);
+
+        // Second association should fail due to rate limit
+        assert_noop!(
+            SubtensorModule::associate_evm_key(
+                RuntimeOrigin::signed(hotkey),
+                netuid,
+                evm_key,
+                block_number,
+                signature.clone(),
+            ),
+            Error::<Test>::EvmKeyAssociateRateLimitExceeded
+        );
+
+        System::set_block_number(System::block_number() + EvmKeyAssociateRateLimit::get());
+        let block_number = frame_system::Pallet::<Test>::block_number();
+        let hashed_block_number = keccak_256(block_number.encode().as_ref());
+        let hotkey_bytes = hotkey.encode();
+        let message = [hotkey_bytes.as_ref(), hashed_block_number.as_ref()].concat();
+        let signature = sign_evm_message(&pair, message);
+
+        assert_ok!(SubtensorModule::associate_evm_key(
+            RuntimeOrigin::signed(hotkey),
+            netuid,
+            evm_key,
+            block_number,
+            signature.clone(),
+        ));
+    });
+}
+
+#[test]
+fn test_associate_evm_key_uid_not_found() {
+    new_test_ext(1).execute_with(|| {
+        let netuid = NetUid::from(1);
+
+        let tempo: u16 = 2;
+        let modality: u16 = 2;
+
+        add_network(netuid, tempo, modality);
+
+        let hotkey = U256::from(2);
+
+        let pair = ecdsa::Pair::generate().0;
+        let public = pair.public();
+        let evm_key = public_to_evm_key(&public);
+        let block_number = frame_system::Pallet::<Test>::block_number();
+        let hashed_block_number = keccak_256(block_number.encode().as_ref());
+        let hotkey_bytes = hotkey.encode();
+
+        let message = [hotkey_bytes.as_ref(), hashed_block_number.as_ref()].concat();
+        let signature = sign_evm_message(&pair, message);
+
+        assert_noop!(
+            SubtensorModule::associate_evm_key(
+                RuntimeOrigin::signed(hotkey),
+                netuid,
+                evm_key,
+                block_number,
+                signature.clone(),
+            ),
+            Error::<Test>::NonAssociatedColdKey
+        );
+    });
+}