def test_raw_message_serialization():
"""Test the serialization for 'raw_message' speech-act works."""
raw_message_arg = ContractApiMessage.RawMessage("some_ledger_id",
b"some_body")
msg = ContractApiMessage(
performative=ContractApiMessage.Performative.RAW_MESSAGE,
raw_message=raw_message_arg,
)
msg.to = "receiver"
envelope = Envelope(
to=msg.to,
sender="sender",
message=msg,
)
envelope_bytes = envelope.encode()
actual_envelope = Envelope.decode(envelope_bytes)
expected_envelope = envelope
assert expected_envelope.to == actual_envelope.to
assert expected_envelope.sender == actual_envelope.sender
assert (expected_envelope.protocol_specification_id ==
actual_envelope.protocol_specification_id)
assert expected_envelope.message != actual_envelope.message
actual_msg = ContractApiMessage.serializer.decode(actual_envelope.message)
actual_msg.to = actual_envelope.to
actual_msg.sender = actual_envelope.sender
expected_msg = msg
assert expected_msg == actual_msg
def test_incorrect_message(mocked_enforce):
"""Test that we raise an exception when the message is incorrect."""
with mock.patch.object(contract_api_message_logger, "error") as mock_logger:
ContractApiMessage(
message_id=1,
dialogue_reference=(str(0), ""),
target=0,
performative=ContractApiMessage.Performative.RAW_MESSAGE,
raw_message=ContractApiMessage.RawMessage("some_ledger_id", b"some_body"),
)
mock_logger.assert_any_call("some error")
def test_encoding_unknown_performative():
"""Test that we raise an exception when the performative is unknown during encoding."""
msg = ContractApiMessage(
message_id=1,
dialogue_reference=(str(0), ""),
target=0,
performative=ContractApiMessage.Performative.RAW_MESSAGE,
raw_message=ContractApiMessage.RawMessage("some_ledger_id", b"some_body"),
)
with pytest.raises(ValueError, match="Performative not valid:"):
with mock.patch.object(
ContractApiMessage.Performative, "__eq__", return_value=False
):
ContractApiMessage.serializer.encode(msg)