refactor(rosetta): use new system-state summary (#5507)

Author: kodemartin

crates/iota-rosetta/src/network.rs

@@ -7,7 +7,9 @@
 use axum::{Extension, Json, extract::State};
 use axum_extra::extract::WithRejection;
 use fastcrypto::encoding::Hex;
-use iota_types::base_types::ObjectID;
+use iota_types::{
+    base_types::ObjectID, iota_system_state::iota_system_state_summary::IotaSystemStateSummary,
+};
 use serde_json::json;
 use strum::IntoEnumIterator;
 
@@ -50,8 +52,12 @@ pub async fn status(
         .get_latest_iota_system_state()
         .await?;
 
-    let peers = system_state
-        .active_validators
+    let active_validators = match system_state {
+        IotaSystemStateSummary::V1(v1) => v1.active_validators,
+        IotaSystemStateSummary::V2(v2) => v2.active_validators,
+        _ => return Err(anyhow::anyhow!("unsupported IotaSystemStateSummary"))?,
+    };
+    let peers = active_validators
         .iter()
         .map(|validator| Peer {
             peer_id: ObjectID::from(validator.iota_address).into(),

crates/iota-rosetta/src/unit_tests/balance_changing_tx_tests.rs

@@ -26,6 +26,7 @@ use iota_types::{
     TypeTag,
     base_types::{IotaAddress, ObjectID, ObjectRef},
     gas_coin::GasCoin,
+    iota_system_state::iota_system_state_summary::IotaSystemStateSummary,
     programmable_transaction_builder::ProgrammableTransactionBuilder,
     quorum_driver_types::ExecuteTransactionRequestType,
     transaction::{
@@ -490,13 +491,17 @@ async fn test_stake_iota() {
     let sender = get_random_address(&network.get_addresses(), vec![]);
     let coin1 = get_random_iota(&client, sender, vec![]).await;
     let coin2 = get_random_iota(&client, sender, vec![coin1.0]).await;
-    let validator = client
+    let system_state = client
         .governance_api()
         .get_latest_iota_system_state()
         .await
-        .unwrap()
-        .active_validators[0]
-        .iota_address;
+        .unwrap();
+    let active_validators = match system_state {
+        IotaSystemStateSummary::V1(v1) => v1.active_validators,
+        IotaSystemStateSummary::V2(v2) => v2.active_validators,
+        _ => panic!("unsupported IotaSystemStateSummary"),
+    };
+    let validator = active_validators[0].iota_address;
     let tx = client
         .transaction_builder()
         .request_add_stake(
@@ -538,13 +543,17 @@ async fn test_stake_iota_with_none_amount() {
     let sender = get_random_address(&network.get_addresses(), vec![]);
     let coin1 = get_random_iota(&client, sender, vec![]).await;
     let coin2 = get_random_iota(&client, sender, vec![coin1.0]).await;
-    let validator = client
+    let system_state = client
         .governance_api()
         .get_latest_iota_system_state()
         .await
-        .unwrap()
-        .active_validators[0]
-        .iota_address;
+        .unwrap();
+    let active_validators = match system_state {
+        IotaSystemStateSummary::V1(v1) => v1.active_validators,
+        IotaSystemStateSummary::V2(v2) => v2.active_validators,
+        _ => panic!("unsupported IotaSystemStateSummary"),
+    };
+    let validator = active_validators[0].iota_address;
     let tx = client
         .transaction_builder()
         .request_add_stake(
@@ -614,13 +623,17 @@ async fn test_delegation_parsing() -> Result<(), anyhow::Error> {
     let client = network.wallet.get_client().await.unwrap();
     let sender = get_random_address(&network.get_addresses(), vec![]);
     let gas = get_random_iota(&client, sender, vec![]).await;
-    let validator = client
+    let system_state = client
         .governance_api()
         .get_latest_iota_system_state()
         .await
-        .unwrap()
-        .active_validators[0]
-        .iota_address;
+        .unwrap();
+    let active_validators = match system_state {
+        IotaSystemStateSummary::V1(v1) => v1.active_validators,
+        IotaSystemStateSummary::V2(v2) => v2.active_validators,
+        _ => anyhow::bail!("unsupported IotaSystemStateSummary"),
+    };
+    let validator = active_validators[0].iota_address;
 
     let ops: Operations = serde_json::from_value(json!(
         [{

crates/iota-rosetta/tests/end_to_end_tests.rs

@@ -16,6 +16,7 @@ use iota_rosetta::{
 use iota_sdk::rpc_types::{IotaExecutionStatus, IotaTransactionBlockEffectsAPI};
 use iota_swarm_config::genesis_config::{DEFAULT_GAS_AMOUNT, DEFAULT_NUMBER_OF_OBJECT_PER_ACCOUNT};
 use iota_types::{
+    iota_system_state::iota_system_state_summary::IotaSystemStateSummary,
     quorum_driver_types::ExecuteTransactionRequestType, utils::to_sender_signed_transaction,
 };
 use rosetta_client::start_rosetta_test_server;
@@ -78,13 +79,17 @@ async fn test_get_staked_iota() {
     assert_eq!(response.balances[0].value, 0);
 
     // Stake some iota
-    let validator = client
+    let system_state = client
         .governance_api()
         .get_latest_iota_system_state()
         .await
-        .unwrap()
-        .active_validators[0]
-        .iota_address;
+        .unwrap();
+    let active_validators = match system_state {
+        IotaSystemStateSummary::V1(v1) => v1.active_validators,
+        IotaSystemStateSummary::V2(v2) => v2.active_validators,
+        _ => panic!("unsupported IotaSystemStateSummary"),
+    };
+    let validator = active_validators[0].iota_address;
     let coins = client
         .coin_read_api()
         .get_coins(address, None, None, None)
@@ -136,13 +141,17 @@ async fn test_stake() {
 
     let (rosetta_client, _handle) = start_rosetta_test_server(client.clone()).await;
 
-    let validator = client
+    let system_state = client
         .governance_api()
         .get_latest_iota_system_state()
         .await
-        .unwrap()
-        .active_validators[0]
-        .iota_address;
+        .unwrap();
+    let active_validators = match system_state {
+        IotaSystemStateSummary::V1(v1) => v1.active_validators,
+        IotaSystemStateSummary::V2(v2) => v2.active_validators,
+        _ => panic!("unsupported IotaSystemStateSummary"),
+    };
+    let validator = active_validators[0].iota_address;
 
     let ops = serde_json::from_value(json!(
         [{
@@ -197,13 +206,17 @@ async fn test_stake_all() {
 
     let (rosetta_client, _handle) = start_rosetta_test_server(client.clone()).await;
 
-    let validator = client
+    let system_state = client
         .governance_api()
         .get_latest_iota_system_state()
         .await
-        .unwrap()
-        .active_validators[0]
-        .iota_address;
+        .unwrap();
+    let active_validators = match system_state {
+        IotaSystemStateSummary::V1(v1) => v1.active_validators,
+        IotaSystemStateSummary::V2(v2) => v2.active_validators,
+        _ => panic!("unsupported IotaSystemStateSummary"),
+    };
+    let validator = active_validators[0].iota_address;
 
     let ops = serde_json::from_value(json!(
         [{
@@ -263,13 +276,17 @@ async fn test_withdraw_stake() {
     let (rosetta_client, _handle) = start_rosetta_test_server(client.clone()).await;
 
     // First add some stakes
-    let validator = client
+    let system_state = client
         .governance_api()
         .get_latest_iota_system_state()
         .await
-        .unwrap()
-        .active_validators[0]
-        .iota_address;
+        .unwrap();
+    let active_validators = match system_state {
+        IotaSystemStateSummary::V1(v1) => v1.active_validators,
+        IotaSystemStateSummary::V2(v2) => v2.active_validators,
+        _ => panic!("unsupported IotaSystemStateSummary"),
+    };
+    let validator = active_validators[0].iota_address;
 
     let ops = serde_json::from_value(json!(
         [{