test: add tests for network magic and DRep serialization and TextEnvelope save and load

Author: KINGH242

test/pycardano/backend/test_cardano_cli.py

@@ -21,6 +21,7 @@
     TransactionInput,
     CardanoCliError,
 )
+from pycardano.backend.cardano_cli import network_magic
 
 QUERY_TIP_RESULT = {
     "block": 1460093,
@@ -647,6 +648,10 @@ def override_run_command_fail(cmd: List[str]):
     return context
 
 
+def test_network_magic():
+    assert network_magic(1) == ["--testnet-magic", "1"]
+
+
 class TestCardanoCliChainContext:
     def test_protocol_param(self, chain_context):
         assert (

test/pycardano/test_certificate.py

@@ -7,7 +7,9 @@
 from pycardano.certificate import (
     Anchor,
     AuthCommitteeHotCertificate,
+    DRep,
     DRepCredential,
+    DRepKind,
     PoolRegistration,
     PoolRetirement,
     ResignCommitteeColdCertificate,
@@ -22,8 +24,10 @@
 from pycardano.hash import (  # plutus_script_hash,
     POOL_KEY_HASH_SIZE,
     SCRIPT_HASH_SIZE,
+    VERIFICATION_KEY_HASH_SIZE,
     PoolKeyHash,
     ScriptHash,
+    VerificationKeyHash,
 )
 
 TEST_ADDR = Address.from_primitive(
@@ -197,6 +201,104 @@ def test_drep_credential():
     assert DRepCredential.from_cbor(drep_credential_cbor_hex) == drep_credential
 
 
+@pytest.mark.parametrize(
+    "input_values,expected_kind,expected_credential,expected_exception,case_id",
+    [
+        # Happy path: VERIFICATION_KEY_HASH
+        (
+            [DRepKind.VERIFICATION_KEY_HASH.value, b"1" * VERIFICATION_KEY_HASH_SIZE],
+            DRepKind.VERIFICATION_KEY_HASH,
+            VerificationKeyHash(b"1" * VERIFICATION_KEY_HASH_SIZE),
+            None,
+            "verification_key_hash",
+        ),
+        # Happy path: SCRIPT_HASH
+        (
+            [DRepKind.SCRIPT_HASH.value, b"1" * SCRIPT_HASH_SIZE],
+            DRepKind.SCRIPT_HASH,
+            ScriptHash(b"1" * SCRIPT_HASH_SIZE),
+            None,
+            "script_hash",
+        ),
+        # Happy path: ALWAYS_ABSTAIN
+        (
+            [DRepKind.ALWAYS_ABSTAIN.value],
+            DRepKind.ALWAYS_ABSTAIN,
+            None,
+            None,
+            "always_abstain",
+        ),
+        # Happy path: ALWAYS_NO_CONFIDENCE
+        (
+            [DRepKind.ALWAYS_NO_CONFIDENCE.value],
+            DRepKind.ALWAYS_NO_CONFIDENCE,
+            None,
+            None,
+            "always_no_confidence",
+        ),
+        # Error: invalid DRepKind value (not in enum)
+        ([99], None, None, DeserializeException, "invalid_drep_kind"),
+        # Error: valid kind but missing credential for VERIFICATION_KEY_HASH
+        (
+            [DRepKind.VERIFICATION_KEY_HASH.value],
+            None,
+            None,
+            IndexError,
+            "missing_credential_verification_key_hash",
+        ),
+        # Error: valid kind but missing credential for SCRIPT_HASH
+        (
+            [DRepKind.SCRIPT_HASH.value],
+            None,
+            None,
+            IndexError,
+            "missing_credential_script_hash",
+        ),
+        # Error: valid kind but extra credential for ALWAYS_ABSTAIN
+        (
+            [DRepKind.ALWAYS_ABSTAIN.value, b"extra"],
+            DRepKind.ALWAYS_ABSTAIN,
+            None,
+            None,
+            "abstain_with_extra",
+        ),
+        # Error: valid kind but extra credential for ALWAYS_NO_CONFIDENCE
+        (
+            [DRepKind.ALWAYS_NO_CONFIDENCE.value, b"extra"],
+            DRepKind.ALWAYS_NO_CONFIDENCE,
+            None,
+            None,
+            "no_confidence_with_extra",
+        ),
+        # Error: input is empty
+        ([], None, None, IndexError, "empty_input"),
+        # Error: input is not a list or tuple
+        ("notalist", None, None, DeserializeException, "input_not_list"),
+    ],
+    ids=lambda p: p if isinstance(p, str) else None,
+)
+def test_drep_from_primitive(
+    input_values, expected_kind, expected_credential, expected_exception, case_id
+):
+
+    # Arrange
+    # (All input values are provided via test parameters)
+
+    # Act / Assert
+    if expected_exception:
+        with pytest.raises(expected_exception):
+            DRep.from_primitive(input_values)
+    else:
+        result = DRep.from_primitive(input_values)
+        # Assert
+        assert result.kind == expected_kind
+        if expected_credential is not None:
+            assert isinstance(result.credential, type(expected_credential))
+            assert result.credential.payload == expected_credential.payload
+        else:
+            assert result.credential is None
+
+
 def test_unreg_drep_certificate():
     staking_key = StakeSigningKey.from_cbor(
         "5820ff3a330df8859e4e5f42a97fcaee73f6a00d0cf864f4bca902bd106d423f02c0"

test/pycardano/test_transaction.py

@@ -1,3 +1,4 @@
+import tempfile
 from dataclasses import dataclass
 from fractions import Fraction
 from test.pycardano.util import check_two_way_cbor
@@ -243,6 +244,32 @@ def test_transaction():
     assert expected_tx_id == signed_tx.id
 
 
+def test_transaction_save_load():
+    tx_cbor = (
+        "84a70081825820b35a4ba9ef3ce21adcd6879d08553642224304704d206c74d3ffb3e6eed3ca28000d80018182581d60cc"
+        "30497f4ff962f4c1dca54cceefe39f86f1d7179668009f8eb71e598200a1581cec8b7d1dd0b124e8333d3fa8d818f6eac0"
+        "68231a287554e9ceae490ea24f5365636f6e6454657374746f6b656e1a009896804954657374746f6b656e1a0098968002"
+        "1a000493e00e8009a1581cec8b7d1dd0b124e8333d3fa8d818f6eac068231a287554e9ceae490ea24f5365636f6e645465"
+        "7374746f6b656e1a009896804954657374746f6b656e1a00989680075820592a2df0e091566969b3044626faa8023dabe6"
+        "f39c78f33bed9e105e55159221a200828258206443a101bdb948366fc87369336224595d36d8b0eee5602cba8b81a024e5"
+        "84735840846f408dee3b101fda0f0f7ca89e18b724b7ca6266eb29775d3967d6920cae7457accb91def9b77571e15dd2ed"
+        "e38b12cf92496ce7382fa19eb90ab7f73e49008258205797dc2cc919dfec0bb849551ebdf30d96e5cbe0f33f734a87fe82"
+        "6db30f7ef95840bdc771aa7b8c86a8ffcbe1b7a479c68503c8aa0ffde8059443055bf3e54b92f4fca5e0b9ca5bb11ab23b"
+        "1390bb9ffce414fa398fc0b17f4dc76fe9f7e2c99c09018182018482051a075bcd1582041a075bcd0c8200581c9139e5c0"
+        "a42f0f2389634c3dd18dc621f5594c5ba825d9a8883c66278200581c835600a2be276a18a4bebf0225d728f090f724f4c0"
+        "acd591d066fa6ff5d90103a100a11902d1a16b7b706f6c6963795f69647da16d7b706f6c6963795f6e616d657da66b6465"
+        "736372697074696f6e6a3c6f7074696f6e616c3e65696d6167656a3c72657175697265643e686c6f636174696f6ea36761"
+        "7277656176656a3c6f7074696f6e616c3e6568747470736a3c6f7074696f6e616c3e64697066736a3c7265717569726564"
+        "3e646e616d656a3c72657175697265643e667368613235366a3c72657175697265643e64747970656a3c72657175697265643e"
+    )
+    tx = Transaction.from_cbor(tx_cbor)
+
+    with tempfile.NamedTemporaryFile() as f:
+        tx.save(f.name)
+        loaded_tx = Transaction.load(f.name)
+        assert tx == loaded_tx
+
+
 def test_multi_asset():
     serialized_value = [
         100,

test/pycardano/test_witness.py

@@ -0,0 +1,22 @@
+import tempfile
+
+from pycardano import (
+    PaymentKeyPair,
+    PaymentSigningKey,
+    PaymentVerificationKey,
+    VerificationKeyWitness,
+)
+
+
+def test_witness_save_load():
+    sk = PaymentSigningKey.generate()
+    vk = PaymentVerificationKey.from_signing_key(sk)
+    witness = VerificationKeyWitness(
+        vkey=vk,
+        signature=sk.sign(b"test"),
+    )
+
+    with tempfile.NamedTemporaryFile() as f:
+        witness.save(f.name)
+        loaded_witness = VerificationKeyWitness.load(f.name)
+        assert witness == loaded_witness