def test_payload(self): class MQTTMessage: def __init__(self, payload): self.payload = payload self.assertEqual( CommonsManager.payload(MQTTMessage("{'test': 2}")), dict()) self.assertEqual( CommonsManager.payload(MQTTMessage(None)), dict()) self.assertEqual( CommonsManager.payload(MQTTMessage(b'\x81')), dict()) self.assertEqual( CommonsManager.payload(MQTTMessage('{"test": 2}')), {'test': 2}) self.assertEqual( CommonsManager.payload(MQTTMessage(b'{"test": 2}')), {'test': 2}) self.assertEqual( CommonsManager.payload(MQTTMessage('true')), {'true': True}) self.assertEqual( CommonsManager.payload(MQTTMessage('false')), {'false': False}) self.assertEqual( CommonsManager.payload(MQTTMessage(b'true')), {'true': True})
def update(self, message: MQTTMessage): from core.commons.CommonsManager import CommonsManager self.message = message self.intentName = message.topic self.payload = CommonsManager.payload(message) self.slots.update(CommonsManager.parseSlots(message)) self.slotsAsObjects.update(CommonsManager.parseSlotsToObjects(message)) self.customData.update(CommonsManager.parseCustomData(message))
def extend(self, message: MQTTMessage, sessionId: str = None): if sessionId: self.sessionId = sessionId from core.commons.CommonsManager import CommonsManager self.message = message self.intentName = message.topic self.payload = CommonsManager.payload(message) self.slots = CommonsManager.parseSlots(message) self.slotsAsObjects = CommonsManager.parseSlotsToObjects(message) self.customData = CommonsManager.parseCustomData(message)
def test_payload(self): class MQTTMessage(object): def __init__(self, payload): self.payload = payload self.topic = 'test' self.assertEqual(CommonsManager.payload(MQTTMessage(None)), {'test': None}) self.assertEqual(CommonsManager.payload(MQTTMessage(b'\x81')), {'test': b'\x81'}) self.assertEqual(CommonsManager.payload(MQTTMessage('{"test": 2}')), {'test': 2}) self.assertEqual(CommonsManager.payload(MQTTMessage(b'{"test": 2}')), {'test': 2}) self.assertEqual(CommonsManager.payload(MQTTMessage('true')), {'test': True}) self.assertEqual(CommonsManager.payload(MQTTMessage('false')), {'test': False}) self.assertEqual(CommonsManager.payload(MQTTMessage(b'true')), {'test': True})