#![cfg(feature = "test-bpf")]
mod program_test;
use program_test::{tools::ProgramInstructionError, *};
use solana_program_test::tokio;
use solana_sdk::{signature::Keypair, signer::Signer};
use spl_governance::{
error::GovernanceError,
instruction::{set_governance_config, Vote},
state::enums::VoteThresholdPercentage,
};
#[tokio::test]
async fn test_set_governance_config() {
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 mut proposal_cookie = governance_test
.with_proposal(&token_owner_record_cookie, &mut account_governance_cookie)
.await
.unwrap();
let signatory_record_cookie = governance_test
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
.await
.unwrap();
let mut new_governance_config = governance_test.get_default_governance_config();
new_governance_config.vote_threshold_percentage = VoteThresholdPercentage::YesVote(40);
let proposal_instruction_cookie = governance_test
.with_set_governance_config_instruction(
&mut proposal_cookie,
&token_owner_record_cookie,
&new_governance_config,
)
.await
.unwrap();
governance_test
.sign_off_proposal(&proposal_cookie, &signatory_record_cookie)
.await
.unwrap();
governance_test
.with_cast_vote(&proposal_cookie, &token_owner_record_cookie, Vote::Yes)
.await
.unwrap();
governance_test
.advance_clock_by_min_timespan(proposal_instruction_cookie.account.hold_up_time as u64)
.await;
governance_test
.execute_instruction(&proposal_cookie, &proposal_instruction_cookie)
.await
.unwrap();
let governance_account = governance_test
.get_governance_account(&account_governance_cookie.address)
.await;
assert_eq!(new_governance_config, governance_account.config);
}
#[tokio::test]
async fn test_set_governance_config_with_governance_must_sign_error() {
let mut governance_test = GovernanceProgramTest::start_new().await;
let realm_cookie = governance_test.with_realm().await;
let new_governance_config = governance_test.get_default_governance_config();
let mut set_governance_config_ix = set_governance_config(
&governance_test.program_id,
&realm_cookie.address,
new_governance_config.clone(),
);
set_governance_config_ix.accounts[0].is_signer = false;
let err = governance_test
.process_transaction(&[set_governance_config_ix], None)
.await
.err()
.unwrap();
assert_eq!(err, GovernanceError::GovernancePdaMustSign.into());
}
#[tokio::test]
async fn test_set_governance_config_with_fake_governance_signer_error() {
let mut governance_test = GovernanceProgramTest::start_new().await;
let realm_cookie = governance_test.with_realm().await;
let new_governance_config = governance_test.get_default_governance_config();
let mut set_governance_config_ix = set_governance_config(
&governance_test.program_id,
&realm_cookie.address,
new_governance_config.clone(),
);
let governance_signer = Keypair::new();
set_governance_config_ix.accounts[0].pubkey = governance_signer.pubkey();
let err = governance_test
.process_transaction(&[set_governance_config_ix], Some(&[&governance_signer]))
.await
.err()
.unwrap();
assert_eq!(err, GovernanceError::AccountDoesNotExist.into());
}
#[tokio::test]
async fn test_set_governance_config_with_invalid_governance_authority_error() {
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 mut proposal_cookie = governance_test
.with_proposal(&token_owner_record_cookie, &mut account_governance_cookie)
.await
.unwrap();
let signatory_record_cookie = governance_test
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
.await
.unwrap();
let governed_account_cookie2 = governance_test.with_governed_account().await;
let account_governance_cookie2 = governance_test
.with_account_governance(&realm_cookie, &governed_account_cookie2)
.await
.unwrap();
let new_governance_config = governance_test.get_default_governance_config();
let mut set_governance_config_ix = set_governance_config(
&governance_test.program_id,
&account_governance_cookie2.address,
new_governance_config,
);
let proposal_instruction_cookie = governance_test
.with_instruction(
&mut proposal_cookie,
&token_owner_record_cookie,
None,
&mut set_governance_config_ix,
)
.await
.unwrap();
governance_test
.sign_off_proposal(&proposal_cookie, &signatory_record_cookie)
.await
.unwrap();
governance_test
.with_cast_vote(&proposal_cookie, &token_owner_record_cookie, Vote::Yes)
.await
.unwrap();
governance_test
.advance_clock_by_min_timespan(proposal_instruction_cookie.account.hold_up_time as u64)
.await;
let err = governance_test
.execute_instruction(&proposal_cookie, &proposal_instruction_cookie)
.await
.err()
.unwrap();
assert_eq!(err, ProgramInstructionError::PrivilegeEscalation.into());
}