Merge branch 'Python-Cardano:main' into feat_cert_script

Author: SCMusson

pycardano/backend/ogmios_v6.py

@@ -263,7 +263,7 @@ def _utxo_from_ogmios_result(self, utxo: OgmiosUtxo) -> UTxO:
             # TODO: Need to test with native scripts
             if script["language"] == "plutus:v3":
                 script = PlutusV3Script(bytes.fromhex(script["cbor"]))
-            if script["language"] == "plutus:v2":
+            elif script["language"] == "plutus:v2":
                 script = PlutusV2Script(bytes.fromhex(script["cbor"]))
             elif script["language"] == "plutus:v1":
                 script = PlutusV1Script(bytes.fromhex(script["cbor"]))

pycardano/certificate.py

@@ -61,7 +61,10 @@ def __post_init__(self):
     def from_primitive(
         cls: Type[StakeRegistration], values: Union[list, tuple]
     ) -> StakeRegistration:
-        return cls(stake_credential=StakeCredential.from_primitive(values[1]))
+        if values[0] == 0:
+            return cls(stake_credential=StakeCredential.from_primitive(values[1]))
+        else:
+            raise DeserializeException(f"Invalid StakeRegistration type {values[0]}")
 
 
 @dataclass(repr=False)

test/pycardano/test_certificate.py

@@ -1,3 +1,6 @@
+import os
+
+from pycardano import StakeSigningKey, TransactionBody
 from pycardano.address import Address
 from pycardano.certificate import (
     PoolRegistration,
@@ -121,3 +124,28 @@ def test_pool_retirement():
     )
 
     assert PoolRetirement.from_cbor(pool_retirement_cbor_hex) == pool_retirement
+
+
+def test_staking_certificate_serdes():
+    staking_key = StakeSigningKey.generate()
+    stake_pool_key_hash = os.urandom(28)
+
+    transaction_body = TransactionBody(
+        certificates=[
+            StakeRegistration(
+                stake_credential=StakeCredential(
+                    staking_key.to_verification_key().hash()
+                )
+            ),
+            StakeDelegation(
+                stake_credential=StakeCredential(
+                    staking_key.to_verification_key().hash()
+                ),
+                pool_keyhash=PoolKeyHash(stake_pool_key_hash),
+            ),
+        ]
+    )
+
+    after_serdes = TransactionBody.from_cbor(transaction_body.to_cbor())
+
+    assert after_serdes == transaction_body