solana-program-library/governance/program/tests/process_execute_transaction.rs

750 lines
20 KiB
Rust

#![cfg(feature = "test-sbf")]
mod program_test;
use solana_program::{
instruction::{AccountMeta, Instruction},
program_error::ProgramError,
sysvar::clock,
};
use solana_program_test::tokio;
use program_test::*;
use spl_governance::{
error::GovernanceError,
state::enums::{ProposalState, TransactionExecutionStatus},
};
#[tokio::test]
async fn test_execute_mint_transaction() {
// Arrange
let mut governance_test = GovernanceProgramTest::start_new().await;
let realm_cookie = governance_test.with_realm().await;
let governed_mint_cookie = governance_test.with_governed_mint().await;
let token_owner_record_cookie = governance_test
.with_community_token_deposit(&realm_cookie)
.await
.unwrap();
let mut mint_governance_cookie = governance_test
.with_mint_governance(
&realm_cookie,
&governed_mint_cookie,
&token_owner_record_cookie,
)
.await
.unwrap();
let mut proposal_cookie = governance_test
.with_proposal(&token_owner_record_cookie, &mut mint_governance_cookie)
.await
.unwrap();
let signatory_record_cookie = governance_test
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
.await
.unwrap();
let proposal_transaction_cookie = governance_test
.with_mint_tokens_transaction(
&governed_mint_cookie,
&mut proposal_cookie,
&token_owner_record_cookie,
0,
None,
None,
)
.await
.unwrap();
governance_test
.sign_off_proposal(&proposal_cookie, &signatory_record_cookie)
.await
.unwrap();
governance_test
.with_cast_yes_no_vote(&proposal_cookie, &token_owner_record_cookie, YesNoVote::Yes)
.await
.unwrap();
// Advance timestamp past hold_up_time
governance_test
.advance_clock_by_min_timespan(proposal_transaction_cookie.account.hold_up_time as u64)
.await;
let clock = governance_test.bench.get_clock().await;
// Act
governance_test
.execute_proposal_transaction(&proposal_cookie, &proposal_transaction_cookie)
.await
.unwrap();
// Assert
let proposal_account = governance_test
.get_proposal_account(&proposal_cookie.address)
.await;
let yes_option = proposal_account.options.first().unwrap();
assert_eq!(1, yes_option.transactions_executed_count);
assert_eq!(ProposalState::Completed, proposal_account.state);
assert_eq!(Some(clock.unix_timestamp), proposal_account.closed_at);
assert_eq!(Some(clock.unix_timestamp), proposal_account.executing_at);
let proposal_transaction_account = governance_test
.get_proposal_transaction_account(&proposal_transaction_cookie.address)
.await;
assert_eq!(
Some(clock.unix_timestamp),
proposal_transaction_account.executed_at
);
assert_eq!(
TransactionExecutionStatus::Success,
proposal_transaction_account.execution_status
);
let instruction_token_account = governance_test
.get_token_account(&proposal_transaction_cookie.account.instructions[0].accounts[1].pubkey)
.await;
assert_eq!(10, instruction_token_account.amount);
}
#[tokio::test]
async fn test_execute_transfer_transaction() {
// Arrange
let mut governance_test = GovernanceProgramTest::start_new().await;
let realm_cookie = governance_test.with_realm().await;
let governed_token_cookie = governance_test.with_governed_token().await;
let token_owner_record_cookie = governance_test
.with_community_token_deposit(&realm_cookie)
.await
.unwrap();
let mut token_governance_cookie = governance_test
.with_token_governance(
&realm_cookie,
&governed_token_cookie,
&token_owner_record_cookie,
)
.await
.unwrap();
let mut proposal_cookie = governance_test
.with_proposal(&token_owner_record_cookie, &mut token_governance_cookie)
.await
.unwrap();
let signatory_record_cookie = governance_test
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
.await
.unwrap();
let proposal_transaction_cookie = governance_test
.with_transfer_tokens_transaction(
&governed_token_cookie,
&mut proposal_cookie,
&token_owner_record_cookie,
None,
)
.await
.unwrap();
governance_test
.sign_off_proposal(&proposal_cookie, &signatory_record_cookie)
.await
.unwrap();
governance_test
.with_cast_yes_no_vote(&proposal_cookie, &token_owner_record_cookie, YesNoVote::Yes)
.await
.unwrap();
// Advance timestamp past hold_up_time
governance_test
.advance_clock_by_min_timespan(proposal_transaction_cookie.account.hold_up_time as u64)
.await;
let clock = governance_test.bench.get_clock().await;
// Act
governance_test
.execute_proposal_transaction(&proposal_cookie, &proposal_transaction_cookie)
.await
.unwrap();
// Assert
let proposal_account = governance_test
.get_proposal_account(&proposal_cookie.address)
.await;
let yes_option = proposal_account.options.first().unwrap();
assert_eq!(1, yes_option.transactions_executed_count);
assert_eq!(ProposalState::Completed, proposal_account.state);
assert_eq!(Some(clock.unix_timestamp), proposal_account.closed_at);
assert_eq!(Some(clock.unix_timestamp), proposal_account.executing_at);
let proposal_transaction_account = governance_test
.get_proposal_transaction_account(&proposal_transaction_cookie.address)
.await;
assert_eq!(
Some(clock.unix_timestamp),
proposal_transaction_account.executed_at
);
assert_eq!(
TransactionExecutionStatus::Success,
proposal_transaction_account.execution_status
);
let instruction_token_account = governance_test
.get_token_account(&proposal_transaction_cookie.account.instructions[0].accounts[1].pubkey)
.await;
assert_eq!(15, instruction_token_account.amount);
}
#[tokio::test]
async fn test_execute_upgrade_program_transaction() {
// Arrange
let mut governance_test = GovernanceProgramTest::start_new().await;
let realm_cookie = governance_test.with_realm().await;
let governed_program_cookie = governance_test.with_governed_program().await;
let token_owner_record_cookie = governance_test
.with_community_token_deposit(&realm_cookie)
.await
.unwrap();
let mut program_governance_cookie = governance_test
.with_program_governance(
&realm_cookie,
&governed_program_cookie,
&token_owner_record_cookie,
)
.await
.unwrap();
let mut proposal_cookie = governance_test
.with_proposal(&token_owner_record_cookie, &mut program_governance_cookie)
.await
.unwrap();
let signatory_record_cookie = governance_test
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
.await
.unwrap();
let proposal_transaction_cookie = governance_test
.with_upgrade_program_transaction(
&program_governance_cookie,
&mut proposal_cookie,
&token_owner_record_cookie,
)
.await
.unwrap();
governance_test
.sign_off_proposal(&proposal_cookie, &signatory_record_cookie)
.await
.unwrap();
governance_test
.with_cast_yes_no_vote(&proposal_cookie, &token_owner_record_cookie, YesNoVote::Yes)
.await
.unwrap();
// Advance timestamp past hold_up_time
governance_test
.advance_clock_by_min_timespan(proposal_transaction_cookie.account.hold_up_time as u64)
.await;
// Ensure we can invoke the governed program before upgrade
let governed_program_ix = Instruction::new_with_bytes(
governed_program_cookie.address,
&[0],
vec![AccountMeta::new(clock::id(), false)],
);
let err = governance_test
.bench
.process_transaction(&[governed_program_ix.clone()], None)
.await
.err()
.unwrap();
// solana_bpf_rust_upgradable returns CustomError == 42
assert_eq!(ProgramError::Custom(42), err);
let clock = governance_test.bench.get_clock().await;
// Act
governance_test
.execute_proposal_transaction(&proposal_cookie, &proposal_transaction_cookie)
.await
.unwrap();
// Assert
let proposal_account = governance_test
.get_proposal_account(&proposal_cookie.address)
.await;
let yes_option = proposal_account.options.first().unwrap();
assert_eq!(1, yes_option.transactions_executed_count);
assert_eq!(ProposalState::Completed, proposal_account.state);
assert_eq!(Some(clock.unix_timestamp), proposal_account.closed_at);
assert_eq!(Some(clock.unix_timestamp), proposal_account.executing_at);
let proposal_transaction_account = governance_test
.get_proposal_transaction_account(&proposal_transaction_cookie.address)
.await;
assert_eq!(
Some(clock.unix_timestamp),
proposal_transaction_account.executed_at
);
assert_eq!(
TransactionExecutionStatus::Success,
proposal_transaction_account.execution_status
);
// Assert we can invoke the governed program after upgrade
governance_test.advance_clock().await;
let err = governance_test
.bench
.process_transaction(&[governed_program_ix.clone()], None)
.await
.err()
.unwrap();
// solana_bpf_rust_upgraded returns CustomError == 43
assert_eq!(ProgramError::Custom(43), err);
// --------------------------- !!! Voila !!! -----------------------------
}
#[tokio::test]
#[ignore]
async fn test_execute_proposal_transaction_with_invalid_state_errors() {
// Arrange
let mut governance_test = GovernanceProgramTest::start_new().await;
let realm_cookie = governance_test.with_realm().await;
let governed_mint_cookie = governance_test.with_governed_mint().await;
let token_owner_record_cookie = governance_test
.with_community_token_deposit(&realm_cookie)
.await
.unwrap();
let mut mint_governance_cookie = governance_test
.with_mint_governance(
&realm_cookie,
&governed_mint_cookie,
&token_owner_record_cookie,
)
.await
.unwrap();
let mut proposal_cookie = governance_test
.with_proposal(&token_owner_record_cookie, &mut mint_governance_cookie)
.await
.unwrap();
let signatory_record_cookie1 = governance_test
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
.await
.unwrap();
let signatory_record_cookie2 = governance_test
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
.await
.unwrap();
let proposal_transaction_cookie = governance_test
.with_mint_tokens_transaction(
&governed_mint_cookie,
&mut proposal_cookie,
&token_owner_record_cookie,
0,
None,
None,
)
.await
.unwrap();
// Act
let err = governance_test
.execute_proposal_transaction(&proposal_cookie, &proposal_transaction_cookie)
.await
.err()
.unwrap();
// Assert
assert_eq!(
err,
GovernanceError::InvalidStateCannotExecuteTransaction.into()
);
// Arrange
governance_test
.sign_off_proposal(&proposal_cookie, &signatory_record_cookie1)
.await
.unwrap();
// Act
let err = governance_test
.execute_proposal_transaction(&proposal_cookie, &proposal_transaction_cookie)
.await
.err()
.unwrap();
// Assert
assert_eq!(
err,
GovernanceError::InvalidStateCannotExecuteTransaction.into()
);
// Arrange
governance_test
.sign_off_proposal(&proposal_cookie, &signatory_record_cookie2)
.await
.unwrap();
// Act
let err = governance_test
.execute_proposal_transaction(&proposal_cookie, &proposal_transaction_cookie)
.await
.err()
.unwrap();
// Assert
assert_eq!(
err,
GovernanceError::InvalidStateCannotExecuteTransaction.into()
);
// Arrange
governance_test
.with_cast_yes_no_vote(&proposal_cookie, &token_owner_record_cookie, YesNoVote::Yes)
.await
.unwrap();
governance_test.advance_clock().await;
// Act
let err = governance_test
.execute_proposal_transaction(&proposal_cookie, &proposal_transaction_cookie)
.await
.err()
.unwrap();
// Assert
assert_eq!(
err,
GovernanceError::CannotExecuteTransactionWithinHoldUpTime.into()
);
// Arrange
// Advance timestamp past hold_up_time
governance_test
.advance_clock_by_min_timespan(proposal_transaction_cookie.account.hold_up_time as u64)
.await;
// Act
governance_test
.execute_proposal_transaction(&proposal_cookie, &proposal_transaction_cookie)
.await
.unwrap();
// Assert
let proposal_account = governance_test
.get_proposal_account(&proposal_cookie.address)
.await;
assert_eq!(ProposalState::Completed, proposal_account.state);
// Arrange
governance_test.advance_clock().await;
// Act
let err = governance_test
.execute_proposal_transaction(&proposal_cookie, &proposal_transaction_cookie)
.await
.err()
.unwrap();
// Assert
assert_eq!(
err,
GovernanceError::InvalidStateCannotExecuteTransaction.into()
);
}
#[tokio::test]
async fn test_execute_proposal_transaction_for_other_proposal_error() {
// Arrange
let mut governance_test = GovernanceProgramTest::start_new().await;
let realm_cookie = governance_test.with_realm().await;
let governed_mint_cookie = governance_test.with_governed_mint().await;
let token_owner_record_cookie = governance_test
.with_community_token_deposit(&realm_cookie)
.await
.unwrap();
let mut mint_governance_cookie = governance_test
.with_mint_governance(
&realm_cookie,
&governed_mint_cookie,
&token_owner_record_cookie,
)
.await
.unwrap();
let mut proposal_cookie = governance_test
.with_proposal(&token_owner_record_cookie, &mut mint_governance_cookie)
.await
.unwrap();
let signatory_record_cookie = governance_test
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
.await
.unwrap();
let proposal_transaction_cookie = governance_test
.with_mint_tokens_transaction(
&governed_mint_cookie,
&mut proposal_cookie,
&token_owner_record_cookie,
0,
None,
None,
)
.await
.unwrap();
governance_test
.sign_off_proposal(&proposal_cookie, &signatory_record_cookie)
.await
.unwrap();
governance_test
.with_cast_yes_no_vote(&proposal_cookie, &token_owner_record_cookie, YesNoVote::Yes)
.await
.unwrap();
// Advance clock past hold_up_time
governance_test
.advance_clock_by_min_timespan(proposal_transaction_cookie.account.hold_up_time as u64)
.await;
let token_owner_record_cookie2 = governance_test
.with_community_token_deposit(&realm_cookie)
.await
.unwrap();
let proposal_cookie2 = governance_test
.with_proposal(&token_owner_record_cookie2, &mut mint_governance_cookie)
.await
.unwrap();
governance_test.advance_clock().await;
// Act
let err = governance_test
.execute_proposal_transaction(&proposal_cookie2, &proposal_transaction_cookie)
.await
.err()
.unwrap();
// Assert
assert_eq!(
err,
GovernanceError::InvalidProposalForProposalTransaction.into()
);
}
#[tokio::test]
async fn test_execute_mint_transaction_twice_error() {
// Arrange
let mut governance_test = GovernanceProgramTest::start_new().await;
let realm_cookie = governance_test.with_realm().await;
let governed_mint_cookie = governance_test.with_governed_mint().await;
let token_owner_record_cookie = governance_test
.with_community_token_deposit(&realm_cookie)
.await
.unwrap();
let mut mint_governance_cookie = governance_test
.with_mint_governance(
&realm_cookie,
&governed_mint_cookie,
&token_owner_record_cookie,
)
.await
.unwrap();
let mut proposal_cookie = governance_test
.with_proposal(&token_owner_record_cookie, &mut mint_governance_cookie)
.await
.unwrap();
let signatory_record_cookie = governance_test
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
.await
.unwrap();
let proposal_transaction_cookie = governance_test
.with_mint_tokens_transaction(
&governed_mint_cookie,
&mut proposal_cookie,
&token_owner_record_cookie,
0,
None,
None,
)
.await
.unwrap();
governance_test
.with_nop_transaction(&mut proposal_cookie, &token_owner_record_cookie, 0, None)
.await
.unwrap();
governance_test
.sign_off_proposal(&proposal_cookie, &signatory_record_cookie)
.await
.unwrap();
governance_test
.with_cast_yes_no_vote(&proposal_cookie, &token_owner_record_cookie, YesNoVote::Yes)
.await
.unwrap();
// Advance clock past hold_up_time
governance_test
.advance_clock_by_min_timespan(proposal_transaction_cookie.account.hold_up_time as u64)
.await;
governance_test
.execute_proposal_transaction(&proposal_cookie, &proposal_transaction_cookie)
.await
.unwrap();
governance_test.advance_clock().await;
// Act
let err = governance_test
.execute_proposal_transaction(&proposal_cookie, &proposal_transaction_cookie)
.await
.err()
.unwrap();
// Assert
assert_eq!(err, GovernanceError::TransactionAlreadyExecuted.into());
}
#[tokio::test]
async fn test_execute_transaction_with_create_proposal_and_execute_in_single_slot_error() {
// Arrange
let mut governance_test = GovernanceProgramTest::start_new().await;
let realm_cookie = governance_test.with_realm().await;
let governed_mint_cookie = governance_test.with_governed_mint().await;
let token_owner_record_cookie = governance_test
.with_community_token_deposit(&realm_cookie)
.await
.unwrap();
let mut governance_config = governance_test.get_default_governance_config();
governance_config.min_transaction_hold_up_time = 0;
let mut mint_governance_cookie = governance_test
.with_mint_governance_using_config(
&realm_cookie,
&governed_mint_cookie,
&token_owner_record_cookie,
&governance_config,
)
.await
.unwrap();
let mut proposal_cookie = governance_test
.with_proposal(&token_owner_record_cookie, &mut mint_governance_cookie)
.await
.unwrap();
let signatory_record_cookie = governance_test
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
.await
.unwrap();
let proposal_transaction_cookie = governance_test
.with_mint_tokens_transaction(
&governed_mint_cookie,
&mut proposal_cookie,
&token_owner_record_cookie,
0,
None,
Some(0),
)
.await
.unwrap();
governance_test
.sign_off_proposal(&proposal_cookie, &signatory_record_cookie)
.await
.unwrap();
governance_test
.with_cast_yes_no_vote(&proposal_cookie, &token_owner_record_cookie, YesNoVote::Yes)
.await
.unwrap();
// Act
let err = governance_test
.execute_proposal_transaction(&proposal_cookie, &proposal_transaction_cookie)
.await
.err()
.unwrap();
// Assert
assert_eq!(
err,
GovernanceError::CannotExecuteTransactionWithinHoldUpTime.into()
);
}