def validateNodeMsg(self, wrappedMsg): node_message_factory.set_message_class(TestMsg) return super().validateNodeMsg(wrappedMsg)
from plenum.common.messages.message_base import MessageBase from plenum.common.messages.node_message_factory import node_message_factory from plenum.common.messages.node_messages import NonEmptyStringField from plenum.common.util import randomString def randomMsg(): return TestMsg('subject ' + randomString(), 'content ' + randomString()) class TestMsg(MessageBase): typename = "TESTMSG" schema = ( ("subject", NonEmptyStringField()), ("content", NonEmptyStringField()), ) node_message_factory.set_message_class(TestMsg)
def validateNodeMsg(self, wrappedMsg): node_message_factory.set_message_class(TestMsg) return super().validateNodeMsg(wrappedMsg)
from plenum.common.messages.fields import NonEmptyStringField from plenum.common.messages.message_base import MessageBase from plenum.common.messages.node_message_factory import node_message_factory from plenum.common.util import randomString def randomMsg(): return TestMsg('subject ' + randomString(), 'content ' + randomString()) class TestMsg(MessageBase): typename = "TESTMSG" schema = ( ("subject", NonEmptyStringField()), ("content", NonEmptyStringField()), ) node_message_factory.set_message_class(TestMsg)