Unit tests for EVM template

evm-template/Cargo.toml

@@ -184,6 +184,8 @@ xcm-primitives = { git = "https://github.com/OpenZeppelin/moonbeam.git", branch
 substrate-runtimes-fuzzers = { git = "https://github.com/srlabs/substrate-runtime-fuzzer.git", default-features = false, tag = "polkadot-v1.12.0" }
 ziggy = { version = "0.8", default-features = false }
 
+ethereum = { version = "0.15.0" }
+
 [workspace.lints.clippy]
 large_enum_variant = "allow"
 too_many_arguments = "allow"

evm-template/runtime/Cargo.toml

@@ -112,6 +112,7 @@ xcm-primitives = { workspace = true }
 
 
 [dev-dependencies]
+ethereum = { workspace = true }
 polkadot-runtime-parachains = { workspace = true }
 sp-io = { workspace = true }
 sp-tracing = { workspace = true }

evm-template/runtime/src/configs/asset_config.rs

@@ -171,3 +171,167 @@ impl AccountIdAssetIdConversion<AccountId, AssetId> for Runtime {
         AccountId::from(data)
     }
 }
+
+#[cfg(test)]
+mod tests {
+    mod asset_registrar {
+        use pallet_asset_manager::AssetRegistrar;
+        use sp_io::TestExternalities;
+
+        use crate::{
+            configs::{AssetRegistrar as Registrar, AssetRegistrarMetadata},
+            types::AssetId,
+            AssetManager, Assets, Runtime, RuntimeOrigin,
+        };
+
+        #[test]
+        fn test_destroy_asset_dispatch_info_weight() {
+            let asset_id: AssetId = 1;
+            let _ = <Registrar as AssetRegistrar<Runtime>>::destroy_asset_dispatch_info_weight(
+                asset_id,
+            );
+        }
+
+        #[test]
+        fn test_destroy_foreign_asset() {
+            TestExternalities::default().execute_with(|| {
+                let asset_id: AssetId = 1;
+                let _ = Assets::force_create(
+                    RuntimeOrigin::root(),
+                    asset_id.into(),
+                    AssetManager::account_id(),
+                    true,
+                    1,
+                );
+                let res = <Registrar as AssetRegistrar<Runtime>>::destroy_foreign_asset(asset_id);
+                assert!(res.is_ok());
+            });
+        }
+
+        #[test]
+        fn test_create_foreign_asset() {
+            TestExternalities::default().execute_with(|| {
+                let asset_id = 1;
+                let res = <Registrar as AssetRegistrar<Runtime>>::create_foreign_asset(
+                    asset_id,
+                    1,
+                    AssetRegistrarMetadata {
+                        name: vec![0, 1, 2, 3],
+                        symbol: vec![4, 5, 6, 7],
+                        decimals: 6,
+                        is_frozen: false,
+                    },
+                    false,
+                );
+                assert!(res.is_ok());
+            });
+        }
+    }
+
+    mod account_asset_id_conversion {
+        use core::str::FromStr;
+
+        use fp_account::AccountId20;
+
+        use crate::{
+            configs::{
+                asset_config::FOREIGN_ASSET_PRECOMPILE_ADDRESS_PREFIX, AccountIdAssetIdConversion,
+            },
+            types::AssetId,
+            AccountId, Runtime,
+        };
+
+        #[test]
+        fn test_account_to_asset_id_success() {
+            let expected_asset_id: AssetId = 1;
+            let mut data = [0u8; 32];
+            data[0..4].copy_from_slice(FOREIGN_ASSET_PRECOMPILE_ADDRESS_PREFIX);
+            data[4..20].copy_from_slice(&expected_asset_id.to_be_bytes());
+            let account_id = AccountId::from(data);
+            let (prefix, asset_id) = Runtime::account_to_asset_id(account_id)
+                .expect("Account to asset id conversion failed");
+            assert_eq!(prefix, FOREIGN_ASSET_PRECOMPILE_ADDRESS_PREFIX);
+            assert_eq!(asset_id, expected_asset_id);
+        }
+
+        #[test]
+        fn test_account_to_asset_id_error() {
+            let expected_asset_id: AssetId = 1;
+            let mut data = [0u8; 32];
+            data[0..3].copy_from_slice(&FOREIGN_ASSET_PRECOMPILE_ADDRESS_PREFIX[0..3]);
+            data[3..4].copy_from_slice(&[0]);
+            data[4..20].copy_from_slice(&expected_asset_id.to_be_bytes());
+            let account_id = AccountId::from(data);
+            let res = Runtime::account_to_asset_id(account_id);
+            assert_eq!(res, None);
+        }
+
+        #[test]
+        fn test_asset_id_to_account() {
+            let expected =
+                AccountId20::from_str("0xFFFFFFFF00000000000000000000000000000001").unwrap();
+            let asset_id = 1;
+            let result =
+                Runtime::asset_id_to_account(FOREIGN_ASSET_PRECOMPILE_ADDRESS_PREFIX, asset_id);
+            assert_eq!(result, expected);
+        }
+    }
+
+    mod asset_type {
+        use crate::{configs::AssetType, types::AssetId};
+
+        #[test]
+        fn test_asset_type_default() {
+            let default_asset_type = AssetType::default();
+            assert_eq!(
+                default_asset_type,
+                AssetType::Xcm(xcm::v3::Location {
+                    parents: 0,
+                    interior: xcm::v3::Junctions::Here
+                })
+            );
+        }
+
+        #[test]
+        fn test_asset_type_from_location_v3() {
+            let location = xcm::v3::Location {
+                parents: 0,
+                interior: xcm::v3::Junctions::X1(xcm::v3::Junction::OnlyChild),
+            };
+            let asset_type = AssetType::from(location);
+
+            assert_eq!(asset_type, AssetType::Xcm(location));
+        }
+
+        #[test]
+        fn test_asset_type_try_from_location_v4() {
+            let location =
+                xcm::latest::Location { parents: 0, interior: xcm::latest::Junctions::Here };
+            let old_location: xcm::v3::Location =
+                xcm::v3::Location { parents: 0, interior: xcm::v3::Junctions::Here };
+            let asset_type = AssetType::try_from(location)
+                .expect("AssetType conversion from location v4 failed");
+
+            assert_eq!(asset_type, AssetType::Xcm(old_location));
+        }
+
+        #[test]
+        fn test_asset_type_into_location() {
+            let location = xcm::v3::Location { parents: 0, interior: xcm::v3::Junctions::Here };
+            let asset_type = AssetType::Xcm(location);
+            let converted: Option<xcm::v3::Location> = asset_type.into();
+            assert_eq!(converted, Some(location));
+        }
+
+        #[test]
+        fn test_asset_type_into_asset_id() {
+            let location = xcm::v3::Location { parents: 0, interior: xcm::v3::Junctions::Here };
+            let expected_asset_id: u128 = 114990615950639921045101060455076456094;
+            let asset_type = AssetType::Xcm(location);
+
+            let asset_id = AssetId::from(asset_type);
+
+            assert_eq!(asset_id, expected_asset_id);
+        }
+    }
+}

evm-template/runtime/src/configs/governance/origins.rs

@@ -113,3 +113,172 @@ pub mod pallet_custom_origins {
         }
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use frame_support::traits::EnsureOrigin;
+
+    use crate::{
+        configs::{governance::Spender, pallet_custom_origins::Origin},
+        constants::currency::{CENTS, GRAND},
+        Balance,
+    };
+
+    #[derive(Debug, PartialEq)]
+    enum TestOrigin {
+        SmallTipper,
+        SmallSpender,
+        Treasurer,
+        BigTipper,
+        MediumSpender,
+        BigSpender,
+        WhitelistedCaller,
+        ReferendumCanceller,
+        ReferendumKiller,
+    }
+
+    impl From<TestOrigin> for Result<Origin, TestOrigin> {
+        fn from(value: TestOrigin) -> Self {
+            match value {
+                TestOrigin::SmallTipper => Ok(Origin::SmallTipper),
+                TestOrigin::SmallSpender => Ok(Origin::SmallSpender),
+                TestOrigin::Treasurer => Ok(Origin::Treasurer),
+                TestOrigin::ReferendumCanceller => Ok(Origin::ReferendumCanceller),
+                TestOrigin::ReferendumKiller => Ok(Origin::ReferendumKiller),
+                TestOrigin::BigTipper => Ok(Origin::BigTipper),
+                TestOrigin::MediumSpender => Ok(Origin::MediumSpender),
+                TestOrigin::BigSpender => Ok(Origin::BigSpender),
+                TestOrigin::WhitelistedCaller => Ok(Origin::WhitelistedCaller),
+            }
+        }
+    }
+
+    impl From<Origin> for TestOrigin {
+        fn from(value: Origin) -> Self {
+            match value {
+                Origin::SmallTipper => TestOrigin::SmallTipper,
+                Origin::SmallSpender => TestOrigin::SmallSpender,
+                Origin::Treasurer => TestOrigin::Treasurer,
+                Origin::ReferendumCanceller => TestOrigin::ReferendumCanceller,
+                Origin::ReferendumKiller => TestOrigin::ReferendumKiller,
+                Origin::BigTipper => TestOrigin::BigTipper,
+                Origin::MediumSpender => TestOrigin::MediumSpender,
+                Origin::BigSpender => TestOrigin::BigSpender,
+                Origin::WhitelistedCaller => TestOrigin::WhitelistedCaller,
+            }
+        }
+    }
+    mod spender {
+        use super::*;
+
+        #[test]
+        fn test_small_tipper() {
+            let a: Balance =
+                Spender::try_origin(TestOrigin::SmallTipper).expect("SmallTipper misconfigured");
+            assert_eq!(a, 250 * 3 * CENTS);
+        }
+
+        #[test]
+        fn test_small_spender() {
+            let a: Balance =
+                Spender::try_origin(TestOrigin::SmallSpender).expect("SmallSpender misconfigured");
+            assert_eq!(a, 10 * GRAND);
+        }
+
+        #[test]
+        fn test_big_tipper() {
+            let a: Balance =
+                Spender::try_origin(TestOrigin::BigTipper).expect("BigTipper misconfigured");
+            assert_eq!(a, GRAND);
+        }
+
+        #[test]
+        fn test_medium_spender() {
+            let a: Balance = Spender::try_origin(TestOrigin::MediumSpender)
+                .expect("MediumSpender misconfigured");
+            assert_eq!(a, 100 * GRAND);
+        }
+
+        #[test]
+        fn test_big_spender() {
+            let a: Balance =
+                Spender::try_origin(TestOrigin::BigSpender).expect("BigSpender misconfigured");
+            assert_eq!(a, 1_000 * GRAND);
+        }
+
+        #[test]
+        fn test_treasurer() {
+            let a: Balance =
+                Spender::try_origin(TestOrigin::Treasurer).expect("Treasurer misconfigured");
+            assert_eq!(a, 10_000 * GRAND);
+        }
+
+        #[test]
+        fn test_referendum_killer() {
+            let a: TestOrigin = Spender::try_origin(TestOrigin::ReferendumKiller)
+                .expect_err("ReferendumKiller misconfigured");
+            assert_eq!(a, TestOrigin::ReferendumKiller);
+        }
+
+        #[test]
+        fn test_referendum_canceller() {
+            let a: TestOrigin = Spender::try_origin(TestOrigin::ReferendumCanceller)
+                .expect_err("ReferendumCanceller misconfigured");
+            assert_eq!(a, TestOrigin::ReferendumCanceller);
+        }
+
+        #[test]
+        fn test_whitelisted_caller() {
+            let a: TestOrigin = Spender::try_origin(TestOrigin::WhitelistedCaller)
+                .expect_err("WhitelistedCaller misconfigured");
+            assert_eq!(a, TestOrigin::WhitelistedCaller);
+        }
+
+        #[test]
+        #[cfg(feature = "runtime-benchmarks")]
+        fn test_try_successful_origin() {
+            let a: TestOrigin = Spender::try_successful_origin().expect("incorrect setup");
+            assert_eq!(a, TestOrigin::Treasurer);
+        }
+    }
+
+    mod treasurer {
+        use super::*;
+        use crate::configs::pallet_custom_origins::Treasurer;
+
+        #[test]
+        pub fn test_treasurer_try_origin() {
+            let () = Treasurer::try_origin(TestOrigin::Treasurer).expect("incorrect configuration");
+            let a = Treasurer::try_origin(TestOrigin::MediumSpender).expect_err("should be err");
+            assert_eq!(a, TestOrigin::MediumSpender)
+        }
+
+        #[test]
+        #[cfg(feature = "runtime-benchmarks")]
+        fn test_treasurer_try_successful_origin() {
+            let a: Result<TestOrigin, ()> = Treasurer::try_successful_origin();
+            assert_eq!(a, Ok(TestOrigin::Treasurer));
+        }
+    }
+
+    mod referendum_canceller {
+        use super::*;
+        use crate::configs::pallet_custom_origins::ReferendumCanceller;
+
+        #[test]
+        pub fn test_referendum_canceller_try_origin() {
+            let () = ReferendumCanceller::try_origin(TestOrigin::ReferendumCanceller)
+                .expect("incorrect configuration");
+            let a = ReferendumCanceller::try_origin(TestOrigin::MediumSpender)
+                .expect_err("should be err");
+            assert_eq!(a, TestOrigin::MediumSpender)
+        }
+
+        #[test]
+        #[cfg(feature = "runtime-benchmarks")]
+        fn test_treasurer_try_successful_origin() {
+            let a: Result<TestOrigin, ()> = ReferendumCanceller::try_successful_origin();
+            assert_eq!(a, Ok(TestOrigin::ReferendumCanceller));
+        }
+    }
+}