Skip to content

Commit c00bab0

Browse files
authored
test: improve torus0 and governance pallet tests (#84)
Closes CHAIN-70
1 parent 95ff1c3 commit c00bab0

File tree

8 files changed

+942
-457
lines changed

8 files changed

+942
-457
lines changed

pallets/governance/tests/application.rs

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
use pallet_governance::{
2-
application::ApplicationStatus, AgentApplications, GlobalGovernanceConfig,
2+
application::ApplicationStatus, AgentApplications, GlobalGovernanceConfig, Whitelist,
33
};
44
use polkadot_sdk::frame_support::assert_err;
55
use test_utils::*;
@@ -118,7 +118,7 @@ fn whitelist_executes_application_correctly_remove() {
118118
zero_min_burn();
119119

120120
let key = 0;
121-
let adding_key = 1;
121+
let removing_key = 1;
122122
pallet_governance::Curators::<Test>::insert(key, ());
123123

124124
let proposal_cost = GlobalGovernanceConfig::<Test>::get().agent_application_cost;
@@ -128,19 +128,21 @@ fn whitelist_executes_application_correctly_remove() {
128128
// first submit an application
129129
let balance_before = get_balance(key);
130130

131+
Whitelist::<Test>::set(removing_key, Some(()));
132+
131133
assert_ok!(pallet_governance::Pallet::<Test>::submit_application(
132134
get_origin(key),
133-
adding_key,
135+
removing_key,
134136
data.clone(),
135-
false
137+
true
136138
));
137139

138140
let balance_after = get_balance(key);
139141
assert_eq!(balance_after, balance_before - proposal_cost);
140142

141143
let mut application_id: u32 = u32::MAX;
142144
for (_, value) in AgentApplications::<Test>::iter() {
143-
assert_eq!(value.agent_key, adding_key);
145+
assert_eq!(value.agent_key, removing_key);
144146
assert_eq!(value.data, data);
145147
application_id = value.id;
146148
}
@@ -154,8 +156,8 @@ fn whitelist_executes_application_correctly_remove() {
154156

155157
assert_eq!(balance_after_accept, balance_before);
156158

157-
assert!(pallet_governance::whitelist::is_whitelisted::<Test>(
158-
&adding_key
159+
assert!(!pallet_governance::whitelist::is_whitelisted::<Test>(
160+
&removing_key
159161
));
160162

161163
let application =

pallets/governance/tests/config.rs

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
use polkadot_sdk::{
2+
frame_support::assert_err,
3+
sp_runtime::{traits::BadOrigin, Percent},
4+
};
5+
use test_utils::{assert_ok, get_origin, new_test_ext, RuntimeOrigin, Test};
6+
7+
#[test]
8+
fn set_emission_params() {
9+
new_test_ext().execute_with(|| {
10+
assert_ok!(pallet_governance::Pallet::<Test>::set_emission_params(
11+
RuntimeOrigin::root(),
12+
Percent::from_percent(15),
13+
Percent::from_percent(15),
14+
));
15+
16+
assert_eq!(
17+
pallet_emission0::EmissionRecyclingPercentage::<Test>::get(),
18+
Percent::from_percent(15)
19+
);
20+
21+
assert_eq!(
22+
pallet_governance::TreasuryEmissionFee::<Test>::get(),
23+
Percent::from_percent(15)
24+
);
25+
});
26+
}
27+
28+
#[test]
29+
fn set_emission_params_non_root() {
30+
new_test_ext().execute_with(|| {
31+
assert_err!(
32+
pallet_governance::Pallet::<Test>::set_emission_params(
33+
get_origin(0),
34+
Percent::from_percent(15),
35+
Percent::from_percent(15),
36+
),
37+
BadOrigin
38+
);
39+
});
40+
}
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
use pallet_governance::NotDelegatingVotingPower;
2+
use polkadot_sdk::frame_support::assert_ok;
3+
use test_utils::{get_origin, new_test_ext, Test};
4+
5+
#[test]
6+
fn delegates_voting_power_correctly() {
7+
new_test_ext().execute_with(|| {
8+
const MODULE: u32 = 0;
9+
10+
assert_ok!(pallet_governance::Pallet::<Test>::enable_vote_delegation(
11+
get_origin(MODULE)
12+
));
13+
14+
assert!(!NotDelegatingVotingPower::<Test>::get().contains(&MODULE));
15+
});
16+
}
17+
18+
#[test]
19+
fn disable_voting_power_delegation_correctly() {
20+
new_test_ext().execute_with(|| {
21+
const MODULE: u32 = 0;
22+
23+
assert_ok!(pallet_governance::Pallet::<Test>::enable_vote_delegation(
24+
get_origin(MODULE)
25+
));
26+
27+
assert_ok!(pallet_governance::Pallet::<Test>::disable_vote_delegation(
28+
get_origin(MODULE)
29+
));
30+
31+
assert!(NotDelegatingVotingPower::<Test>::get().contains(&MODULE));
32+
});
33+
}

0 commit comments

Comments
 (0)