540 lines
14 KiB
Rust
540 lines
14 KiB
Rust
#![cfg(feature = "test-bpf")]
|
|
|
|
mod program_test;
|
|
|
|
use solana_program::{
|
|
instruction::{AccountMeta, Instruction},
|
|
program_error::ProgramError,
|
|
sysvar::{clock, fees},
|
|
};
|
|
use solana_program_test::tokio;
|
|
|
|
use program_test::*;
|
|
use spl_governance::{error::GovernanceError, instruction::Vote, state::enums::ProposalState};
|
|
|
|
#[tokio::test]
|
|
async fn test_execute_mint_instruction() {
|
|
// 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 mut mint_governance_cookie = governance_test
|
|
.with_mint_governance(&realm_cookie, &governed_mint_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 mint_governance_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
let signatory_record_cookie = governance_test
|
|
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
let proposal_instruction_cookie = governance_test
|
|
.with_mint_tokens_instruction(
|
|
&governed_mint_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_vote(&proposal_cookie, &token_owner_record_cookie, Vote::Yes)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Advance slot past hold_up_time
|
|
let execute_at_slot = 1 + proposal_instruction_cookie.account.hold_up_time + 1;
|
|
|
|
governance_test
|
|
.context
|
|
.warp_to_slot(execute_at_slot)
|
|
.unwrap();
|
|
|
|
// Act
|
|
governance_test
|
|
.execute_instruction(&proposal_cookie, &proposal_instruction_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Assert
|
|
|
|
let proposal_account = governance_test
|
|
.get_proposal_account(&proposal_cookie.address)
|
|
.await;
|
|
|
|
assert_eq!(1, proposal_account.instructions_executed_count);
|
|
assert_eq!(ProposalState::Completed, proposal_account.state);
|
|
assert_eq!(Some(execute_at_slot), proposal_account.closed_at);
|
|
assert_eq!(Some(execute_at_slot), proposal_account.executing_at);
|
|
|
|
let proposal_instruction_account = governance_test
|
|
.get_proposal_instruction_account(&proposal_instruction_cookie.address)
|
|
.await;
|
|
|
|
assert_eq!(
|
|
Some(execute_at_slot),
|
|
proposal_instruction_account.executed_at
|
|
);
|
|
|
|
let instruction_token_account = governance_test
|
|
.get_token_account(&proposal_instruction_cookie.account.instruction.accounts[1].pubkey)
|
|
.await;
|
|
|
|
assert_eq!(10, instruction_token_account.amount);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_execute_upgrade_program_instruction() {
|
|
// 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 mut program_governance_cookie = governance_test
|
|
.with_program_governance(&realm_cookie, &governed_program_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 program_governance_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
let signatory_record_cookie = governance_test
|
|
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
let proposal_instruction_cookie = governance_test
|
|
.with_upgrade_program_instruction(
|
|
&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_vote(&proposal_cookie, &token_owner_record_cookie, Vote::Yes)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Advance slot past hold_up_time
|
|
let execute_at_slot = 1 + proposal_instruction_cookie.account.hold_up_time + 1;
|
|
|
|
governance_test
|
|
.context
|
|
.warp_to_slot(execute_at_slot)
|
|
.unwrap();
|
|
|
|
// Ensure we can invoke the governed program before upgrade
|
|
let governed_program_instruction = Instruction::new_with_bytes(
|
|
governed_program_cookie.address,
|
|
&[0],
|
|
vec![
|
|
AccountMeta::new(governed_program_cookie.address, false),
|
|
AccountMeta::new(clock::id(), false),
|
|
AccountMeta::new(fees::id(), false),
|
|
],
|
|
);
|
|
|
|
let err = governance_test
|
|
.process_transaction(&[governed_program_instruction.clone()], None)
|
|
.await
|
|
.err()
|
|
.unwrap();
|
|
|
|
// solana_bpf_rust_upgradable returns CustomError == 42
|
|
assert_eq!(ProgramError::Custom(42), err);
|
|
|
|
// Act
|
|
governance_test
|
|
.execute_instruction(&proposal_cookie, &proposal_instruction_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Assert
|
|
|
|
let proposal_account = governance_test
|
|
.get_proposal_account(&proposal_cookie.address)
|
|
.await;
|
|
|
|
assert_eq!(1, proposal_account.instructions_executed_count);
|
|
assert_eq!(ProposalState::Completed, proposal_account.state);
|
|
assert_eq!(Some(execute_at_slot), proposal_account.closed_at);
|
|
assert_eq!(Some(execute_at_slot), proposal_account.executing_at);
|
|
|
|
let proposal_instruction_account = governance_test
|
|
.get_proposal_instruction_account(&proposal_instruction_cookie.address)
|
|
.await;
|
|
|
|
assert_eq!(
|
|
Some(execute_at_slot),
|
|
proposal_instruction_account.executed_at
|
|
);
|
|
|
|
// Assert we can invoke the governed program after upgrade
|
|
governance_test
|
|
.context
|
|
.warp_to_slot(execute_at_slot + 10)
|
|
.unwrap();
|
|
|
|
let err = governance_test
|
|
.process_transaction(&[governed_program_instruction.clone()], None)
|
|
.await
|
|
.err()
|
|
.unwrap();
|
|
|
|
// solana_bpf_rust_upgraded returns CustomError == 43
|
|
assert_eq!(ProgramError::Custom(43), err);
|
|
|
|
// --------------------------- !!! Voila !!! -----------------------------
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_execute_instruction_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 mut mint_governance_cookie = governance_test
|
|
.with_mint_governance(&realm_cookie, &governed_mint_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 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_instruction_cookie = governance_test
|
|
.with_mint_tokens_instruction(
|
|
&governed_mint_cookie,
|
|
&mut proposal_cookie,
|
|
&token_owner_record_cookie,
|
|
None,
|
|
)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Act
|
|
|
|
let err = governance_test
|
|
.execute_instruction(&proposal_cookie, &proposal_instruction_cookie)
|
|
.await
|
|
.err()
|
|
.unwrap();
|
|
|
|
// Assert
|
|
assert_eq!(
|
|
err,
|
|
GovernanceError::InvalidStateCannotExecuteInstruction.into()
|
|
);
|
|
|
|
// Arrange
|
|
|
|
governance_test
|
|
.sign_off_proposal(&proposal_cookie, &signatory_record_cookie1)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Act
|
|
|
|
let err = governance_test
|
|
.execute_instruction(&proposal_cookie, &proposal_instruction_cookie)
|
|
.await
|
|
.err()
|
|
.unwrap();
|
|
|
|
// Assert
|
|
assert_eq!(
|
|
err,
|
|
GovernanceError::InvalidStateCannotExecuteInstruction.into()
|
|
);
|
|
|
|
// Arrange
|
|
|
|
governance_test
|
|
.sign_off_proposal(&proposal_cookie, &signatory_record_cookie2)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Act
|
|
|
|
let err = governance_test
|
|
.execute_instruction(&proposal_cookie, &proposal_instruction_cookie)
|
|
.await
|
|
.err()
|
|
.unwrap();
|
|
|
|
// Assert
|
|
assert_eq!(
|
|
err,
|
|
GovernanceError::InvalidStateCannotExecuteInstruction.into()
|
|
);
|
|
|
|
// Arrange
|
|
|
|
governance_test
|
|
.with_cast_vote(&proposal_cookie, &token_owner_record_cookie, Vote::Yes)
|
|
.await
|
|
.unwrap();
|
|
|
|
governance_test.context.warp_to_slot(5).unwrap();
|
|
|
|
// Act
|
|
let err = governance_test
|
|
.execute_instruction(&proposal_cookie, &proposal_instruction_cookie)
|
|
.await
|
|
.err()
|
|
.unwrap();
|
|
|
|
// Assert
|
|
assert_eq!(
|
|
err,
|
|
GovernanceError::CannotExecuteInstructionWithinHoldUpTime.into()
|
|
);
|
|
|
|
// Arrange
|
|
// Advance slot past hold_up_time
|
|
let execute_at_slot = 1 + proposal_instruction_cookie.account.hold_up_time + 1;
|
|
|
|
governance_test
|
|
.context
|
|
.warp_to_slot(execute_at_slot)
|
|
.unwrap();
|
|
|
|
// Act
|
|
governance_test
|
|
.execute_instruction(&proposal_cookie, &proposal_instruction_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
|
|
.context
|
|
.warp_to_slot(execute_at_slot + 10)
|
|
.unwrap();
|
|
|
|
// Act
|
|
let err = governance_test
|
|
.execute_instruction(&proposal_cookie, &proposal_instruction_cookie)
|
|
.await
|
|
.err()
|
|
.unwrap();
|
|
|
|
// Assert
|
|
assert_eq!(
|
|
err,
|
|
GovernanceError::InvalidStateCannotExecuteInstruction.into()
|
|
);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_execute_instruction_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 mut mint_governance_cookie = governance_test
|
|
.with_mint_governance(&realm_cookie, &governed_mint_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 mint_governance_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
let signatory_record_cookie = governance_test
|
|
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
let proposal_instruction_cookie = governance_test
|
|
.with_mint_tokens_instruction(
|
|
&governed_mint_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_vote(&proposal_cookie, &token_owner_record_cookie, Vote::Yes)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Advance slot past hold_up_time
|
|
let execute_at_slot = 1 + proposal_instruction_cookie.account.hold_up_time + 1;
|
|
|
|
governance_test
|
|
.context
|
|
.warp_to_slot(execute_at_slot)
|
|
.unwrap();
|
|
|
|
let proposal_cookie2 = governance_test
|
|
.with_proposal(&token_owner_record_cookie, &mut mint_governance_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Act
|
|
let err = governance_test
|
|
.execute_instruction(&proposal_cookie2, &proposal_instruction_cookie)
|
|
.await
|
|
.err()
|
|
.unwrap();
|
|
|
|
// Assert
|
|
assert_eq!(
|
|
err,
|
|
GovernanceError::InvalidProposalForProposalInstruction.into()
|
|
);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_execute_mint_instruction_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 mut mint_governance_cookie = governance_test
|
|
.with_mint_governance(&realm_cookie, &governed_mint_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 mint_governance_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
let signatory_record_cookie = governance_test
|
|
.with_signatory(&proposal_cookie, &token_owner_record_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
let proposal_instruction_cookie = governance_test
|
|
.with_mint_tokens_instruction(
|
|
&governed_mint_cookie,
|
|
&mut proposal_cookie,
|
|
&token_owner_record_cookie,
|
|
None,
|
|
)
|
|
.await
|
|
.unwrap();
|
|
|
|
governance_test
|
|
.with_instruction(&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_vote(&proposal_cookie, &token_owner_record_cookie, Vote::Yes)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Advance slot past hold_up_time
|
|
let execute_at_slot = 1 + proposal_instruction_cookie.account.hold_up_time + 1;
|
|
|
|
governance_test
|
|
.context
|
|
.warp_to_slot(execute_at_slot)
|
|
.unwrap();
|
|
|
|
governance_test
|
|
.execute_instruction(&proposal_cookie, &proposal_instruction_cookie)
|
|
.await
|
|
.unwrap();
|
|
|
|
governance_test
|
|
.context
|
|
.warp_to_slot(execute_at_slot + 10)
|
|
.unwrap();
|
|
|
|
// Act
|
|
|
|
let err = governance_test
|
|
.execute_instruction(&proposal_cookie, &proposal_instruction_cookie)
|
|
.await
|
|
.err()
|
|
.unwrap();
|
|
|
|
// Assert
|
|
assert_eq!(err, GovernanceError::InstructionAlreadyExecuted.into());
|
|
}
|