def test_serialisation_fipa(self):
"""Tests a Value Error flag for wrong CFP query."""
with pytest.raises(ValueError):
msg = FIPAMessage(performative=FIPAMessage.Performative.CFP,
message_id=0,
dialogue_reference=(str(0), ''),
target=1,
query=None)
with mock.patch(
"aea.protocols.fipa.message.FIPAMessage.Performative"
) as mock_performative_enum:
mock_performative_enum.CFP.value = "unknown"
FIPASerializer().encode(msg), "Raises Value Error"
with pytest.raises(ValueError):
msg.set("query", "Hello")
# query type is not supported
FIPASerializer().encode(msg)
with pytest.raises(ValueError):
cfp_msg = FIPAMessage(message_id=0,
dialogue_reference=(str(0), ''),
target=0,
performative=FIPAMessage.Performative.CFP,
query=b"hello")
cfp_msg.set("query", "hello")
fipa_msg = fipa_pb2.FIPAMessage()
fipa_msg.message_id = cfp_msg.get("message_id")
dialogue_reference = cast(Dict[str, str],
cfp_msg.get("dialogue_reference"))
fipa_msg.dialogue_starter_reference = dialogue_reference[0]
fipa_msg.dialogue_responder_reference = dialogue_reference[1]
fipa_msg.target = cfp_msg.get("target")
performative = fipa_pb2.FIPAMessage.CFP()
fipa_msg.cfp.CopyFrom(performative)
fipa_bytes = fipa_msg.SerializeToString()
# The encoded message is not a valid FIPA message.
FIPASerializer().decode(fipa_bytes)
with pytest.raises(ValueError):
cfp_msg = FIPAMessage(message_id=0,
dialogue_reference=(str(0), ''),
target=0,
performative=FIPAMessage.Performative.CFP,
query=b"hello")
with mock.patch(
"aea.protocols.fipa.message.FIPAMessage.Performative"
) as mock_performative_enum:
mock_performative_enum.CFP.value = "unknown"
fipa_msg = fipa_pb2.FIPAMessage()
fipa_msg.message_id = cfp_msg.get("message_id")
dialogue_reference = cast(Dict[str, str],
cfp_msg.get("dialogue_reference"))
fipa_msg.dialogue_starter_reference = dialogue_reference[0]
fipa_msg.dialogue_responder_reference = dialogue_reference[1]
fipa_msg.target = cfp_msg.get("target")
performative = fipa_pb2.FIPAMessage.CFP()
fipa_msg.cfp.CopyFrom(performative)
fipa_bytes = fipa_msg.SerializeToString()
# The encoded message is not a FIPA message
FIPASerializer().decode(fipa_bytes)
def test_fipa_cfp_serialization():
"""Test that the serialization for the 'fipa' protocol works."""
query = Query([Constraint('something', ConstraintType('>', 1))])
msg = FIPAMessage(message_id=0,
dialogue_reference=(str(0), ''),
target=0,
performative=FIPAMessage.Performative.CFP,
query=query)
msg_bytes = FIPASerializer().encode(msg)
envelope = Envelope(to="receiver",
sender="sender",
protocol_id=FIPAMessage.protocol_id,
message=msg_bytes)
envelope_bytes = envelope.encode()
actual_envelope = Envelope.decode(envelope_bytes)
expected_envelope = envelope
assert expected_envelope == actual_envelope
actual_msg = FIPASerializer().decode(actual_envelope.message)
expected_msg = msg
assert expected_msg == actual_msg
msg.set("query", "not_supported_query")
with pytest.raises(ValueError, match="Query type not supported:"):
FIPASerializer().encode(msg)
def test_serialisation_fipa(self):
"""Tests a Value Error flag for wrong CFP query."""
with pytest.raises(ValueError):
msg = FIPAMessage(performative=FIPAMessage.Performative.CFP,
message_id=0,
dialogue_id=0,
destination="publicKey",
target=1)
with mock.patch("aea.protocols.fipa.message.FIPAMessage.Performative")\
as mock_performative_enum:
mock_performative_enum.CFP.value = "unknown"
assert FIPASerializer().encode(msg), "Raises Value Error"
with pytest.raises(ValueError):
msg.set("query", "Hello")
assert FIPASerializer().encode(msg), "Query type is Supported!"
with pytest.raises(ValueError):
cfp_msg = FIPAMessage(message_id=0,
dialogue_id=0,
target=0,
performative=FIPAMessage.Performative.CFP,
query=b"hello")
cfp_msg.set("query", "hello")
fipa_msg = fipa_pb2.FIPAMessage()
fipa_msg.message_id = cfp_msg.get("id")
fipa_msg.dialogue_id = cfp_msg.get("dialogue_id")
fipa_msg.target = cfp_msg.get("target")
performative = fipa_pb2.FIPAMessage.CFP()
fipa_msg.cfp.CopyFrom(performative)
fipa_bytes = fipa_msg.SerializeToString()
assert FIPASerializer().decode(fipa_bytes),\
"The encoded message is a valid FIPA message."
with pytest.raises(ValueError):
cfp_msg = FIPAMessage(message_id=0,
dialogue_id=0,
target=0,
performative=FIPAMessage.Performative.CFP,
query=b"hello")
with mock.patch("aea.protocols.fipa.message.FIPAMessage.Performative")\
as mock_performative_enum:
mock_performative_enum.CFP.value = "unknown"
fipa_msg = fipa_pb2.FIPAMessage()
fipa_msg.message_id = cfp_msg.get("id")
fipa_msg.dialogue_id = cfp_msg.get("dialogue_id")
fipa_msg.target = cfp_msg.get("target")
performative = fipa_pb2.FIPAMessage.CFP()
fipa_msg.cfp.CopyFrom(performative)
fipa_bytes = fipa_msg.SerializeToString()
assert FIPASerializer().decode(fipa_bytes),\
"The encoded message is a FIPA message"