diff --git a/package/test_tools/__private/wallet/constants.py b/package/test_tools/__private/wallet/constants.py index 0e4e83a51d6d1289e4a60c5a19055c8fbbec7452..e5e5fabec01c1e7003e51f6f0280ace256db861f 100644 --- a/package/test_tools/__private/wallet/constants.py +++ b/package/test_tools/__private/wallet/constants.py @@ -8,6 +8,7 @@ from schemas.fields.basic import AccountName, EmptyString, PrivateKey, PublicKey from schemas.fields.compound import HbdExchangeRate from schemas.fields.hive_datetime import HiveDateTime from schemas.operations.representations.hf26_representation import HF26Representation +from schemas.operations.representations.legacy_representation import LegacyRepresentation from schemas.transaction import Transaction if TYPE_CHECKING: @@ -62,6 +63,11 @@ class SimpleTransaction(Transaction): self.operations.append(HF26Representation(type=operation.get_name_with_suffix(), value=operation)) +class SimpleTransactionLegacy(Transaction): + def add_operation(self, operation: AnyOperation) -> None: + self.operations.append(LegacyRepresentation(type=operation.get_name(), value=operation)) + + class WalletResponseBase(SimpleTransaction): transaction_id: str