Add mana field to Build enum variants in transaction builder tests (#…

…2141) * Add mana field to transaction builder tests * Use Build enum in more tests --------- Co-authored-by: Thibault Martinez <[email protected]>
iotaledger · Mar 6, 2024 · eb1873e · eb1873e
1 parent a3d8012
commit eb1873e
Show file tree

Hide file tree

Showing 16 changed files with 696 additions and 127 deletions.
diff --git a/sdk/tests/client/mod.rs b/sdk/tests/client/mod.rs
@@ -65,6 +65,7 @@ const SLOT_COMMITMENT_ID: SlotCommitmentId = SlotCommitmentHash::null().const_in
 enum Build<'a> {
     Basic {
         amount: u64,
+        mana: u64,
         address: Address,
         native_token: Option<(&'a str, u64)>,
         sender: Option<Address>,
@@ -74,6 +75,7 @@ enum Build<'a> {
     },
     Nft {
         amount: u64,
+        mana: u64,
         nft_id: NftId,
         address: Address,
         sender: Option<Address>,
@@ -83,6 +85,7 @@ enum Build<'a> {
     },
     Account {
         amount: u64,
+        mana: u64,
         account_id: AccountId,
         address: Address,
         sender: Option<Address>,
@@ -111,6 +114,7 @@ impl<'a> Build<'a> {
         match self {
             Build::Basic {
                 amount,
+                mana,
                 address,
                 native_token,
                 sender,
@@ -119,6 +123,7 @@ impl<'a> Build<'a> {
                 expiration,
             } => {
                 let mut builder = BasicOutputBuilder::new_with_amount(amount)
+                    .with_mana(mana)
                     .add_unlock_condition(AddressUnlockCondition::new(address.clone()));
 
                 if let Some((id, amount)) = native_token {
@@ -149,6 +154,7 @@ impl<'a> Build<'a> {
             }
             Build::Nft {
                 amount,
+                mana,
                 nft_id,
                 address,
                 sender,
@@ -157,6 +163,7 @@ impl<'a> Build<'a> {
                 expiration,
             } => {
                 let mut builder = NftOutputBuilder::new_with_amount(amount, nft_id)
+                    .with_mana(mana)
                     .add_unlock_condition(AddressUnlockCondition::new(address));
 
                 if let Some(sender) = sender {
@@ -180,12 +187,14 @@ impl<'a> Build<'a> {
             }
             Build::Account {
                 amount,
+                mana,
                 account_id,
                 address,
                 sender,
                 issuer,
             } => {
                 let mut builder = AccountOutputBuilder::new_with_amount(amount, account_id)
+                    .with_mana(mana)
                     .add_unlock_condition(AddressUnlockCondition::new(address));
 
                 if let Some(sender) = sender {

diff --git a/sdk/tests/client/signing/account.rs b/sdk/tests/client/signing/account.rs
@@ -54,6 +54,7 @@ async fn sign_account_state_transition() -> Result<(), Box<dyn std::error::Error
         [(
             Account {
                 amount: 1_000_000,
+                mana: 0,
                 account_id: account_id,
                 address: address.clone(),
                 sender: None,
@@ -66,6 +67,7 @@ async fn sign_account_state_transition() -> Result<(), Box<dyn std::error::Error
 
     let outputs = build_outputs([Account {
         amount: 1_000_000,
+        mana: 0,
         account_id: account_id,
         address: address.clone(),
         sender: None,
@@ -131,6 +133,7 @@ async fn account_reference_unlocks() -> Result<(), Box<dyn std::error::Error>> {
             (
                 Account {
                     amount: 1_000_000,
+                    mana: 0,
                     account_id: account_id,
                     address: address.clone(),
                     sender: None,
@@ -141,6 +144,7 @@ async fn account_reference_unlocks() -> Result<(), Box<dyn std::error::Error>> {
             (
                 Basic {
                     amount: 1_000_000,
+                    mana: 0,
                     address: account_address.clone(),
                     native_token: None,
                     sender: None,
@@ -153,6 +157,7 @@ async fn account_reference_unlocks() -> Result<(), Box<dyn std::error::Error>> {
             (
                 Basic {
                     amount: 1_000_000,
+                    mana: 0,
                     address: account_address.clone(),
                     native_token: None,
                     sender: None,
@@ -169,13 +174,15 @@ async fn account_reference_unlocks() -> Result<(), Box<dyn std::error::Error>> {
     let outputs = build_outputs([
         Account {
             amount: 1_000_000,
+            mana: 0,
             account_id: account_id,
             address: address,
             sender: None,
             issuer: None,
         },
         Basic {
             amount: 2_000_000,
+            mana: 0,
             address: account_address,
             native_token: None,
             sender: None,

diff --git a/sdk/tests/client/signing/basic.rs b/sdk/tests/client/signing/basic.rs
@@ -45,6 +45,7 @@ async fn single_ed25519_unlock() -> Result<(), Box<dyn std::error::Error>> {
         [(
             Basic {
                 amount: 1_000_000,
+                mana: 0,
                 address: address_0.clone(),
                 native_token: None,
                 sender: None,
@@ -59,6 +60,7 @@ async fn single_ed25519_unlock() -> Result<(), Box<dyn std::error::Error>> {
 
     let outputs = build_outputs([Basic {
         amount: 1_000_000,
+        mana: 0,
         address: address_0,
         native_token: None,
         sender: None,
@@ -124,6 +126,7 @@ async fn ed25519_reference_unlocks() -> Result<(), Box<dyn std::error::Error>> {
             (
                 Basic {
                     amount: 1_000_000,
+                    mana: 0,
                     address: address_0.clone(),
                     native_token: None,
                     sender: None,
@@ -136,6 +139,7 @@ async fn ed25519_reference_unlocks() -> Result<(), Box<dyn std::error::Error>> {
             (
                 Basic {
                     amount: 1_000_000,
+                    mana: 0,
                     address: address_0.clone(),
                     native_token: None,
                     sender: None,
@@ -148,6 +152,7 @@ async fn ed25519_reference_unlocks() -> Result<(), Box<dyn std::error::Error>> {
             (
                 Basic {
                     amount: 1_000_000,
+                    mana: 0,
                     address: address_0.clone(),
                     native_token: None,
                     sender: None,
@@ -163,6 +168,7 @@ async fn ed25519_reference_unlocks() -> Result<(), Box<dyn std::error::Error>> {
 
     let outputs = build_outputs([Basic {
         amount: 3_000_000,
+        mana: 0,
         address: address_0,
         native_token: None,
         sender: None,
@@ -249,6 +255,7 @@ async fn two_signature_unlocks() -> Result<(), Box<dyn std::error::Error>> {
             (
                 Basic {
                     amount: 1_000_000,
+                    mana: 0,
                     address: address_0.clone(),
                     native_token: None,
                     sender: None,
@@ -261,6 +268,7 @@ async fn two_signature_unlocks() -> Result<(), Box<dyn std::error::Error>> {
             (
                 Basic {
                     amount: 1_000_000,
+                    mana: 0,
                     address: address_1,
                     native_token: None,
                     sender: None,
@@ -276,6 +284,7 @@ async fn two_signature_unlocks() -> Result<(), Box<dyn std::error::Error>> {
 
     let outputs = build_outputs([Basic {
         amount: 2_000_000,
+        mana: 0,
         address: address_0,
         native_token: None,
         sender: None,

diff --git a/sdk/tests/client/signing/delegation.rs b/sdk/tests/client/signing/delegation.rs
@@ -55,6 +55,7 @@ async fn valid_creation() -> Result<(), Box<dyn std::error::Error>> {
         [(
             Basic {
                 amount: 1_000_000,
+                mana: 0,
                 address: address.clone(),
                 native_token: None,
                 sender: None,
@@ -135,6 +136,7 @@ async fn creation_missing_commitment_input() -> Result<(), Box<dyn std::error::E
         [(
             Basic {
                 amount: 1_000_000,
+                mana: 0,
                 address: address.clone(),
                 native_token: None,
                 sender: None,
@@ -197,6 +199,7 @@ async fn non_null_id_creation() -> Result<(), Box<dyn std::error::Error>> {
         [(
             Basic {
                 amount: 1_000_000,
+                mana: 0,
                 address: address.clone(),
                 native_token: None,
                 sender: None,
@@ -281,6 +284,7 @@ async fn mismatch_amount_creation() -> Result<(), Box<dyn std::error::Error>> {
         [(
             Basic {
                 amount: 1_000_000,
+                mana: 0,
                 address: address.clone(),
                 native_token: None,
                 sender: None,
@@ -365,6 +369,7 @@ async fn non_zero_end_epoch_creation() -> Result<(), Box<dyn std::error::Error>>
         [(
             Basic {
                 amount: 1_000_000,
+                mana: 0,
                 address: address.clone(),
                 native_token: None,
                 sender: None,
@@ -449,6 +454,7 @@ async fn invalid_start_epoch_creation() -> Result<(), Box<dyn std::error::Error>
         [(
             Basic {
                 amount: 1_000_000,
+                mana: 0,
                 address: address.clone(),
                 native_token: None,
                 sender: None,
@@ -1019,6 +1025,7 @@ async fn destroy_null_id() -> Result<(), Box<dyn std::error::Error>> {
 
     let outputs = build_outputs([Basic {
         amount: 1_000_000,
+        mana: 0,
         address,
         native_token: None,
         sender: None,
@@ -1111,6 +1118,7 @@ async fn destroy_reward_missing() -> Result<(), Box<dyn std::error::Error>> {
 
     let outputs = build_outputs([Basic {
         amount: 1_000_000,
+        mana: 0,
         address,
         native_token: None,
         sender: None,