test: add save and load tests for Address and CBORSerializable classes

Author: KINGH242

test/pycardano/test_address.py

@@ -1,3 +1,5 @@
+import tempfile
+
 import pytest
 
 from pycardano.address import Address, AddressType, PointerAddress
@@ -211,3 +213,13 @@ def test_from_primitive_invalid_type_addr():
 
     with pytest.raises(DeserializeException):
         Address.from_primitive(value)
+
+
+def test_save_load_address():
+    address_string = "addr_test1vr2p8st5t5cxqglyjky7vk98k7jtfhdpvhl4e97cezuhn0cqcexl7"
+    address = Address.from_primitive(address_string)
+
+    with tempfile.NamedTemporaryFile() as f:
+        address.save(f.name)
+        loaded_address = Address.load(f.name)
+        assert address == loaded_address

test/pycardano/test_serialization.py

@@ -26,6 +26,7 @@
     CBORBase,
     Datum,
     MultiAsset,
+    Primitive,
     RawPlutusData,
     Transaction,
     TransactionWitnessSet,
@@ -49,7 +50,6 @@
     NonEmptyOrderedSet,
     OrderedSet,
     RawCBOR,
-    TextEnvelope,
     default_encoder,
     limit_primitive_type,
 )
@@ -991,39 +991,37 @@ class TestData(MapCBORSerializable):
     assert s_copy[0].value == 100
 
 
-def test_text_envelope():
+def test_save_load():
     @dataclass
-    class Test1(ArrayCBORSerializable, TextEnvelope):
+    class Test1(CBORSerializable):
         a: str
         b: Union[str, None] = None
 
-        KEY_TYPE = "Test1"
-        DESCRIPTION = "A test class for TextEnvelope serialization"
-
-        def __init__(
-            self,
-            a: str,
-            b: Union[str, None] = None,
-            payload: Optional[bytes] = None,
-            key_type: Optional[str] = None,
-            description: Optional[str] = None,
-        ):
-            self.a = a
-            self.b = b
-            TextEnvelope.__init__(self, payload, key_type, description)
+        @property
+        def json_type(self) -> str:
+            return "Test Type"
 
-    test1 = Test1(a="a")
+        @property
+        def json_description(self) -> str:
+            return "Test Description"
 
-    wrong_type = {
-        "type": "Test2",
-        "description": "A test class for TextEnvelope serialization",
-        "cborHex": "826161f6",
-    }
+        @classmethod
+        def from_primitive(
+            cls: Type[CBORSerializable], value: Any, type_args: Optional[tuple] = None
+        ) -> CBORSerializable:
+            if not isinstance(value, dict):
+                raise DeserializeException(f"Expected dict, got {type(value)}")
+            return Test1(a=value["a"], b=value.get("b"))
 
-    with pytest.raises(InvalidKeyTypeException):
-        invalid_test1 = Test1.from_json(json.dumps(wrong_type), validate_type=True)
+        def to_shallow_primitive(self) -> Union[Primitive, CBORSerializable]:
+            return {"a": self.a, "b": self.b}
+
+    test1 = Test1(a="a")
+    test1_json = json.loads(test1.to_json())
 
-    assert test1.payload == b"\x82aa\xf6"
+    assert test1_json["type"] == "Test Type"
+    assert test1_json["description"] == "Test Description"
+    assert test1_json["cborHex"] == test1.to_cbor_hex()
 
     with tempfile.NamedTemporaryFile() as f:
         test1.save(f.name)