12345678910111213141516171819202122232425262728293031323334353637use frame_support::{38 assert_noop, assert_ok, ord_parameter_types, parameter_types,39 traits::{ConstU32, ConstU64, EitherOfDiverse},40 BoundedVec,41};42use frame_system::{EnsureRoot, EnsureSignedBy};43use parity_scale_codec::{Decode, Encode};44use sp_core::H256;45use sp_runtime::{46 traits::{BadOrigin, BlakeTwo256, IdentityLookup},47 BuildStorage,48};4950use super::*;51use crate as pallet_identity;5253type Block = frame_system::mocking::MockBlockU32<Test>;5455frame_support::construct_runtime!(56 pub enum Test {57 System: frame_system,58 Balances: pallet_balances,59 Identity: pallet_identity,60 }61);6263parameter_types! {64 pub BlockWeights: frame_system::limits::BlockWeights =65 frame_system::limits::BlockWeights::simple_max(frame_support::weights::Weight::from_parts(1024, 0));66}67impl frame_system::Config for Test {68 type BaseCallFilter = frame_support::traits::Everything;69 type Block = Block;70 type BlockWeights = ();71 type BlockLength = ();72 type RuntimeOrigin = RuntimeOrigin;73 type Nonce = u64;74 type Hash = H256;75 type RuntimeCall = RuntimeCall;76 type Hashing = BlakeTwo256;77 type AccountId = u64;78 type Lookup = IdentityLookup<Self::AccountId>;79 type RuntimeEvent = RuntimeEvent;80 type BlockHashCount = ConstU32<250>;81 type DbWeight = ();82 type Version = ();83 type PalletInfo = PalletInfo;84 type AccountData = pallet_balances::AccountData<u64>;85 type OnNewAccount = ();86 type OnKilledAccount = ();87 type SystemWeightInfo = ();88 type SS58Prefix = ();89 type OnSetCode = ();90 type MaxConsumers = ConstU32<16>;91}9293impl pallet_balances::Config for Test {94 type Balance = u64;95 type RuntimeEvent = RuntimeEvent;96 type DustRemoval = ();97 type ExistentialDeposit = ConstU64<1>;98 type AccountStore = System;99 type MaxLocks = ();100 type MaxReserves = ();101 type ReserveIdentifier = [u8; 8];102 type WeightInfo = ();103 type RuntimeHoldReason = RuntimeHoldReason;104 type FreezeIdentifier = ();105 type MaxHolds = ();106 type MaxFreezes = ();107}108109parameter_types! {110 pub const MaxAdditionalFields: u32 = 2;111 pub const MaxRegistrars: u32 = 20;112}113114ord_parameter_types! {115 pub const One: u64 = 1;116 pub const Two: u64 = 2;117}118type EnsureOneOrRoot = EitherOfDiverse<EnsureRoot<u64>, EnsureSignedBy<One, u64>>;119type EnsureTwoOrRoot = EitherOfDiverse<EnsureRoot<u64>, EnsureSignedBy<Two, u64>>;120impl pallet_identity::Config for Test {121 type RuntimeEvent = RuntimeEvent;122 type Currency = Balances;123 type Slashed = ();124 type BasicDeposit = ConstU64<10>;125 type FieldDeposit = ConstU64<10>;126 type SubAccountDeposit = ConstU64<10>;127 type MaxSubAccounts = ConstU32<2>;128 type MaxAdditionalFields = MaxAdditionalFields;129 type MaxRegistrars = MaxRegistrars;130 type RegistrarOrigin = EnsureOneOrRoot;131 type ForceOrigin = EnsureTwoOrRoot;132 type WeightInfo = ();133}134135pub fn new_test_ext() -> sp_io::TestExternalities {136 let mut t = <frame_system::GenesisConfig<Test>>::default()137 .build_storage()138 .unwrap();139 pallet_balances::GenesisConfig::<Test> {140 balances: vec![(1, 10), (2, 10), (3, 10), (10, 100), (20, 100), (30, 100)],141 }142 .assimilate_storage(&mut t)143 .unwrap();144 t.into()145}146147fn ten() -> IdentityInfo<MaxAdditionalFields> {148 IdentityInfo {149 display: Data::Raw(b"ten".to_vec().try_into().unwrap()),150 legal: Data::Raw(b"The Right Ordinal Ten, Esq.".to_vec().try_into().unwrap()),151 ..Default::default()152 }153}154155fn twenty() -> IdentityInfo<MaxAdditionalFields> {156 IdentityInfo {157 display: Data::Raw(b"twenty".to_vec().try_into().unwrap()),158 legal: Data::Raw(159 b"The Right Ordinal Twenty, Esq."160 .to_vec()161 .try_into()162 .unwrap(),163 ),164 ..Default::default()165 }166}167168#[test]169fn editing_subaccounts_should_work() {170 new_test_ext().execute_with(|| {171 let data = |x| Data::Raw(vec![x; 1].try_into().unwrap());172173 assert_noop!(174 Identity::add_sub(RuntimeOrigin::signed(10), 20, data(1)),175 Error::<Test>::NoIdentity176 );177178 assert_ok!(Identity::set_identity(179 RuntimeOrigin::signed(10),180 Box::new(ten())181 ));182183 184 assert_ok!(Identity::add_sub(RuntimeOrigin::signed(10), 1, data(1)));185 assert_eq!(SuperOf::<Test>::get(1), Some((10, data(1))));186 assert_eq!(Balances::free_balance(10), 80);187188 189 assert_ok!(Identity::add_sub(RuntimeOrigin::signed(10), 2, data(2)));190 assert_eq!(SuperOf::<Test>::get(1), Some((10, data(1))));191 assert_eq!(SuperOf::<Test>::get(2), Some((10, data(2))));192 assert_eq!(Balances::free_balance(10), 70);193194 195 assert_noop!(196 Identity::add_sub(RuntimeOrigin::signed(10), 3, data(3)),197 Error::<Test>::TooManySubAccounts198 );199200 201 assert_ok!(Identity::rename_sub(RuntimeOrigin::signed(10), 1, data(11)));202 assert_eq!(SuperOf::<Test>::get(1), Some((10, data(11))));203 assert_eq!(SuperOf::<Test>::get(2), Some((10, data(2))));204 assert_eq!(Balances::free_balance(10), 70);205206 207 assert_ok!(Identity::remove_sub(RuntimeOrigin::signed(10), 1));208 assert_eq!(SuperOf::<Test>::get(1), None);209 assert_eq!(SuperOf::<Test>::get(2), Some((10, data(2))));210 assert_eq!(Balances::free_balance(10), 80);211212 213 assert_ok!(Identity::add_sub(RuntimeOrigin::signed(10), 3, data(3)));214 assert_eq!(SuperOf::<Test>::get(1), None);215 assert_eq!(SuperOf::<Test>::get(2), Some((10, data(2))));216 assert_eq!(SuperOf::<Test>::get(3), Some((10, data(3))));217 assert_eq!(Balances::free_balance(10), 70);218 });219}220221#[test]222fn resolving_subaccount_ownership_works() {223 new_test_ext().execute_with(|| {224 let data = |x| Data::Raw(vec![x; 1].try_into().unwrap());225226 assert_ok!(Identity::set_identity(227 RuntimeOrigin::signed(10),228 Box::new(ten())229 ));230 assert_ok!(Identity::set_identity(231 RuntimeOrigin::signed(20),232 Box::new(twenty())233 ));234235 236 assert_ok!(Identity::add_sub(RuntimeOrigin::signed(10), 1, data(1)));237 assert_eq!(Balances::free_balance(1), 10);238 assert_eq!(Balances::free_balance(10), 80);239 assert_eq!(Balances::reserved_balance(10), 20);240 241 assert_noop!(242 Identity::add_sub(RuntimeOrigin::signed(20), 1, data(1)),243 Error::<Test>::AlreadyClaimed244 );245 246 assert_ok!(Identity::quit_sub(RuntimeOrigin::signed(1)));247 248 assert_eq!(Balances::free_balance(1), 20);249 assert_eq!(Balances::free_balance(10), 80);250 assert_eq!(Balances::reserved_balance(10), 10);251 252 assert_ok!(Identity::add_sub(RuntimeOrigin::signed(20), 1, data(1)));253 });254}255256#[test]257fn trailing_zeros_decodes_into_default_data() {258 let encoded = Data::Raw(b"Hello".to_vec().try_into().unwrap()).encode();259 assert!(<(Data, Data)>::decode(&mut &encoded[..]).is_err());260 let input = &mut &encoded[..];261 let (a, b) = <(Data, Data)>::decode(&mut AppendZerosInput::new(input)).unwrap();262 assert_eq!(a, Data::Raw(b"Hello".to_vec().try_into().unwrap()));263 assert_eq!(b, Data::None);264}265266#[test]267fn adding_registrar_should_work() {268 new_test_ext().execute_with(|| {269 assert_ok!(Identity::add_registrar(RuntimeOrigin::signed(1), 3));270 assert_ok!(Identity::set_fee(RuntimeOrigin::signed(3), 0, 10));271 let fields = IdentityFields(IdentityField::Display | IdentityField::Legal);272 assert_ok!(Identity::set_fields(RuntimeOrigin::signed(3), 0, fields));273 assert_eq!(274 Identity::registrars(),275 vec![Some(RegistrarInfo {276 account: 3,277 fee: 10,278 fields279 })]280 );281 });282}283284#[test]285fn amount_of_registrars_is_limited() {286 new_test_ext().execute_with(|| {287 for i in 1..MaxRegistrars::get() + 1 {288 assert_ok!(Identity::add_registrar(RuntimeOrigin::signed(1), i as u64));289 }290 let last_registrar = MaxRegistrars::get() as u64 + 1;291 assert_noop!(292 Identity::add_registrar(RuntimeOrigin::signed(1), last_registrar),293 Error::<Test>::TooManyRegistrars294 );295 });296}297298#[test]299fn registration_should_work() {300 new_test_ext().execute_with(|| {301 assert_ok!(Identity::add_registrar(RuntimeOrigin::signed(1), 3));302 assert_ok!(Identity::set_fee(RuntimeOrigin::signed(3), 0, 10));303 let mut three_fields = ten();304 three_fields305 .additional306 .try_push(Default::default())307 .unwrap();308 three_fields309 .additional310 .try_push(Default::default())311 .unwrap();312 assert!(three_fields313 .additional314 .try_push(Default::default())315 .is_err());316 assert_ok!(Identity::set_identity(317 RuntimeOrigin::signed(10),318 Box::new(ten())319 ));320 assert_eq!(Identity::identity(10).unwrap().info, ten());321 assert_eq!(Balances::free_balance(10), 90);322 assert_ok!(Identity::clear_identity(RuntimeOrigin::signed(10)));323 assert_eq!(Balances::free_balance(10), 100);324 assert_noop!(325 Identity::clear_identity(RuntimeOrigin::signed(10)),326 Error::<Test>::NotNamed327 );328 });329}330331#[test]332fn uninvited_judgement_should_work() {333 new_test_ext().execute_with(|| {334 assert_noop!(335 Identity::provide_judgement(336 RuntimeOrigin::signed(3),337 0,338 10,339 Judgement::Reasonable,340 H256::random()341 ),342 Error::<Test>::InvalidIndex343 );344345 assert_ok!(Identity::add_registrar(RuntimeOrigin::signed(1), 3));346 assert_noop!(347 Identity::provide_judgement(348 RuntimeOrigin::signed(3),349 0,350 10,351 Judgement::Reasonable,352 H256::random()353 ),354 Error::<Test>::InvalidTarget355 );356357 assert_ok!(Identity::set_identity(358 RuntimeOrigin::signed(10),359 Box::new(ten())360 ));361 assert_noop!(362 Identity::provide_judgement(363 RuntimeOrigin::signed(3),364 0,365 10,366 Judgement::Reasonable,367 H256::random()368 ),369 Error::<Test>::JudgementForDifferentIdentity370 );371372 let identity_hash = BlakeTwo256::hash_of(&ten());373374 assert_noop!(375 Identity::provide_judgement(376 RuntimeOrigin::signed(10),377 0,378 10,379 Judgement::Reasonable,380 identity_hash381 ),382 Error::<Test>::InvalidIndex383 );384 assert_noop!(385 Identity::provide_judgement(386 RuntimeOrigin::signed(3),387 0,388 10,389 Judgement::FeePaid(1),390 identity_hash391 ),392 Error::<Test>::InvalidJudgement393 );394395 assert_ok!(Identity::provide_judgement(396 RuntimeOrigin::signed(3),397 0,398 10,399 Judgement::Reasonable,400 identity_hash401 ));402 assert_eq!(403 Identity::identity(10).unwrap().judgements,404 vec![(0, Judgement::Reasonable)]405 );406 });407}408409#[test]410fn clearing_judgement_should_work() {411 new_test_ext().execute_with(|| {412 assert_ok!(Identity::add_registrar(RuntimeOrigin::signed(1), 3));413 assert_ok!(Identity::set_identity(414 RuntimeOrigin::signed(10),415 Box::new(ten())416 ));417 assert_ok!(Identity::provide_judgement(418 RuntimeOrigin::signed(3),419 0,420 10,421 Judgement::Reasonable,422 BlakeTwo256::hash_of(&ten())423 ));424 assert_ok!(Identity::clear_identity(RuntimeOrigin::signed(10)));425 assert_eq!(Identity::identity(10), None);426 });427}428429#[test]430fn killing_slashing_should_work() {431 new_test_ext().execute_with(|| {432 assert_ok!(Identity::set_identity(433 RuntimeOrigin::signed(10),434 Box::new(ten())435 ));436 assert_noop!(437 Identity::kill_identity(RuntimeOrigin::signed(1), 10),438 BadOrigin439 );440 assert_ok!(Identity::kill_identity(RuntimeOrigin::signed(2), 10));441 assert_eq!(Identity::identity(10), None);442 assert_eq!(Balances::free_balance(10), 90);443 assert_noop!(444 Identity::kill_identity(RuntimeOrigin::signed(2), 10),445 Error::<Test>::NotNamed446 );447 });448}449450#[test]451fn setting_subaccounts_should_work() {452 new_test_ext().execute_with(|| {453 let mut subs = vec![(20, Data::Raw(vec![40; 1].try_into().unwrap()))];454 assert_noop!(455 Identity::set_subs(RuntimeOrigin::signed(10), subs.clone()),456 Error::<Test>::NotFound457 );458459 assert_ok!(Identity::set_identity(460 RuntimeOrigin::signed(10),461 Box::new(ten())462 ));463 assert_ok!(Identity::set_subs(RuntimeOrigin::signed(10), subs.clone()));464 assert_eq!(Balances::free_balance(10), 80);465 assert_eq!(Identity::subs_of(10), (10, vec![20].try_into().unwrap()));466 assert_eq!(467 Identity::super_of(20),468 Some((10, Data::Raw(vec![40; 1].try_into().unwrap())))469 );470471 472 subs.push((30, Data::Raw(vec![50; 1].try_into().unwrap())));473 assert_ok!(Identity::set_subs(RuntimeOrigin::signed(10), subs.clone()));474 assert_eq!(Balances::free_balance(10), 70);475 assert_eq!(476 Identity::subs_of(10),477 (20, vec![20, 30].try_into().unwrap())478 );479 assert_eq!(480 Identity::super_of(20),481 Some((10, Data::Raw(vec![40; 1].try_into().unwrap())))482 );483 assert_eq!(484 Identity::super_of(30),485 Some((10, Data::Raw(vec![50; 1].try_into().unwrap())))486 );487488 489 subs[0] = (40, Data::Raw(vec![60; 1].try_into().unwrap()));490 assert_ok!(Identity::set_subs(RuntimeOrigin::signed(10), subs.clone()));491 assert_eq!(Balances::free_balance(10), 70); 492 assert_eq!(493 Identity::subs_of(10),494 (20, vec![40, 30].try_into().unwrap())495 );496 assert_eq!(Identity::super_of(20), None);497 assert_eq!(498 Identity::super_of(30),499 Some((10, Data::Raw(vec![50; 1].try_into().unwrap())))500 );501 assert_eq!(502 Identity::super_of(40),503 Some((10, Data::Raw(vec![60; 1].try_into().unwrap())))504 );505506 507 assert_ok!(Identity::set_subs(RuntimeOrigin::signed(10), vec![]));508 assert_eq!(Balances::free_balance(10), 90);509 assert_eq!(Identity::subs_of(10), (0, BoundedVec::default()));510 assert_eq!(Identity::super_of(30), None);511 assert_eq!(Identity::super_of(40), None);512513 subs.push((20, Data::Raw(vec![40; 1].try_into().unwrap())));514 assert_noop!(515 Identity::set_subs(RuntimeOrigin::signed(10), subs.clone()),516 Error::<Test>::TooManySubAccounts517 );518 });519}520521#[test]522fn clearing_account_should_remove_subaccounts_and_refund() {523 new_test_ext().execute_with(|| {524 assert_ok!(Identity::set_identity(525 RuntimeOrigin::signed(10),526 Box::new(ten())527 ));528 assert_ok!(Identity::set_subs(529 RuntimeOrigin::signed(10),530 vec![(20, Data::Raw(vec![40; 1].try_into().unwrap()))]531 ));532 assert_ok!(Identity::clear_identity(RuntimeOrigin::signed(10)));533 assert_eq!(Balances::free_balance(10), 100);534 assert!(Identity::super_of(20).is_none());535 });536}537538#[test]539fn killing_account_should_remove_subaccounts_and_not_refund() {540 new_test_ext().execute_with(|| {541 assert_ok!(Identity::set_identity(542 RuntimeOrigin::signed(10),543 Box::new(ten())544 ));545 assert_ok!(Identity::set_subs(546 RuntimeOrigin::signed(10),547 vec![(20, Data::Raw(vec![40; 1].try_into().unwrap()))]548 ));549 assert_ok!(Identity::kill_identity(RuntimeOrigin::signed(2), 10));550 assert_eq!(Balances::free_balance(10), 80);551 assert!(Identity::super_of(20).is_none());552 });553}554555#[test]556fn cancelling_requested_judgement_should_work() {557 new_test_ext().execute_with(|| {558 assert_ok!(Identity::add_registrar(RuntimeOrigin::signed(1), 3));559 assert_ok!(Identity::set_fee(RuntimeOrigin::signed(3), 0, 10));560 assert_noop!(561 Identity::cancel_request(RuntimeOrigin::signed(10), 0),562 Error::<Test>::NoIdentity563 );564 assert_ok!(Identity::set_identity(565 RuntimeOrigin::signed(10),566 Box::new(ten())567 ));568 assert_ok!(Identity::request_judgement(569 RuntimeOrigin::signed(10),570 0,571 10572 ));573 assert_ok!(Identity::cancel_request(RuntimeOrigin::signed(10), 0));574 assert_eq!(Balances::free_balance(10), 90);575 assert_noop!(576 Identity::cancel_request(RuntimeOrigin::signed(10), 0),577 Error::<Test>::NotFound578 );579580 assert_ok!(Identity::provide_judgement(581 RuntimeOrigin::signed(3),582 0,583 10,584 Judgement::Reasonable,585 BlakeTwo256::hash_of(&ten())586 ));587 assert_noop!(588 Identity::cancel_request(RuntimeOrigin::signed(10), 0),589 Error::<Test>::JudgementGiven590 );591 });592}593594#[test]595fn requesting_judgement_should_work() {596 new_test_ext().execute_with(|| {597 assert_ok!(Identity::add_registrar(RuntimeOrigin::signed(1), 3));598 assert_ok!(Identity::set_fee(RuntimeOrigin::signed(3), 0, 10));599 assert_ok!(Identity::set_identity(600 RuntimeOrigin::signed(10),601 Box::new(ten())602 ));603 assert_noop!(604 Identity::request_judgement(RuntimeOrigin::signed(10), 0, 9),605 Error::<Test>::FeeChanged606 );607 assert_ok!(Identity::request_judgement(608 RuntimeOrigin::signed(10),609 0,610 10611 ));612 613 assert_eq!(Balances::free_balance(10), 80);614615 616 assert_noop!(617 Identity::request_judgement(RuntimeOrigin::signed(10), 0, 10),618 Error::<Test>::StickyJudgement619 );620 assert_ok!(Identity::provide_judgement(621 RuntimeOrigin::signed(3),622 0,623 10,624 Judgement::Erroneous,625 BlakeTwo256::hash_of(&ten())626 ));627 628 assert_eq!(Balances::free_balance(3), 20);629630 631 assert_noop!(632 Identity::request_judgement(RuntimeOrigin::signed(10), 0, 10),633 Error::<Test>::StickyJudgement634 );635636 637 assert_ok!(Identity::add_registrar(RuntimeOrigin::signed(1), 4));638 assert_ok!(Identity::request_judgement(639 RuntimeOrigin::signed(10),640 1,641 10642 ));643644 645 assert_ok!(Identity::provide_judgement(646 RuntimeOrigin::signed(3),647 0,648 10,649 Judgement::OutOfDate,650 BlakeTwo256::hash_of(&ten())651 ));652 assert_ok!(Identity::request_judgement(653 RuntimeOrigin::signed(10),654 0,655 10656 ));657 });658}659660#[test]661fn provide_judgement_should_return_judgement_payment_failed_error() {662 new_test_ext().execute_with(|| {663 assert_ok!(Identity::add_registrar(RuntimeOrigin::signed(1), 3));664 assert_ok!(Identity::set_fee(RuntimeOrigin::signed(3), 0, 10));665 assert_ok!(Identity::set_identity(666 RuntimeOrigin::signed(10),667 Box::new(ten())668 ));669 assert_ok!(Identity::request_judgement(670 RuntimeOrigin::signed(10),671 0,672 10673 ));674 675 assert_eq!(Balances::free_balance(10), 80);676677 678 Balances::make_free_balance_be(&3, 0);679 assert_noop!(680 Identity::provide_judgement(681 RuntimeOrigin::signed(3),682 0,683 10,684 Judgement::Erroneous,685 BlakeTwo256::hash_of(&ten())686 ),687 Error::<Test>::JudgementPaymentFailed688 );689 });690}691692#[test]693fn field_deposit_should_work() {694 new_test_ext().execute_with(|| {695 assert_ok!(Identity::add_registrar(RuntimeOrigin::signed(1), 3));696 assert_ok!(Identity::set_fee(RuntimeOrigin::signed(3), 0, 10));697 assert_ok!(Identity::set_identity(698 RuntimeOrigin::signed(10),699 Box::new(IdentityInfo {700 additional: vec![701 (702 Data::Raw(b"number".to_vec().try_into().unwrap()),703 Data::Raw(10u32.encode().try_into().unwrap())704 ),705 (706 Data::Raw(b"text".to_vec().try_into().unwrap()),707 Data::Raw(b"10".to_vec().try_into().unwrap())708 ),709 ]710 .try_into()711 .unwrap(),712 ..Default::default()713 })714 ));715 assert_eq!(Balances::free_balance(10), 70);716 });717}718719#[test]720fn setting_account_id_should_work() {721 new_test_ext().execute_with(|| {722 assert_ok!(Identity::add_registrar(RuntimeOrigin::signed(1), 3));723 724 assert_noop!(725 Identity::set_account_id(RuntimeOrigin::signed(4), 0, 3),726 Error::<Test>::InvalidIndex727 );728 729 assert_ok!(Identity::set_account_id(RuntimeOrigin::signed(3), 0, 4));730 731 assert_ok!(Identity::set_account_id(RuntimeOrigin::signed(4), 0, 3));732 });733}734735#[test]736fn test_has_identity() {737 new_test_ext().execute_with(|| {738 assert_ok!(Identity::set_identity(739 RuntimeOrigin::signed(10),740 Box::new(ten())741 ));742 assert!(Identity::has_identity(&10, IdentityField::Display as u64));743 assert!(Identity::has_identity(&10, IdentityField::Legal as u64));744 assert!(Identity::has_identity(745 &10,746 IdentityField::Display as u64 | IdentityField::Legal as u64747 ));748 assert!(!Identity::has_identity(749 &10,750 IdentityField::Display as u64 | IdentityField::Legal as u64 | IdentityField::Web as u64751 ));752 });753}