Skip to content

Commit

Permalink
added some more tests
Browse files Browse the repository at this point in the history
  • Loading branch information
NoahSaso committed Feb 4, 2025
1 parent ae3d765 commit ba32eee
Showing 1 changed file with 52 additions and 0 deletions.
52 changes: 52 additions & 0 deletions contracts/delegation/dao-vote-delegation/src/testing/tests.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1071,6 +1071,58 @@ fn test_migrate() {
assert_eq!(version.version, CONTRACT_VERSION);
}

#[test]
fn test_change_member_vp() {
let mut suite = Cw4DaoVoteDelegationTestingSuite::new().build();

suite.register(ADDR0);
suite.delegate(ADDR1, ADDR0, Decimal::percent(100));
suite.advance_block();

// ensure all of ADDR0's voting power is delegated to ADDR1
let initial_vp = suite.members[1].weight;
suite.assert_delegate_total_delegated_vp(ADDR0, initial_vp);

// double ADDR0's voting power
let new_vp = initial_vp * 2;
let dao_core_addr = suite.dao.core_addr.clone();
let group_addr = suite.dao.x.group_addr.clone();
suite.execute_smart_ok(
&dao_core_addr,
&group_addr,
&cw4_group::msg::ExecuteMsg::UpdateMembers {
add: vec![cw4::Member {
addr: ADDR1.to_string(),
weight: new_vp,
}],
remove: vec![],
},
&[],
);
suite.advance_block();

// ensure all of ADDR0's new voting power is now delegated to ADDR1
suite.assert_delegate_total_delegated_vp(ADDR0, new_vp);
}

#[test]
fn test_auto_unregister() {
let mut suite = TokenDaoVoteDelegationTestingSuite::new().build();

suite.register(ADDR0);

suite.advance_block();

suite.assert_delegate_registered(ADDR0, None);

// unstake all tokens, which should automatically unregister the delegate
suite.unstake(ADDR0, suite.members[0].amount);

suite.advance_block();

suite.assert_delegate_not_registered(ADDR0, None);
}

/// this test does not actually test gas limits, since cw-multi-test does not
/// run a real chain, but it is demonstrative of what behaviors may lead to high
/// gas usage. this test is replicated in the DAO DAO UI codebase using an
Expand Down

0 comments on commit ba32eee

Please sign in to comment.