def _compare(self, obj1, obj2): msg_type = to_message_type(obj1) msg_type2 = to_message_type(obj2) if msg_type != msg_type2: return False if msg_type in self._comparators: return self._comparators[msg_type](obj1, obj2) return obj1 == obj2
def _compare(self, obj1, obj2): msg_type = to_message_type(obj1) msg_type2 = to_message_type(obj1) if msg_type != msg_type2: raise UnexpectedMessageException(msg_type, obj1, obj2) if msg_type in self._comparators: return self._comparators[msg_type](obj1, obj2) return obj1 == obj2
def __init__(self, message_type, expected, received): super().__init__("{}: Expected {}({}):'{}', Got {}({}):'{}'".format( to_protobuf_class(message_type).__name__, to_protobuf_class(to_message_type(expected)), to_message_type(expected), str(expected).strip(), to_protobuf_class(to_message_type(received)), to_message_type(received), str(received).strip())) self.message_type_name = to_protobuf_class(message_type).__name__ self.expected = expected self.received = received
def __init__(self, message_type, expected, received): super().__init__("{}: Expected {}({}):'{}', Got {}({}):'{}'".format( to_protobuf_class(message_type).__name__, to_protobuf_class(to_message_type(expected)), to_message_type(expected), str(expected).strip(), to_protobuf_class(to_message_type(received)), to_message_type(received), str(received).strip() )) self.message_type_name = to_protobuf_class(message_type).__name__ self.expected = expected self.received = received
def _compare(self, obj1, obj2): msg_type = to_message_type(obj1) if msg_type in self._comparators: return self._comparators[msg_type](obj1, obj2) else: return obj1 == obj2
def send(self, message_content, correlation_id=None): """ Convert the message content to a protobuf message, including serialization of the content and insertion of the content type. Optionally include the correlation id in the message. Messages are sent with the name of this class as the sender. """ message = Message(message_type=to_message_type(message_content), content=message_content.SerializeToString(), correlation_id=correlation_id, sender=self.__class__.__name__) return self._loop.run_until_complete( self._send(self._tp_ident, message))
def send(self, message_content, correlation_id=None): """ Convert the message content to a protobuf message, including serialization of the content and insertion of the content type. Optionally include the correlation id in the message. Messages are sent with the name of this class as the sender. """ message = Message( message_type=to_message_type(message_content), content=message_content.SerializeToString(), correlation_id=correlation_id, ) return self._loop.run_until_complete( self._send(self._tp_ident, message))