gaspersic/solana
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

chore: fix flaky web3 stake-program test (#14504)

Browse Source 
* chore: remove dangling skipPreflight

* chore: rearrange stake test to deactivate sooner after delegate

* chore: ensure stake is inactive before trying to withdraw
This commit is contained in:
Tyera Eulberg
2021-01-08 20:17:57 -07:00
committed by GitHub
parent ec48631fc5
commit 0d8acd1f79
1 changed files with 38 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
web3.js/test/stake-program.test.js
View File

@ -369,6 +369,36 @@ test('live staking actions', async () => {
}),
).rejects.toThrow();
// Deactivate stake
let deactivate = StakeProgram.deactivate({
stakePubkey: newAccountPubkey,
authorizedPubkey: authorized.publicKey,
});
await sendAndConfirmTransaction(connection, deactivate, [authorized], {
commitment: 'singleGossip',
});
let stakeActivationState;
do {
stakeActivationState = await connection.getStakeActivation(
newAccountPubkey,
);
} while (stakeActivationState.state != 'inactive');
// Test that withdraw succeeds after deactivation
withdraw = StakeProgram.withdraw({
stakePubkey: newAccountPubkey,
authorizedPubkey: authorized.publicKey,
toPubkey: recipient.publicKey,
lamports: minimumAmount + 20,
});
await sendAndConfirmTransaction(connection, withdraw, [authorized], {
commitment: 'singleGossip',
});
const recipientBalance = await connection.getBalance(recipient.publicKey);
expect(recipientBalance).toEqual(minimumAmount + 20);
// Split stake
const newStake = new Account();
let split = StakeProgram.split({
@ -380,6 +410,8 @@ test('live staking actions', async () => {
await sendAndConfirmTransaction(connection, split, [authorized, newStake], {
commitment: 'singleGossip',
});
const balance = await connection.getBalance(newAccountPubkey);
expect(balance).toEqual(minimumAmount + 2);
// Authorize to new account
const newAuthorized = new Account();
@ -404,45 +436,18 @@ test('live staking actions', async () => {
commitment: 'singleGossip',
});
// Test old authorized can't deactivate
let deactivateNotAuthorized = StakeProgram.deactivate({
// Test old authorized can't delegate
let delegateNotAuthorized = StakeProgram.delegate({
stakePubkey: newAccountPubkey,
authorizedPubkey: authorized.publicKey,
votePubkey,
});
await expect(
sendAndConfirmTransaction(
connection,
deactivateNotAuthorized,
[authorized],
{commitment: 'singleGossip', skipPreflight: true},
),
sendAndConfirmTransaction(connection, delegateNotAuthorized, [authorized], {
commitment: 'singleGossip',
}),
).rejects.toThrow();
// Deactivate stake
let deactivate = StakeProgram.deactivate({
stakePubkey: newAccountPubkey,
authorizedPubkey: newAuthorized.publicKey,
});
await sendAndConfirmTransaction(connection, deactivate, [newAuthorized], {
commitment: 'singleGossip',
});
// Test that withdraw succeeds after deactivation
withdraw = StakeProgram.withdraw({
stakePubkey: newAccountPubkey,
authorizedPubkey: newAuthorized.publicKey,
toPubkey: recipient.publicKey,
lamports: minimumAmount + 20,
});
await sendAndConfirmTransaction(connection, withdraw, [newAuthorized], {
commitment: 'singleGossip',
});
const balance = await connection.getBalance(newAccountPubkey);
expect(balance).toEqual(minimumAmount + 2);
const recipientBalance = await connection.getBalance(recipient.publicKey);
expect(recipientBalance).toEqual(minimumAmount + 20);
// Authorize a derived address
authorize = StakeProgram.authorize({
stakePubkey: newAccountPubkey,