test: update EIP-712 roundtrip tests

LedgerHQ · Oct 23, 2024 · 2dd1dbd · 2dd1dbd
1 parent 83e3b74
commit 2dd1dbd
Showing 1 changed file with 14 additions and 7 deletions.
diff --git a/tests/convert/ledger/eip712/test_convert_eip712_round_trip.py b/tests/convert/ledger/eip712/test_convert_eip712_round_trip.py
@@ -104,13 +104,20 @@ def _cleanup_formats(formats: dict[str, Any]) -> Any:
             del_by_path(message, "screens")
             if "fields" in message:
                 for field in message["fields"]:
-                    # Other formats are always converted to RAW
-                    if "format" in field and field["format"] not in (
-                        FieldFormat.AMOUNT,
-                        FieldFormat.TOKEN_AMOUNT,
-                        FieldFormat.DATE,
-                    ):
-                        field["format"] = "raw"
+                    if "format" in field:
+                        format = field["format"]
+
+                        # Address name format parameters cannot be preserved
+                        if format == "addressName":
+                            del_by_path(field, "params")
+
+                        # Other formats are always converted to RAW
+                        if format not in (
+                            FieldFormat.AMOUNT,
+                            FieldFormat.TOKEN_AMOUNT,
+                            FieldFormat.DATE,
+                        ):
+                            field["format"] = "raw"
 
         return formats