Tests for LockupKind::None
This commit is contained in:
Christian Kamm
parent
5b63ba8339
commit
1e10adb2f1
|
|
@ -31,6 +31,7 @@ pub struct ExchangeRateCookie {
|
|||
pub struct VoterCookie {
|
||||
pub address: Pubkey,
|
||||
pub authority: Pubkey,
|
||||
pub voter_weight_record: Pubkey,
|
||||
}
|
||||
|
||||
impl AddinCookie {
|
||||
|
|
@ -216,6 +217,7 @@ impl AddinCookie {
|
|||
VoterCookie {
|
||||
address: voter,
|
||||
authority: authority.pubkey(),
|
||||
voter_weight_record,
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -271,6 +273,54 @@ impl AddinCookie {
|
|||
.await
|
||||
}
|
||||
|
||||
pub async fn update_deposit(
|
||||
&self,
|
||||
registrar: &RegistrarCookie,
|
||||
voter: &VoterCookie,
|
||||
exchange_rate: &ExchangeRateCookie,
|
||||
authority: &Keypair,
|
||||
token_address: Pubkey,
|
||||
id: u8,
|
||||
amount: u64,
|
||||
) -> std::result::Result<(), TransportError> {
|
||||
let data =
|
||||
anchor_lang::InstructionData::data(&governance_registry::instruction::UpdateDeposit {
|
||||
id,
|
||||
amount,
|
||||
});
|
||||
|
||||
let accounts = anchor_lang::ToAccountMetas::to_account_metas(
|
||||
&governance_registry::accounts::UpdateDeposit {
|
||||
registrar: registrar.address,
|
||||
voter: voter.address,
|
||||
exchange_vault: exchange_rate.exchange_vault,
|
||||
deposit_token: token_address,
|
||||
voting_token: voter.voting_token(exchange_rate),
|
||||
authority: authority.pubkey(),
|
||||
deposit_mint: exchange_rate.deposit_mint.pubkey.unwrap(),
|
||||
voting_mint: exchange_rate.voting_mint,
|
||||
token_program: spl_token::id(),
|
||||
associated_token_program: spl_associated_token_account::id(),
|
||||
system_program: solana_sdk::system_program::id(),
|
||||
rent: solana_program::sysvar::rent::id(),
|
||||
},
|
||||
None,
|
||||
);
|
||||
|
||||
let instructions = vec![Instruction {
|
||||
program_id: self.program_id,
|
||||
accounts,
|
||||
data,
|
||||
}];
|
||||
|
||||
// clone the secrets
|
||||
let signer = Keypair::from_base58_string(&authority.to_base58_string());
|
||||
|
||||
self.solana
|
||||
.process_transaction(&instructions, Some(&[&signer]))
|
||||
.await
|
||||
}
|
||||
|
||||
pub async fn withdraw(
|
||||
&self,
|
||||
registrar: &RegistrarCookie,
|
||||
|
|
@ -317,6 +367,48 @@ impl AddinCookie {
|
|||
.process_transaction(&instructions, Some(&[&signer]))
|
||||
.await
|
||||
}
|
||||
|
||||
pub async fn update_voter_weight_record(
|
||||
&self,
|
||||
registrar: &RegistrarCookie,
|
||||
voter: &VoterCookie,
|
||||
authority: &Keypair,
|
||||
) -> std::result::Result<governance_registry::account::VoterWeightRecord, TransportError> {
|
||||
let data = anchor_lang::InstructionData::data(
|
||||
&governance_registry::instruction::UpdateVoterWeightRecord {},
|
||||
);
|
||||
|
||||
let accounts = anchor_lang::ToAccountMetas::to_account_metas(
|
||||
&governance_registry::accounts::UpdateVoterWeightRecord {
|
||||
registrar: registrar.address,
|
||||
voter: voter.address,
|
||||
voter_weight_record: voter.voter_weight_record,
|
||||
authority: authority.pubkey(),
|
||||
system_program: solana_sdk::system_program::id(),
|
||||
},
|
||||
None,
|
||||
);
|
||||
|
||||
let instructions = vec![Instruction {
|
||||
program_id: self.program_id,
|
||||
accounts,
|
||||
data,
|
||||
}];
|
||||
|
||||
// clone the secrets
|
||||
let signer = Keypair::from_base58_string(&authority.to_base58_string());
|
||||
|
||||
self.solana
|
||||
.process_transaction(&instructions, Some(&[&signer]))
|
||||
.await?;
|
||||
|
||||
Ok(self
|
||||
.solana
|
||||
.get_account::<governance_registry::account::VoterWeightRecord>(
|
||||
voter.voter_weight_record,
|
||||
)
|
||||
.await)
|
||||
}
|
||||
}
|
||||
|
||||
impl ExchangeRateCookie {
|
||||
|
|
|
|||
|
|
@ -0,0 +1,280 @@
|
|||
use anchor_spl::token::TokenAccount;
|
||||
use solana_program_test::*;
|
||||
use solana_sdk::{pubkey::Pubkey, signature::Keypair, signer::Signer, transport::TransportError};
|
||||
|
||||
use program_test::*;
|
||||
|
||||
mod program_test;
|
||||
|
||||
struct Balances {
|
||||
token: u64,
|
||||
vault: u64,
|
||||
deposit: u64,
|
||||
voter_weight: u64,
|
||||
}
|
||||
|
||||
async fn balances(
|
||||
context: &TestContext,
|
||||
registrar: &RegistrarCookie,
|
||||
address: Pubkey,
|
||||
voter: &VoterCookie,
|
||||
voter_authority: &Keypair,
|
||||
rate: &ExchangeRateCookie,
|
||||
deposit_id: u8,
|
||||
) -> Balances {
|
||||
// Advance slots to avoid caching of the UpdateVoterWeightRecord call
|
||||
// TODO: Is this something that could be an issue on a live node?
|
||||
context.solana.advance_clock_by_slots(2).await;
|
||||
|
||||
let token = context.solana.token_account_balance(address).await;
|
||||
let vault = rate.vault_balance(&context.solana).await;
|
||||
let deposit = voter.deposit_amount(&context.solana, deposit_id).await;
|
||||
let vwr = context
|
||||
.addin
|
||||
.update_voter_weight_record(®istrar, &voter, &voter_authority)
|
||||
.await
|
||||
.unwrap();
|
||||
Balances {
|
||||
token,
|
||||
vault,
|
||||
deposit,
|
||||
voter_weight: vwr.voter_weight,
|
||||
}
|
||||
}
|
||||
|
||||
#[allow(unaligned_references)]
|
||||
#[tokio::test]
|
||||
async fn test_deposit_no_locking() -> Result<(), TransportError> {
|
||||
let context = TestContext::new().await;
|
||||
|
||||
let payer = &context.users[0].key;
|
||||
let realm_authority = Keypair::new();
|
||||
let realm = context
|
||||
.governance
|
||||
.create_realm(
|
||||
"testrealm",
|
||||
realm_authority.pubkey(),
|
||||
&context.mints[0],
|
||||
&payer,
|
||||
&context.addin.program_id,
|
||||
)
|
||||
.await;
|
||||
|
||||
let voter_authority = &context.users[1].key;
|
||||
let token_owner_record = realm
|
||||
.create_token_owner_record(voter_authority.pubkey(), &payer)
|
||||
.await;
|
||||
|
||||
let registrar = context.addin.create_registrar(&realm, payer).await;
|
||||
let mngo_rate = context
|
||||
.addin
|
||||
.create_exchange_rate(®istrar, &realm_authority, payer, 0, &context.mints[0], 1)
|
||||
.await;
|
||||
|
||||
let voter = context
|
||||
.addin
|
||||
.create_voter(®istrar, &voter_authority, &payer)
|
||||
.await;
|
||||
|
||||
let voter2_authority = &context.users[2].key;
|
||||
let voter2 = context
|
||||
.addin
|
||||
.create_voter(®istrar, &voter2_authority, &payer)
|
||||
.await;
|
||||
|
||||
let reference_account = context.users[1].token_accounts[0];
|
||||
let get_balances = |depot_id| {
|
||||
balances(
|
||||
&context,
|
||||
®istrar,
|
||||
reference_account,
|
||||
&voter,
|
||||
&voter_authority,
|
||||
&mngo_rate,
|
||||
depot_id,
|
||||
)
|
||||
};
|
||||
|
||||
// test deposit and withdraw
|
||||
|
||||
let initial = get_balances(0).await;
|
||||
assert_eq!(initial.vault, 0);
|
||||
assert_eq!(initial.deposit, 0);
|
||||
|
||||
context
|
||||
.addin
|
||||
.create_deposit(
|
||||
®istrar,
|
||||
&voter,
|
||||
&mngo_rate,
|
||||
&voter_authority,
|
||||
reference_account,
|
||||
governance_registry::account::LockupKind::None,
|
||||
10000,
|
||||
0,
|
||||
)
|
||||
.await?;
|
||||
|
||||
let after_deposit = get_balances(0).await;
|
||||
assert_eq!(initial.token, after_deposit.token + after_deposit.vault);
|
||||
assert_eq!(after_deposit.voter_weight, after_deposit.vault);
|
||||
assert_eq!(after_deposit.vault, 10000);
|
||||
assert_eq!(after_deposit.deposit, 10000);
|
||||
|
||||
// add to the existing deposit 0
|
||||
context
|
||||
.addin
|
||||
.update_deposit(
|
||||
®istrar,
|
||||
&voter,
|
||||
&mngo_rate,
|
||||
&voter_authority,
|
||||
reference_account,
|
||||
0,
|
||||
5000,
|
||||
)
|
||||
.await?;
|
||||
|
||||
let after_deposit2 = get_balances(0).await;
|
||||
assert_eq!(initial.token, after_deposit2.token + after_deposit2.vault);
|
||||
assert_eq!(after_deposit2.voter_weight, after_deposit2.vault);
|
||||
assert_eq!(after_deposit2.vault, 15000);
|
||||
assert_eq!(after_deposit2.deposit, 15000);
|
||||
|
||||
// create a separate deposit (index 1)
|
||||
context
|
||||
.addin
|
||||
.create_deposit(
|
||||
®istrar,
|
||||
&voter,
|
||||
&mngo_rate,
|
||||
&voter_authority,
|
||||
reference_account,
|
||||
governance_registry::account::LockupKind::None,
|
||||
7000,
|
||||
0,
|
||||
)
|
||||
.await?;
|
||||
|
||||
context
|
||||
.addin
|
||||
.withdraw(
|
||||
®istrar,
|
||||
&voter,
|
||||
&token_owner_record,
|
||||
&mngo_rate,
|
||||
&voter_authority,
|
||||
reference_account,
|
||||
0,
|
||||
10000,
|
||||
)
|
||||
.await
|
||||
.expect_err("deposit happened in the same slot");
|
||||
|
||||
let after_deposit3 = get_balances(1).await;
|
||||
assert_eq!(initial.token, after_deposit3.token + after_deposit3.vault);
|
||||
assert_eq!(after_deposit3.voter_weight, after_deposit3.vault);
|
||||
assert_eq!(after_deposit3.vault, 22000);
|
||||
assert_eq!(after_deposit3.deposit, 7000);
|
||||
|
||||
// Withdraw works now because some slots were advanced (in get_balances())
|
||||
context
|
||||
.addin
|
||||
.withdraw(
|
||||
®istrar,
|
||||
&voter,
|
||||
&token_owner_record,
|
||||
&mngo_rate,
|
||||
&voter_authority,
|
||||
reference_account,
|
||||
0,
|
||||
10000,
|
||||
)
|
||||
.await?;
|
||||
|
||||
let after_withdraw1 = get_balances(0).await;
|
||||
assert_eq!(initial.token, after_withdraw1.token + after_withdraw1.vault);
|
||||
assert_eq!(after_withdraw1.voter_weight, after_withdraw1.vault);
|
||||
assert_eq!(after_withdraw1.vault, 12000);
|
||||
assert_eq!(after_withdraw1.deposit, 5000);
|
||||
|
||||
context
|
||||
.addin
|
||||
.withdraw(
|
||||
®istrar,
|
||||
&voter,
|
||||
&token_owner_record,
|
||||
&mngo_rate,
|
||||
&voter_authority,
|
||||
reference_account,
|
||||
0,
|
||||
5001,
|
||||
)
|
||||
.await
|
||||
.expect_err("withdrew too much");
|
||||
|
||||
context
|
||||
.addin
|
||||
.withdraw(
|
||||
®istrar,
|
||||
&voter,
|
||||
&token_owner_record,
|
||||
&mngo_rate,
|
||||
&voter_authority,
|
||||
reference_account,
|
||||
0,
|
||||
5000,
|
||||
)
|
||||
.await?;
|
||||
|
||||
let after_withdraw2 = get_balances(0).await;
|
||||
assert_eq!(initial.token, after_withdraw2.token + after_withdraw2.vault);
|
||||
assert_eq!(after_withdraw2.voter_weight, after_withdraw2.vault);
|
||||
assert_eq!(after_withdraw2.vault, 7000);
|
||||
assert_eq!(after_withdraw2.deposit, 0);
|
||||
|
||||
// check that the voter2 account is still at 0
|
||||
let voter2_balances = balances(
|
||||
&context,
|
||||
®istrar,
|
||||
reference_account,
|
||||
&voter2,
|
||||
&voter2_authority,
|
||||
&mngo_rate,
|
||||
0,
|
||||
)
|
||||
.await;
|
||||
assert_eq!(voter2_balances.deposit, 0);
|
||||
assert_eq!(voter2_balances.voter_weight, 0);
|
||||
|
||||
// now voter2 deposits
|
||||
context
|
||||
.addin
|
||||
.create_deposit(
|
||||
®istrar,
|
||||
&voter2,
|
||||
&mngo_rate,
|
||||
&voter2_authority,
|
||||
context.users[2].token_accounts[0],
|
||||
governance_registry::account::LockupKind::None,
|
||||
1000,
|
||||
5,
|
||||
)
|
||||
.await?;
|
||||
|
||||
let voter2_balances = balances(
|
||||
&context,
|
||||
®istrar,
|
||||
reference_account,
|
||||
&voter2,
|
||||
&voter2_authority,
|
||||
&mngo_rate,
|
||||
0,
|
||||
)
|
||||
.await;
|
||||
assert_eq!(voter2_balances.deposit, 1000);
|
||||
assert_eq!(voter2_balances.voter_weight, 1000);
|
||||
assert_eq!(voter2_balances.vault, 8000);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
Loading…
Reference in New Issue