E2E Testing: Staking

[TrevorJTClarke]

Unit Tests

• [ ] Add unit tests here: https://github.com/Cron-Near/treasury/blob/main/treasury/src/staking.rs#L688
• [ ] Test: add staking pool
• [ ] Test: remove staking pool
• [ ] Test: Get all infos: get_info, get_config, get_delegations
• [ ] Test: deposit_and_stake, confirm it works for both attached deposit, and optional amount
• [ ] Test: get_staked_balance validate balance matches the deposited amount (requires deposit_and_stake to be called first)
• [ ] Test: unstake, validate this works AND if croncat is specified - withdraw task was created
• [ ] Test: withdraw can be called and works as expected
• [ ] Test: yield_harvest can be called and works as expected for supported pool
• [ ] Test: liquid_unstake works for supported pools and returns valid amount based on optional amount OR returns valid amount for unstaking all

Workspace Tests

• [ ] Add tests here: https://github.com/Cron-Near/treasury/tree/main/e2e/src as a new file "staking.rs"
• [ ] Test Flow: Metapool Staking
  • [ ] Add pool meta-v2.pool.testnet
  • [ ] call deposit_and_stake with attached amount
  • [ ] view get_delegations and verify stake amount
  • [ ] call liquid_unstake and specify 1/4 amount staked
  • [ ] call get_staked_balance to update internal values
  • [ ] view get_delegations and verify stake + unstake amounts
  • [ ] call unstake and specify 1/2 amount staked, validate auto withdraw is setup
  • [ ] add TODO: comment - add fastforward setup to wait until natural unstake 4 epochs, then trigger another withdraw manually
• [ ] Test Flow: Vanilla Validator Staking
  • [ ] call deposit_and_stake with internal amount
  • [ ] call get_staked_balance to update internal values
  • [ ] view get_delegations and verify stake + unstake amounts
  • [ ] call unstake and specify 1/2 amount staked, validate auto withdraw is setup
  • [ ] add TODO: comment - add fastforward setup to wait until natural unstake 4 epochs, then trigger another withdraw manually
• [ ] Test Flow: Yield Harvest: Metapool
  • [ ] Add pool meta-v2.pool.testnet
  • [ ] call deposit_and_stake with attached amount
  • [ ] add TODO: comment - add fastforward setup to wait until natural unstake 4 epochs, then trigger manually
  • [ ] call yield_harvest on the pool contract

Examples:

# add/remove staking pools
near call $TREASURY_ACCOUNT_ID add_staking_pool '{"pool_account_id": "node0"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS
near call $TREASURY_ACCOUNT_ID add_staking_pool '{"pool_account_id": "meta-v2.pool.testnet", "liquid_unstake_function": "liquid_unstake", "yield_function": "harvest_meta", "withdraw_function": "withdraw_all"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS
near call $TREASURY_ACCOUNT_ID add_staking_pool '{"pool_account_id": "hotones.pool.f863973.m0"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS
near call $TREASURY_ACCOUNT_ID remove_staking_pool '{"pool_account_id": "hotones.pool.f863973.m0"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS
near view $TREASURY_ACCOUNT_ID get_delegations
near view $TREASURY_ACCOUNT_ID get_config
near view $TREASURY_ACCOUNT_ID get_info

# deposit&stake
near call $TREASURY_ACCOUNT_ID deposit_and_stake '{"pool_account_id": "meta-v2.pool.testnet"}' --accountId $TREASURY_ACCOUNT_ID --amount 10
near call $TREASURY_ACCOUNT_ID deposit_and_stake '{"pool_account_id": "hotones.pool.f863973.m0"}' --accountId $TREASURY_ACCOUNT_ID --amount 10 --gas $MAX_GAS
near call $TREASURY_ACCOUNT_ID deposit_and_stake '{"pool_account_id": "hotones.pool.f863973.m0", "amount": "100000000000000000000000000"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS

# # update internal balance
near call $TREASURY_ACCOUNT_ID get_staked_balance '{"pool_account_id": "meta-v2.pool.testnet"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS
near call $TREASURY_ACCOUNT_ID get_staked_balance '{"pool_account_id": "hotones.pool.f863973.m0"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS
near view $TREASURY_ACCOUNT_ID get_delegations

# liquidunstake
near call $TREASURY_ACCOUNT_ID liquid_unstake '{"pool_account_id": "meta-v2.pool.testnet", "amount": "5000000000000000000000000"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS
near call $TREASURY_ACCOUNT_ID liquid_unstake '{"pool_account_id": "meta-v2.pool.testnet"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS

# unstake
near call $TREASURY_ACCOUNT_ID unstake '{"pool_account_id": "meta-v2.pool.testnet"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS
near call $TREASURY_ACCOUNT_ID unstake '{"pool_account_id": "hotones.pool.f863973.m0", "amount": "5000000000000000000000000"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS

# withdraw
near call $TREASURY_ACCOUNT_ID withdraw '{"pool_account_id": "meta-v2.pool.testnet"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS

# withdraw
near call $TREASURY_ACCOUNT_ID yield_harvest '{"pool_account_id": "meta-v2.pool.testnet"}' --accountId $TREASURY_ACCOUNT_ID --gas $MAX_GAS