def decode(obj: bytes) -> Message:
"""
Decode bytes into a 'Signing' message.
:param obj: the bytes object.
:return: the 'Signing' message.
"""
message_pb = ProtobufMessage()
signing_pb = signing_pb2.SigningMessage()
message_pb.ParseFromString(obj)
message_id = message_pb.dialogue_message.message_id
dialogue_reference = (
message_pb.dialogue_message.dialogue_starter_reference,
message_pb.dialogue_message.dialogue_responder_reference,
)
target = message_pb.dialogue_message.target
signing_pb.ParseFromString(message_pb.dialogue_message.content)
performative = signing_pb.WhichOneof("performative")
performative_id = SigningMessage.Performative(str(performative))
performative_content = dict() # type: Dict[str, Any]
if performative_id == SigningMessage.Performative.SIGN_TRANSACTION:
pb2_terms = signing_pb.sign_transaction.terms
terms = Terms.decode(pb2_terms)
performative_content["terms"] = terms
pb2_raw_transaction = signing_pb.sign_transaction.raw_transaction
raw_transaction = RawTransaction.decode(pb2_raw_transaction)
performative_content["raw_transaction"] = raw_transaction
elif performative_id == SigningMessage.Performative.SIGN_MESSAGE:
pb2_terms = signing_pb.sign_message.terms
terms = Terms.decode(pb2_terms)
performative_content["terms"] = terms
pb2_raw_message = signing_pb.sign_message.raw_message
raw_message = RawMessage.decode(pb2_raw_message)
performative_content["raw_message"] = raw_message
elif performative_id == SigningMessage.Performative.SIGNED_TRANSACTION:
pb2_signed_transaction = signing_pb.signed_transaction.signed_transaction
signed_transaction = SignedTransaction.decode(
pb2_signed_transaction)
performative_content["signed_transaction"] = signed_transaction
elif performative_id == SigningMessage.Performative.SIGNED_MESSAGE:
pb2_signed_message = signing_pb.signed_message.signed_message
signed_message = SignedMessage.decode(pb2_signed_message)
performative_content["signed_message"] = signed_message
elif performative_id == SigningMessage.Performative.ERROR:
pb2_error_code = signing_pb.error.error_code
error_code = ErrorCode.decode(pb2_error_code)
performative_content["error_code"] = error_code
else:
raise ValueError(
"Performative not valid: {}.".format(performative_id))
return SigningMessage(message_id=message_id,
dialogue_reference=dialogue_reference,
target=target,
performative=performative,
**performative_content)
def encode(msg: Message) -> bytes:
"""
Encode a 'Signing' message into bytes.
:param msg: the message object.
:return: the bytes.
"""
msg = cast(SigningMessage, msg)
message_pb = ProtobufMessage()
dialogue_message_pb = DialogueMessage()
signing_msg = signing_pb2.SigningMessage()
dialogue_message_pb.message_id = msg.message_id
dialogue_reference = msg.dialogue_reference
dialogue_message_pb.dialogue_starter_reference = dialogue_reference[0]
dialogue_message_pb.dialogue_responder_reference = dialogue_reference[
1]
dialogue_message_pb.target = msg.target
performative_id = msg.performative
if performative_id == SigningMessage.Performative.SIGN_TRANSACTION:
performative = signing_pb2.SigningMessage.Sign_Transaction_Performative(
) # type: ignore
terms = msg.terms
Terms.encode(performative.terms, terms)
raw_transaction = msg.raw_transaction
RawTransaction.encode(performative.raw_transaction,
raw_transaction)
signing_msg.sign_transaction.CopyFrom(performative)
elif performative_id == SigningMessage.Performative.SIGN_MESSAGE:
performative = signing_pb2.SigningMessage.Sign_Message_Performative(
) # type: ignore
terms = msg.terms
Terms.encode(performative.terms, terms)
raw_message = msg.raw_message
RawMessage.encode(performative.raw_message, raw_message)
signing_msg.sign_message.CopyFrom(performative)
elif performative_id == SigningMessage.Performative.SIGNED_TRANSACTION:
performative = signing_pb2.SigningMessage.Signed_Transaction_Performative(
) # type: ignore
signed_transaction = msg.signed_transaction
SignedTransaction.encode(performative.signed_transaction,
signed_transaction)
signing_msg.signed_transaction.CopyFrom(performative)
elif performative_id == SigningMessage.Performative.SIGNED_MESSAGE:
performative = signing_pb2.SigningMessage.Signed_Message_Performative(
) # type: ignore
signed_message = msg.signed_message
SignedMessage.encode(performative.signed_message, signed_message)
signing_msg.signed_message.CopyFrom(performative)
elif performative_id == SigningMessage.Performative.ERROR:
performative = signing_pb2.SigningMessage.Error_Performative(
) # type: ignore
error_code = msg.error_code
ErrorCode.encode(performative.error_code, error_code)
signing_msg.error.CopyFrom(performative)
else:
raise ValueError(
"Performative not valid: {}".format(performative_id))
dialogue_message_pb.content = signing_msg.SerializeToString()
message_pb.dialogue_message.CopyFrom(dialogue_message_pb)
message_bytes = message_pb.SerializeToString()
return message_bytes
def setup(cls, **kwargs):
"""Setup the test class."""
super().setup(**kwargs)
cls.ledger_api_handler = cast(
LedgerApiHandler, cls._skill.skill_context.handlers.ledger_api)
cls.logger = cls._skill.skill_context.logger
cls.simple_oracle_behaviour = cast(
SimpleOracleBehaviour,
cls._skill.skill_context.behaviours.simple_oracle_behaviour,
)
cls.ledger_api_dialogues = cast(
LedgerApiDialogues, cls._skill.skill_context.ledger_api_dialogues)
cls.contract_api_dialogues = cast(
ContractApiDialogues,
cls._skill.skill_context.contract_api_dialogues)
cls.signing_dialogues = cast(
SigningDialogues, cls._skill.skill_context.signing_dialogues)
cls.list_of_ledger_api_messages = (
DialogueMessage(
LedgerApiMessage.Performative.GET_BALANCE,
{
"ledger_id": LEDGER_ID,
"address": "some_eth_address"
},
),
DialogueMessage(
LedgerApiMessage.Performative.SEND_SIGNED_TRANSACTION,
{
"signed_transaction": SignedTransaction(
LEDGER_ID, DEFAULT_TX)
},
),
DialogueMessage(
LedgerApiMessage.Performative.GET_TRANSACTION_RECEIPT,
{
"transaction_digest":
TransactionDigest(LEDGER_ID, "some_digest")
},
),
DialogueMessage(
LedgerApiMessage.Performative.GET_STATE,
{
"ledger_id": LEDGER_ID,
"callable": "some_callable",
"args": (),
"kwargs": LedgerApiMessage.Kwargs({}),
},
),
)
cls.list_of_contract_api_messages = (DialogueMessage(
ContractApiMessage.Performative.GET_DEPLOY_TRANSACTION,
{
"ledger_id": LEDGER_ID,
"contract_id": "some_contract_id",
"callable": "some_callable",
"kwargs": ContractApiKwargs({"some_key": "some_value"}),
},
), )
cls.list_of_signing_messages = (DialogueMessage(
SigningMessage.Performative.SIGN_TRANSACTION,
{
"terms": Terms(*DEFAULT_TERMS),
"raw_transaction": RawTransaction(LEDGER_ID, DEFAULT_TX),
},
), )