Failing test

Author: cffls

test/pycardano/test_txbuilder.py

@@ -35,6 +35,7 @@
     VERIFICATION_KEY_HASH_SIZE,
     PoolKeyHash,
     TransactionId,
+    ScriptHash,
     VerificationKeyHash,
 )
 from pycardano.key import VerificationKey
@@ -2326,3 +2327,88 @@ def test_collateral_no_duplicates(chain_context):
             total_collateral_input
             == Value(tx_body.total_collateral) + tx_body.collateral_return.amount
         ), "The total collateral input amount should match the sum of the selected UTxOs"
+
+
+def test_token_transfer_with_change(chain_context):
+    """Test token transfer with change address handling.
+
+    Replicates issue where transaction fails with 'Input UTxOs depleted' when:
+    - Input 1: 4 ADA
+    - Input 2: ~1.03 ADA + 1,876,083 tokens
+    - Output: ~1.32 ADA + 382 tokens
+    - Expected change should handle remaining ADA and tokens
+    """
+    # Create the vault address that holds tokens
+    vault_address = Address.from_primitive(
+        "addr_test1vrs324jltsc0ssuptpa5ngpfk89cps92xa99a2t6vlg6kdqtm5qnv"
+    )
+
+    # Create receiver address
+    receiver_address = Address.from_primitive(
+        "addr_test1vrm9x2zsux7va6w892g38tvchnzahvcd9tykqf3ygnmwtaqyfg52x"
+    )
+
+    # Create token details
+    token_policy_id = ScriptHash(bytes.fromhex("1f847bb9ac60e869780037c0510dbd89f745316db7ec4fee81ff1e97"))
+    token_name = AssetName(b"dux_1")
+
+    # Create the two input UTXOs and patch chain_context.utxos
+    with patch.object(chain_context, "utxos") as mock_utxos:
+        mock_utxos.return_value = [
+            UTxO(
+                TransactionInput.from_primitive([
+                    "e11efc26f94a3cbf724dc052c43abf36f7a631a831acc6d783f1c9c8c52725c5",
+                    0
+                ]),
+                TransactionOutput(
+                    vault_address,
+                    Value(
+                        1038710,  # ~1.03 ADA
+                        MultiAsset.from_primitive({
+                            token_policy_id.payload: {
+                                b"dux_1": 1876083  # 1,876,083 tokens
+                            }
+                        })
+                    )
+                )
+            )
+        ]
+
+        # Create transaction builder
+        tx_builder = TransactionBuilder(chain_context)
+
+        # Add inputs - using add_input_address for the vault input
+        tx_builder.add_input_address(vault_address)
+        tx_builder.add_input(UTxO(
+            TransactionInput.from_primitive([b"1" * 32, 0]),
+            TransactionOutput(receiver_address, Value(4000000)) # 4 ADA input
+        ))
+
+        # Add output for receiver
+        output_value = Value(
+            1326255,  # ~1.32 ADA
+            MultiAsset.from_primitive({
+                token_policy_id.payload: {
+                    b"dux_1": 382  # 382 tokens
+                }
+            })
+        )
+        tx_builder.add_output(TransactionOutput(receiver_address, output_value))
+
+        # Build transaction with change going back to vault
+        tx = tx_builder.build(change_address=vault_address, merge_change=True)
+
+        # Verify the transaction outputs
+        assert len(tx.outputs) == 2  # One for receiver, one for change
+
+        # Verify receiver output
+        receiver_output = tx.outputs[0]
+        assert receiver_output.address == receiver_address
+        assert receiver_output.amount.coin == 1326255
+        assert receiver_output.amount.multi_asset[token_policy_id][token_name] == 382
+
+        # Verify change output
+        change_output = tx.outputs[1]
+        assert change_output.address == vault_address
+        assert change_output.amount.coin == 4000000 + 1038710 - 1326255 - tx.fee
+        assert change_output.amount.multi_asset[token_policy_id][token_name] == 1876083 - 382