-
Notifications
You must be signed in to change notification settings - Fork 145
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
6 changed files
with
468 additions
and
125 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,4 +1,2 @@ | ||
| pub mod suite; | ||
| pub mod tests; | ||
|
|
||
| pub use suite::*; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
114 changes: 114 additions & 0 deletions
114
contracts/delegation/dao-vote-delegation/src/testing/suite/cw4.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,114 @@ | ||
| use std::ops::{Deref, DerefMut}; | ||
|
|
||
| use cosmwasm_std::Decimal; | ||
| use dao_testing::{Cw4TestDao, DaoTestingSuite}; | ||
|
|
||
| use super::base::DaoVoteDelegationTestingSuiteBase; | ||
|
|
||
| pub struct Cw4DaoVoteDelegationTestingSuite { | ||
| /// base testing suite that we're extending | ||
| pub base: DaoVoteDelegationTestingSuiteBase, | ||
|
|
||
| /// cw4-group voting DAO | ||
| pub dao: Cw4TestDao, | ||
| /// members of the DAO | ||
| pub members: Vec<cw4::Member>, | ||
| } | ||
|
|
||
| // allow direct access to base testing suite methods | ||
| impl Deref for Cw4DaoVoteDelegationTestingSuite { | ||
| type Target = DaoVoteDelegationTestingSuiteBase; | ||
|
|
||
| fn deref(&self) -> &Self::Target { | ||
| &self.base | ||
| } | ||
| } | ||
|
|
||
| // allow direct access to base testing suite methods | ||
| impl DerefMut for Cw4DaoVoteDelegationTestingSuite { | ||
| fn deref_mut(&mut self) -> &mut Self::Target { | ||
| &mut self.base | ||
| } | ||
| } | ||
|
|
||
| // CONSTRUCTOR | ||
| impl Cw4DaoVoteDelegationTestingSuite { | ||
| pub fn new() -> Self { | ||
| let mut base = DaoVoteDelegationTestingSuiteBase::new(); | ||
| let mut suite = base.cw4(); | ||
|
|
||
| let members = suite.members.clone(); | ||
| let dao = suite.dao(); | ||
|
|
||
| base.dao_core_addr = dao.core_addr.clone(); | ||
|
|
||
| Self { base, dao, members } | ||
| } | ||
|
|
||
| pub fn with_vp_cap_percent(mut self, vp_cap_percent: Decimal) -> Self { | ||
| self.vp_cap_percent = Some(vp_cap_percent); | ||
| self | ||
| } | ||
|
|
||
| pub fn with_delegation_validity_blocks(mut self, delegation_validity_blocks: u64) -> Self { | ||
| self.delegation_validity_blocks = Some(delegation_validity_blocks); | ||
| self | ||
| } | ||
|
|
||
| pub fn with_max_delegations(mut self, max_delegations: u64) -> Self { | ||
| self.max_delegations = Some(max_delegations); | ||
| self | ||
| } | ||
|
|
||
| pub fn build(mut self) -> Self { | ||
| let code_id = self.delegation_code_id; | ||
| let core_addr = self.dao.core_addr.clone(); | ||
| let group_addr = self.dao.x.group_addr.to_string(); | ||
| let vp_cap_percent = self.vp_cap_percent; | ||
| let delegation_validity_blocks = self.delegation_validity_blocks; | ||
| let max_delegations = self.max_delegations; | ||
|
|
||
| self.delegation_addr = self.instantiate( | ||
| code_id, | ||
| &core_addr, | ||
| &crate::msg::InstantiateMsg { | ||
| dao: None, | ||
| vp_hook_callers: Some(vec![group_addr]), | ||
| no_sync_proposal_modules: None, | ||
| vp_cap_percent, | ||
| delegation_validity_blocks, | ||
| max_delegations, | ||
| }, | ||
| &[], | ||
| "delegation", | ||
| Some(core_addr.to_string()), | ||
| ); | ||
|
|
||
| self.setup_delegation_module(); | ||
|
|
||
| self | ||
| } | ||
|
|
||
| /// set up delegation module by adding necessary hooks and adding it to the | ||
| /// proposal modules | ||
| pub fn setup_delegation_module(&mut self) { | ||
| let dao = self.dao.clone(); | ||
| let delegation_addr = self.delegation_addr.to_string(); | ||
|
|
||
| // add voting power changed hook to cw4-group | ||
| self.execute_smart_ok( | ||
| &dao.core_addr, | ||
| &dao.x.group_addr, | ||
| &cw4::Cw4ExecuteMsg::AddHook { | ||
| addr: delegation_addr.clone(), | ||
| }, | ||
| &[], | ||
| ); | ||
|
|
||
| // add vote hook to all proposal modules | ||
| self.add_vote_hook(&dao, &delegation_addr); | ||
|
|
||
| // set the delegation module for all proposal modules | ||
| self.set_delegation_module(&dao, &delegation_addr); | ||
| } | ||
| } |
3 changes: 3 additions & 0 deletions
3
contracts/delegation/dao-vote-delegation/src/testing/suite/mod.rs
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,3 @@ | ||
| pub mod base; | ||
| pub mod cw4; | ||
| pub mod token; |
Oops, something went wrong.