131 lines
3.7 KiB
Rust
131 lines
3.7 KiB
Rust
#![cfg(feature = "test-bpf")]
|
|
|
|
mod program_test;
|
|
|
|
use solana_program_test::tokio;
|
|
|
|
use program_test::*;
|
|
use spl_governance::{error::GovernanceError, instruction::Vote, state::enums::ProposalState};
|
|
|
|
#[tokio::test]
|
|
async fn test_cancel_proposal() {
|
|
// Arrange
|
|
let mut governance_test = GovernanceProgramTest::start_new().await;
|
|
|
|
let realm_cookie = governance_test.with_realm().await;
|
|
let governed_account_cookie = governance_test.with_governed_account().await;
|
|
|
|
let mut account_governance_cookie = governance_test
|
|
.with_account_governance(&realm_cookie, &governed_account_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
let token_owner_record_cookie = governance_test
|
|
.with_community_token_deposit(&realm_cookie)
|
|
.await;
|
|
|
|
let proposal_cookie = governance_test
|
|
.with_proposal(&token_owner_record_cookie, &mut account_governance_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Act
|
|
governance_test
|
|
.cancel_proposal(&proposal_cookie, &token_owner_record_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Assert
|
|
let proposal_account = governance_test
|
|
.get_proposal_account(&proposal_cookie.address)
|
|
.await;
|
|
|
|
assert_eq!(ProposalState::Cancelled, proposal_account.state);
|
|
assert_eq!(Some(1), proposal_account.closed_at);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_cancel_proposal_with_already_completed_error() {
|
|
// Arrange
|
|
let mut governance_test = GovernanceProgramTest::start_new().await;
|
|
|
|
let realm_cookie = governance_test.with_realm().await;
|
|
let governed_account_cookie = governance_test.with_governed_account().await;
|
|
|
|
let mut account_governance_cookie = governance_test
|
|
.with_account_governance(&realm_cookie, &governed_account_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
let token_owner_record_cookie = governance_test
|
|
.with_community_token_deposit(&realm_cookie)
|
|
.await;
|
|
|
|
let proposal_cookie = governance_test
|
|
.with_signed_off_proposal(&token_owner_record_cookie, &mut account_governance_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
governance_test
|
|
.with_cast_vote(&proposal_cookie, &token_owner_record_cookie, Vote::Yes)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Act
|
|
let err = governance_test
|
|
.cancel_proposal(&proposal_cookie, &token_owner_record_cookie)
|
|
.await
|
|
.err()
|
|
.unwrap();
|
|
|
|
// Assert
|
|
|
|
assert_eq!(
|
|
err,
|
|
GovernanceError::InvalidStateCannotCancelProposal.into()
|
|
);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_cancel_proposal_with_owner_or_delegate_must_sign_error() {
|
|
// Arrange
|
|
let mut governance_test = GovernanceProgramTest::start_new().await;
|
|
|
|
let realm_cookie = governance_test.with_realm().await;
|
|
let governed_account_cookie = governance_test.with_governed_account().await;
|
|
|
|
let mut account_governance_cookie = governance_test
|
|
.with_account_governance(&realm_cookie, &governed_account_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
let mut token_owner_record_cookie = governance_test
|
|
.with_community_token_deposit(&realm_cookie)
|
|
.await;
|
|
|
|
let proposal_cookie = governance_test
|
|
.with_proposal(&token_owner_record_cookie, &mut account_governance_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
let token_owner_record_cookie2 = governance_test
|
|
.with_council_token_deposit(&realm_cookie)
|
|
.await;
|
|
|
|
token_owner_record_cookie.token_owner = token_owner_record_cookie2.token_owner;
|
|
|
|
// Act
|
|
let err = governance_test
|
|
.cancel_proposal(&proposal_cookie, &token_owner_record_cookie)
|
|
.await
|
|
.err()
|
|
.unwrap();
|
|
|
|
// Assert
|
|
|
|
assert_eq!(
|
|
err,
|
|
GovernanceError::GoverningTokenOwnerOrDelegateMustSign.into()
|
|
);
|
|
}
|