Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

(Indexer): Use dedicated keypairs for Governance API tests #3745

Merged
merged 2 commits into from
Oct 29, 2024
Merged

(Indexer): Use dedicated keypairs for Governance API tests #3745

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
136d05f
refactor: use dedicated keypairs and consider epoch progression from …
samuel-rufi Oct 29, 2024
c9cf125
refactor: remove unnecessary gas_price * 2
samuel-rufi Oct 29, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
173 changes: 73 additions & 100 deletions crates/iota-indexer/tests/rpc-tests/governance_api.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,14 +1,8 @@
// Copyright (c) 2024 IOTA Stiftung
// SPDX-License-Identifier: Apache-2.0

use iota_json_rpc_api::{
CoinReadApiClient, GovernanceReadApiClient, IndexerApiClient, TransactionBuilderClient,
WriteApiClient,
};
use iota_json_rpc_types::{
CoinPage, DelegatedStake, IotaObjectDataOptions, IotaObjectResponseQuery,
IotaTransactionBlockResponseOptions, ObjectsPage, StakeStatus, TransactionBlockBytes,
};
use iota_json_rpc_api::{GovernanceReadApiClient, TransactionBuilderClient};
use iota_json_rpc_types::{DelegatedStake, StakeStatus, TransactionBlockBytes};
use iota_test_transaction_builder::TestTransactionBuilder;
use iota_types::{
IOTA_FRAMEWORK_ADDRESS, IOTA_SYSTEM_ADDRESS,
Expand All @@ -17,7 +11,6 @@ use iota_types::{
crypto::{AccountKeyPair, get_key_pair},
gas_coin::GAS,
programmable_transaction_builder::ProgrammableTransactionBuilder,
quorum_driver_types::ExecuteTransactionRequestType,
transaction::{CallArg, ObjectArg},
utils::to_sender_signed_transaction,
};
Expand All @@ -40,26 +33,30 @@ fn test_staking() {
runtime.block_on(async move {
indexer_wait_for_checkpoint(store, 1).await;

let address = cluster.get_address_0();

let objects: ObjectsPage = client
.get_owned_objects(
address,
Some(IotaObjectResponseQuery::new_with_options(
IotaObjectDataOptions::new()
.with_type()
.with_owner()
.with_previous_transaction(),
)),
None,
None,
let (sender, keypair): (_, AccountKeyPair) = get_key_pair();

let gas = cluster
.fund_address_and_return_gas(
cluster.get_reference_gas_price().await,
Some(10_000_000_000),
sender,
)
.await
.unwrap();
assert_eq!(5, objects.data.len());
.await;

indexer_wait_for_object(client, gas.0, gas.1).await;

let iota_coin_ref = cluster
.fund_address_and_return_gas(
cluster.get_reference_gas_price().await,
Some(10_000_000_000),
sender,
)
.await;

indexer_wait_for_object(client, iota_coin_ref.0, iota_coin_ref.1).await;

// Check StakedIota object before test
let staked_iota: Vec<DelegatedStake> = client.get_stakes(address).await.unwrap();
let staked_iota: Vec<DelegatedStake> = client.get_stakes(sender).await.unwrap();
assert!(staked_iota.is_empty());

let validator = client
Expand All @@ -69,40 +66,29 @@ fn test_staking() {
.active_validators[0]
.iota_address;

let coin = objects.data[0].object().unwrap().object_id;
// Delegate some IOTA
let transaction_bytes: TransactionBlockBytes = client
.request_add_stake(
address,
vec![coin],
sender,
vec![iota_coin_ref.0],
Some(1000000000.into()),
validator,
None,
Some(gas.0),
100_000_000.into(),
)
.await
.unwrap();
let tx = cluster
.wallet
.sign_transaction(&transaction_bytes.to_data().unwrap());

let (tx_bytes, signatures) = tx.to_tx_bytes_and_signatures();
let txn = to_sender_signed_transaction(transaction_bytes.to_data().unwrap(), &keypair);

client
.execute_transaction_block(
tx_bytes,
signatures,
Some(IotaTransactionBlockResponseOptions::new()),
Some(ExecuteTransactionRequestType::WaitForLocalExecution),
)
.await
.unwrap();
let res = cluster.wallet.execute_transaction_must_succeed(txn).await;
indexer_wait_for_transaction(res.digest, store, client).await;

cluster.force_new_epoch().await;
indexer_wait_for_latest_checkpoint(store, cluster).await;

// Check DelegatedStake object after epoch transition
let staked_iota: Vec<DelegatedStake> = client.get_stakes(address).await.unwrap();
let staked_iota: Vec<DelegatedStake> = client.get_stakes(sender).await.unwrap();
assert_eq!(1, staked_iota.len());
assert_eq!(1000000000, staked_iota[0].stakes[0].principal);
assert!(matches!(
Expand All @@ -128,16 +114,30 @@ fn test_unstaking() {
runtime.block_on(async move {
indexer_wait_for_checkpoint(store, 1).await;

let address = cluster.get_address_0();
let (sender, keypair): (_, AccountKeyPair) = get_key_pair();

let coins: CoinPage = indexer_client
.get_coins(address, None, None, None)
.await
.unwrap();
assert_eq!(5, coins.data.len());
let gas = cluster
.fund_address_and_return_gas(
cluster.get_reference_gas_price().await,
Some(10_000_000_000),
sender,
)
.await;

indexer_wait_for_object(client, gas.0, gas.1).await;

let iota_coin_ref = cluster
.fund_address_and_return_gas(
cluster.get_reference_gas_price().await,
Some(10_000_000_000),
sender,
)
.await;

indexer_wait_for_object(client, iota_coin_ref.0, iota_coin_ref.1).await;

// Check StakedIota object before test
let staked_iota: Vec<DelegatedStake> = indexer_client.get_stakes(address).await.unwrap();
let staked_iota: Vec<DelegatedStake> = indexer_client.get_stakes(sender).await.unwrap();
assert!(staked_iota.is_empty());

let validator = indexer_client
Expand All @@ -150,36 +150,26 @@ fn test_unstaking() {
// Delegate some IOTA
let transaction_bytes: TransactionBlockBytes = indexer_client
.request_add_stake(
address,
vec![coins.data[0].coin_object_id],
sender,
vec![iota_coin_ref.0],
Some(1000000000.into()),
validator,
None,
Some(gas.0),
100_000_000.into(),
)
.await
.unwrap();
let tx = cluster
.wallet
.sign_transaction(&transaction_bytes.to_data().unwrap());

let (tx_bytes, signatures) = tx.to_tx_bytes_and_signatures();
let txn = to_sender_signed_transaction(transaction_bytes.to_data().unwrap(), &keypair);

indexer_client
.execute_transaction_block(
tx_bytes,
signatures,
Some(IotaTransactionBlockResponseOptions::new()),
Some(ExecuteTransactionRequestType::WaitForLocalExecution),
)
.await
.unwrap();
let res = cluster.wallet.execute_transaction_must_succeed(txn).await;
indexer_wait_for_transaction(res.digest, store, client).await;

cluster.force_new_epoch().await;
indexer_wait_for_latest_checkpoint(store, cluster).await;

// Check DelegatedStake object
let staked_iota: Vec<DelegatedStake> = indexer_client.get_stakes(address).await.unwrap();
let staked_iota: Vec<DelegatedStake> = indexer_client.get_stakes(sender).await.unwrap();
assert_eq!(1, staked_iota.len());
assert_eq!(1000000000, staked_iota[0].stakes[0].principal);
assert!(matches!(
Expand All @@ -191,18 +181,18 @@ fn test_unstaking() {

let transaction_bytes: TransactionBlockBytes = indexer_client
.request_withdraw_stake(
address,
sender,
staked_iota[0].stakes[0].staked_iota_id,
None,
Some(gas.0),
100_000_000.into(),
)
.await
.unwrap();
let tx = cluster
.wallet
.sign_transaction(&transaction_bytes.to_data().unwrap());

let _ = cluster.wallet.execute_transaction_must_succeed(tx).await;
let txn = to_sender_signed_transaction(transaction_bytes.to_data().unwrap(), &keypair);

let res = cluster.wallet.execute_transaction_must_succeed(txn).await;
indexer_wait_for_transaction(res.digest, store, client).await;

cluster.force_new_epoch().await;
indexer_wait_for_latest_checkpoint(store, cluster).await;
Expand All @@ -224,7 +214,7 @@ fn test_unstaking() {
.unwrap();
assert_eq!(0, indexer_response.len());

let staked_iota: Vec<DelegatedStake> = indexer_client.get_stakes(address).await.unwrap();
let staked_iota: Vec<DelegatedStake> = indexer_client.get_stakes(sender).await.unwrap();
assert!(staked_iota.is_empty());
});
}
Expand Down Expand Up @@ -316,7 +306,7 @@ fn test_timelocked_staking() {
let context = &cluster.wallet;
let gas_price = context.get_reference_gas_price().await.unwrap();

let tx_builder = TestTransactionBuilder::new(sender, gas, gas_price);
let tx_builder = TestTransactionBuilder::new(sender, gas, gas_price * 2);
samuel-rufi marked this conversation as resolved.
Show resolved Hide resolved
let txn = to_sender_signed_transaction(tx_builder.programmable(pt).build(), &keypair);

let res = context.execute_transaction_must_succeed(txn).await;
Expand Down Expand Up @@ -504,10 +494,9 @@ fn get_latest_iota_system_state() {
runtime.block_on(async move {
indexer_wait_for_checkpoint(store, 1).await;

let response = client.get_latest_iota_system_state().await.unwrap();
assert_eq!(response.epoch, 0);
assert_eq!(response.protocol_version, 1);
assert_eq!(response.system_state_version, 1);
let system_state = client.get_latest_iota_system_state().await.unwrap();
assert_eq!(system_state.protocol_version, 1);
assert_eq!(system_state.system_state_version, 1);
});
}

Expand All @@ -517,7 +506,7 @@ fn get_committee_info() {
runtime,
store,
client,
cluster,
..
} = ApiTestSetup::get_or_init();

runtime.block_on(async move {
Expand All @@ -526,10 +515,7 @@ fn get_committee_info() {
// Test with no specified epoch
let response = client.get_committee_info(None).await.unwrap();

let (epoch_id, validators) = (response.epoch, response.validators);

assert!(epoch_id == 0);
assert_eq!(validators.len(), 4);
assert_eq!(response.validators.len(), 4);

// Test with specified epoch 0
let response = client.get_committee_info(Some(0.into())).await.unwrap();
Expand All @@ -540,20 +526,9 @@ fn get_committee_info() {
assert_eq!(validators.len(), 4);

// Test with non-existent epoch
let response = client.get_committee_info(Some(1.into())).await;
let response = client.get_committee_info(Some(u64::MAX.into())).await;

assert!(response.is_err());

// Sleep for 5 seconds
cluster.force_new_epoch().await;

// Test with specified epoch 1
let response = client.get_committee_info(Some(1.into())).await.unwrap();

let (epoch_id, validators) = (response.epoch, response.validators);

assert!(epoch_id == 1);
assert_eq!(validators.len(), 4);
});
}

Expand Down Expand Up @@ -586,10 +561,8 @@ fn get_validators_apy() {
runtime.block_on(async move {
indexer_wait_for_checkpoint(store, 1).await;

let response = client.get_validators_apy().await.unwrap();
let (apys, epoch) = (response.apys, response.epoch);
let apys = client.get_validators_apy().await.unwrap().apys;

assert_eq!(epoch, 0);
assert_eq!(apys.len(), 4);
assert!(apys.iter().any(|apy| apy.apy == 0.0));
});
Expand Down
Loading