diff --git a/treasuries/src/tests.rs b/treasuries/src/tests.rs index 7b29c1f9..61ba7511 100644 --- a/treasuries/src/tests.rs +++ b/treasuries/src/tests.rs @@ -16,6 +16,7 @@ //! Unit tests for the encointer_treasuries module. +use std::str::FromStr; use super::*; use crate::mock::{Balances, EncointerTreasuries, System}; use frame_support::assert_ok; @@ -24,43 +25,50 @@ use sp_core::crypto::Ss58Codec; use test_utils::{helpers::*, *}; pub type BalanceOf = - <::Currency as Currency<::AccountId>>::Balance; +<::Currency as Currency<::AccountId>>::Balance; #[test] fn treasury_spending_works() { - new_test_ext().execute_with(|| { - System::set_block_number(System::block_number() + 1); // this is needed to assert events - let beneficiary = AccountId::from(AccountKeyring::Alice); - let amount: BalanceOf = 100_000_000; - let cid = CommunityIdentifier::default(); + new_test_ext().execute_with(|| { + System::set_block_number(System::block_number() + 1); // this is needed to assert events + let beneficiary = AccountId::from(AccountKeyring::Alice); + let amount: BalanceOf = 100_000_000; + let cid = CommunityIdentifier::default(); - let treasury = EncointerTreasuries::get_community_treasury_account_unchecked(Some(cid)); - Balances::make_free_balance_be(&treasury, 500_000_000); + let treasury = EncointerTreasuries::get_community_treasury_account_unchecked(Some(cid)); + Balances::make_free_balance_be(&treasury, 500_000_000); - assert_ok!(EncointerTreasuries::do_spend_native(Some(cid), beneficiary.clone(), amount)); - assert_eq!(Balances::free_balance(&treasury), 400_000_000); - assert_eq!(Balances::free_balance(&beneficiary), amount); - assert_eq!( - last_event::(), - Some(Event::SpentNative { treasury, beneficiary, amount }.into()) - ); - }); + assert_ok!(EncointerTreasuries::do_spend_native(Some(cid), beneficiary.clone(), amount)); + assert_eq!(Balances::free_balance(&treasury), 400_000_000); + assert_eq!(Balances::free_balance(&beneficiary), amount); + assert_eq!( + last_event::(), + Some(Event::SpentNative { treasury, beneficiary, amount }.into()) + ); + }); } #[test] fn treasury_getter_works() { - new_test_ext().execute_with(|| { - let treasury_account = EncointerTreasuries::get_community_treasury_account_unchecked(None); - assert_eq!( - treasury_account.to_ss58check(), - "5FU79FVXdN3RYSj8857XjNT2SgeRrhk8iUzjb75X1yc8YDkZ" - ); - let cid = CommunityIdentifier::default(); - let treasury_account = - EncointerTreasuries::get_community_treasury_account_unchecked(Some(cid)); - assert_eq!( - treasury_account.to_ss58check(), - "5D58hM12H6Gkc1h1chuzbbJ3FitGHAyTM6ECkdz4hi5dFheH" - ) - }); + new_test_ext().execute_with(|| { + let treasury_account = EncointerTreasuries::get_community_treasury_account_unchecked(None); + assert_eq!( + treasury_account.to_ss58check(), + "5FU79FVXdN3RYSj8857XjNT2SgeRrhk8iUzjb75X1yc8YDkZ" + ); + let cid = CommunityIdentifier::default(); + let treasury_account = + EncointerTreasuries::get_community_treasury_account_unchecked(Some(cid)); + assert_eq!( + treasury_account.to_ss58check(), + "5D58hM12H6Gkc1h1chuzbbJ3FitGHAyTM6ECkdz4hi5dFheH" + ); + let cid = CommunityIdentifier::from_str("sqm1v79dF6b").expect("invalid community id"); + let treasury_account = + EncointerTreasuries::get_community_treasury_account_unchecked(Some(cid)); + assert_eq!( + treasury_account.to_ss58check(), + "5CWoc3mGF9VEnuZzBbPWxhKPvY743AGwxUbvkYQHS8yWZbem" + ) + }); }