Skip to content

Commit

Permalink
chore: fix lagging stake program live test (#25107)
Browse files Browse the repository at this point in the history 
* chore: fix lagging stake program live test

* chore: use min stake delegation in stake tests
  • Loading branch information
@jstarry
jstarry authored May 10, 2022
1 parent 61433de commit 42612c9
Show file tree
Hide file tree
Showing 2 changed files with 48 additions and 34 deletions.
13 changes: 7 additions & 6 deletions test/connection.test.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3103,6 +3103,11 @@ describe('Connection', function () {

if (process.env.TEST_LIVE) {
it('stake activation should return activating for new accounts', async () => {
// todo: use `Connection.getMinimumStakeDelegation` when implemented
const MIN_STAKE_DELEGATION = LAMPORTS_PER_SOL;
const STAKE_ACCOUNT_MIN_BALANCE =
await connection.getMinimumBalanceForRentExemption(StakeProgram.space);

const voteAccounts = await connection.getVoteAccounts();
const voteAccount = voteAccounts.current.concat(
voteAccounts.delinquent,
Expand All @@ -3116,17 +3121,13 @@ describe('Connection', function () {
);
await connection.confirmTransaction(signature, 'confirmed');

const minimumAmount = await connection.getMinimumBalanceForRentExemption(
StakeProgram.space,
);

const newStakeAccount = Keypair.generate();
let createAndInitialize = StakeProgram.createAccount({
fromPubkey: authorized.publicKey,
stakePubkey: newStakeAccount.publicKey,
authorized: new Authorized(authorized.publicKey, authorized.publicKey),
lockup: new Lockup(0, 0, new PublicKey(0)),
lamports: minimumAmount + 42,
lamports: STAKE_ACCOUNT_MIN_BALANCE + MIN_STAKE_DELEGATION,
});

await sendAndConfirmTransaction(
Expand Down Expand Up @@ -3164,7 +3165,7 @@ describe('Connection', function () {
'confirmed',
);
expect(activationState.state).to.eq('activating');
expect(activationState.inactive).to.eq(42);
expect(activationState.inactive).to.eq(MIN_STAKE_DELEGATION);
expect(activationState.active).to.eq(0);
});
}
Expand Down
69 changes: 41 additions & 28 deletions test/stake-program.test.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -380,6 +380,13 @@ describe('StakeProgram', () => {
if (process.env.TEST_LIVE) {
it('live staking actions', async () => {
const connection = new Connection(url, 'confirmed');
const SYSTEM_ACCOUNT_MIN_BALANCE =
await connection.getMinimumBalanceForRentExemption(0);
const STAKE_ACCOUNT_MIN_BALANCE =
await connection.getMinimumBalanceForRentExemption(StakeProgram.space);

// todo: use `Connection.getMinimumStakeDelegation` when implemented
const MIN_STAKE_DELEGATION = LAMPORTS_PER_SOL;

const voteAccounts = await connection.getVoteAccounts();
const voteAccount = voteAccounts.current.concat(
Expand All @@ -391,7 +398,7 @@ describe('StakeProgram', () => {
await helpers.airdrop({
connection,
address: payer.publicKey,
amount: 2 * LAMPORTS_PER_SOL,
amount: 10 * LAMPORTS_PER_SOL,
});

const authorized = Keypair.generate();
Expand All @@ -401,16 +408,12 @@ describe('StakeProgram', () => {
amount: 2 * LAMPORTS_PER_SOL,
});

const minimumAmount = await connection.getMinimumBalanceForRentExemption(
StakeProgram.space,
);

expect(await connection.getBalance(payer.publicKey)).to.eq(
2 * LAMPORTS_PER_SOL,
);
expect(await connection.getBalance(authorized.publicKey)).to.eq(
2 * LAMPORTS_PER_SOL,
);
const recipient = Keypair.generate();
await helpers.airdrop({
connection,
address: recipient.publicKey,
amount: SYSTEM_ACCOUNT_MIN_BALANCE,
});

{
// Create Stake account without seed
Expand All @@ -422,7 +425,7 @@ describe('StakeProgram', () => {
authorized.publicKey,
authorized.publicKey,
),
lamports: minimumAmount + 42,
lamports: STAKE_ACCOUNT_MIN_BALANCE + MIN_STAKE_DELEGATION,
});

await sendAndConfirmTransaction(
Expand All @@ -432,7 +435,7 @@ describe('StakeProgram', () => {
{preflightCommitment: 'confirmed'},
);
expect(await connection.getBalance(newStakeAccount.publicKey)).to.eq(
minimumAmount + 42,
STAKE_ACCOUNT_MIN_BALANCE + MIN_STAKE_DELEGATION,
);

const delegation = StakeProgram.delegate({
Expand All @@ -453,14 +456,16 @@ describe('StakeProgram', () => {
StakeProgram.programId,
);

const WITHDRAW_AMOUNT = 1;
const INITIAL_STAKE_DELEGATION = 5 * LAMPORTS_PER_SOL;
let createAndInitializeWithSeed = StakeProgram.createAccountWithSeed({
fromPubkey: payer.publicKey,
stakePubkey: newAccountPubkey,
basePubkey: payer.publicKey,
seed,
authorized: new Authorized(authorized.publicKey, authorized.publicKey),
lockup: new Lockup(0, 0, new PublicKey(0)),
lamports: 4 * minimumAmount + 62,
lamports: STAKE_ACCOUNT_MIN_BALANCE + INITIAL_STAKE_DELEGATION,
});

await sendAndConfirmTransaction(
Expand All @@ -470,7 +475,9 @@ describe('StakeProgram', () => {
{preflightCommitment: 'confirmed'},
);
let originalStakeBalance = await connection.getBalance(newAccountPubkey);
expect(originalStakeBalance).to.eq(4 * minimumAmount + 62);
expect(originalStakeBalance).to.eq(
STAKE_ACCOUNT_MIN_BALANCE + INITIAL_STAKE_DELEGATION,
);

let delegation = StakeProgram.delegate({
stakePubkey: newAccountPubkey,
Expand All @@ -482,12 +489,11 @@ describe('StakeProgram', () => {
});

// Test that withdraw fails before deactivation
const recipient = Keypair.generate();
let withdraw = StakeProgram.withdraw({
stakePubkey: newAccountPubkey,
authorizedPubkey: authorized.publicKey,
toPubkey: recipient.publicKey,
lamports: 1000,
lamports: WITHDRAW_AMOUNT,
});
await expect(
sendAndConfirmTransaction(connection, withdraw, [authorized], {
Expand Down Expand Up @@ -516,22 +522,24 @@ describe('StakeProgram', () => {
stakePubkey: newAccountPubkey,
authorizedPubkey: authorized.publicKey,
toPubkey: recipient.publicKey,
lamports: minimumAmount + 20,
lamports: WITHDRAW_AMOUNT,
});

await sendAndConfirmTransaction(connection, withdraw, [authorized], {
preflightCommitment: 'confirmed',
});
const recipientBalance = await connection.getBalance(recipient.publicKey);
expect(recipientBalance).to.eq(minimumAmount + 20);
expect(recipientBalance).to.eq(
SYSTEM_ACCOUNT_MIN_BALANCE + WITHDRAW_AMOUNT,
);

// Split stake
const newStake = Keypair.generate();
let split = StakeProgram.split({
stakePubkey: newAccountPubkey,
authorizedPubkey: authorized.publicKey,
splitStakePubkey: newStake.publicKey,
lamports: minimumAmount + 20,
lamports: STAKE_ACCOUNT_MIN_BALANCE + MIN_STAKE_DELEGATION,
});
await sendAndConfirmTransaction(
connection,
Expand All @@ -541,8 +549,8 @@ describe('StakeProgram', () => {
preflightCommitment: 'confirmed',
},
);
const balance = await connection.getBalance(newAccountPubkey);
expect(balance).to.eq(2 * minimumAmount + 22);
const balance = await connection.getBalance(newStake.publicKey);
expect(balance).to.eq(STAKE_ACCOUNT_MIN_BALANCE + MIN_STAKE_DELEGATION);

// Split stake with seed
const seed2 = 'test string 2';
Expand All @@ -554,7 +562,7 @@ describe('StakeProgram', () => {
let splitWithSeed = StakeProgram.splitWithSeed({
stakePubkey: newAccountPubkey,
authorizedPubkey: authorized.publicKey,
lamports: minimumAmount + 20,
lamports: STAKE_ACCOUNT_MIN_BALANCE + MIN_STAKE_DELEGATION,
splitStakePubkey: newStake2,
basePubkey: payer.publicKey,
seed: seed2,
Expand All @@ -567,11 +575,12 @@ describe('StakeProgram', () => {
preflightCommitment: 'confirmed',
},
);
expect(await connection.getBalance(newAccountPubkey)).to.eq(
minimumAmount + 2,
expect(await connection.getBalance(newStake2)).to.eq(
STAKE_ACCOUNT_MIN_BALANCE + MIN_STAKE_DELEGATION,
);

// Merge stake
const preMergeBalance = await connection.getBalance(newAccountPubkey);
let merge = StakeProgram.merge({
stakePubkey: newAccountPubkey,
sourceStakePubKey: newStake.publicKey,
Expand All @@ -580,15 +589,19 @@ describe('StakeProgram', () => {
await sendAndConfirmTransaction(connection, merge, [authorized], {
preflightCommitment: 'confirmed',
});
const mergedBalance = await connection.getBalance(newAccountPubkey);
expect(mergedBalance).to.eq(2 * minimumAmount + 22);
const postMergeBalance = await connection.getBalance(newAccountPubkey);
expect(postMergeBalance - preMergeBalance).to.eq(
STAKE_ACCOUNT_MIN_BALANCE + MIN_STAKE_DELEGATION,
);

// Resplit
split = StakeProgram.split({
stakePubkey: newAccountPubkey,
authorizedPubkey: authorized.publicKey,
splitStakePubkey: newStake.publicKey,
lamports: minimumAmount + 20,
// use a different amount than the first split so that this
// transaction is different and won't require a fresh blockhash
lamports: STAKE_ACCOUNT_MIN_BALANCE + MIN_STAKE_DELEGATION,
});
await sendAndConfirmTransaction(
connection,
Expand Down

0 comments on commit 42612c9

Please sign in to comment.