[bugfix] spend utxo with script in the output

Author: cffls

integration-test/test/test_plutus.py

@@ -330,6 +330,45 @@ def test_plutus_v2_ref_script(self):
 
         self.assert_output(taker_address, take_output)
 
+    @retry(tries=TEST_RETRIES, backoff=1.3, delay=2, jitter=(0, 10))
+    @pytest.mark.post_alonzo
+    def test_plutus_v2_spend_ref_script(self):
+        # ----------- Create a reference script ---------------
+        with open("./plutus_scripts/fortytwoV2.plutus", "r") as f:
+            script_hex = f.read()
+            forty_two_script = PlutusV2Script(cbor2.loads(bytes.fromhex(script_hex)))
+
+        giver_address = Address(self.payment_vkey.hash(), network=self.NETWORK)
+
+        builder = TransactionBuilder(self.chain_context)
+        builder.add_input_address(giver_address)
+        builder.add_output(
+            TransactionOutput(giver_address, 50000000, script=forty_two_script)
+        )
+
+        signed_tx = builder.build_and_sign([self.payment_skey], giver_address)
+
+        print("############### Transaction created ###############")
+        print(signed_tx)
+        print(signed_tx.to_cbor_hex())
+        print("############### Submitting transaction ###############")
+        self.chain_context.submit_tx(signed_tx)
+        time.sleep(6)
+
+        # ----------- Spend script utxo ---------------
+        utxo_to_spend = UTxO(
+            TransactionInput(signed_tx.id, 0), signed_tx.transaction_body.outputs[0]
+        )
+
+        builder = TransactionBuilder(self.chain_context)
+        builder.add_input(utxo_to_spend)
+        signed_tx = builder.build_and_sign([self.payment_skey], giver_address)
+        print("############### Transaction created ###############")
+        print(signed_tx)
+        print(signed_tx.to_cbor_hex())
+        print("############### Submitting transaction ###############")
+        self.chain_context.submit_tx(signed_tx)
+
     @retry(tries=TEST_RETRIES, backoff=1.3, delay=2, jitter=(0, 10))
     @pytest.mark.post_alonzo
     def test_transaction_chaining(self):

pycardano/txbuilder.py

@@ -218,6 +218,8 @@ def add_input(self, utxo: UTxO) -> TransactionBuilder:
             TransactionBuilder: Current transaction builder.
         """
         self.inputs.append(utxo)
+        if utxo.output.script:
+            self._reference_scripts.append(utxo.output.script)
         return self
 
     def _consolidate_redeemer(self, redeemer):
@@ -1403,7 +1405,11 @@ def build(
 
             for address in self.input_addresses:
                 for utxo in self.context.utxos(address):
-                    if utxo not in seen_utxos and utxo not in self.excluded_inputs:
+                    if (
+                        utxo not in seen_utxos
+                        and utxo not in self.excluded_inputs
+                        and utxo.output.script is None
+                    ):
                         additional_utxo_pool.append(utxo)
                         additional_amount += utxo.output.amount
                         seen_utxos.add(utxo)

test/pycardano/test_txbuilder.py

@@ -2041,12 +2041,6 @@ def test_build_witness_set_mixed_scripts(chain_context):
     assert witness_set.plutus_v2_script is None
     assert witness_set.plutus_v3_script is None
 
-    # Test with remove_dup_script=False
-    witness_set = builder.build_witness_set(remove_dup_script=False)
-    assert len(witness_set.plutus_v1_script) == 2
-    assert len(witness_set.plutus_v2_script) == 1
-    assert len(witness_set.plutus_v3_script) == 1
-
 
 def test_add_script_input_post_chang(chain_context):
     tx_builder = TransactionBuilder(chain_context)